Update app.py

app.py

@@ -3,7 +3,25 @@ from transformers import pipeline
 
 generator = pipeline("text-generation", model="microsoft/DialoGPT-small")
 
-def simple_gen(text):
-    return generator(text, max_length=50)[0]['generated_text']
+def chat_fn(message, history=None):
+    print(f"User message: {message}")
+    if history is None:
+        history = []
+    conversation = ""
+    for user_msg, bot_msg in history:
+        conversation += f"User: {user_msg}\nBot: {bot_msg}\n"
+    conversation += f"User: {message}\nBot:"
+    print(f"Built conversation prompt:\n{conversation}")
 
-gr.Interface(simple_gen, gr.Textbox(), gr.Textbox()).launch()
+    try:
+        response = generator(conversation, max_length=200, pad_token_id=50256)
+        bot_reply = response[0]['generated_text'].split("Bot:")[-1].strip()
+        print(f"Bot reply: {bot_reply}")
+    except Exception as e:
+        bot_reply = f"Error generating response: {e}"
+        print(f"Error: {e}")
+
+    history.append((message, bot_reply))
+    return bot_reply, history
+
+gr.ChatInterface(fn=chat_fn).launch()