Update app.py

app.py

@@ -7,12 +7,20 @@ tokenizer = BlenderbotTokenizer.from_pretrained(MODEL_NAME)
 model = BlenderbotForConditionalGeneration.from_pretrained(MODEL_NAME)
 
 def chatbot_response(user_input, chat_history=[]):
-    """Generates a response from Blenderbot based on user input."""
-    inputs = tokenizer(user_input, return_tensors="pt")
-    reply_ids = model.generate(**inputs)
+    """Generates a response from Blenderbot with memory."""
+
+    # Format input with chat history
+    history_text = " ".join([f"User: {msg[0]} Assistant: {msg[1]}" for msg in chat_history])
+    
+    formatted_input = f"{history_text} User: {user_input} Assistant:"
+    inputs = tokenizer(formatted_input, return_tensors="pt")
+    reply_ids = model.generate(**inputs, max_length=100)
     response = tokenizer.decode(reply_ids[0], skip_special_tokens=True)
-    chat_history.append((user_input, response))
-    return response, chat_history
+
+    # Update chat history correctly
+    chat_history.append((user_input, response))  # ✅ Now a tuple (user message, bot response)
+
+    return chat_history  # ✅ Returns history formatted correctly
 
 # Set up Gradio interface
 with gr.Blocks() as demo: