Update app.py

app.py

@@ -1,34 +1,39 @@
 import gradio as gr
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 
-MODEL_ID = "microsoft/DialoGPT-medium"  # small free chatbot model
+MODEL_ID = "microsoft/DialoGPT-medium"  # small chatbot model
 
 # Load model & tokenizer
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 model = AutoModelForCausalLM.from_pretrained(MODEL_ID)
-
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, return_full_text=False)
 
 def chat(history, message):
+    # Convert list of tuples -> plain text context
     prompt = ""
-    for i in range(max(0, len(history)-6), len(history), 2):
-        try:
-            prompt += f"User: {history[i]}\nBot: {history[i+1]}\n"
-        except:
-            pass
+    for user, bot in history:
+        prompt += f"User: {user}\nBot: {bot}\n"
     prompt += f"User: {message}\nBot:"
+
     out = generator(prompt, max_length=200, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=1)
     reply = out[0]['generated_text']
+
     if "Bot:" in reply:
         reply = reply.split("Bot:")[-1].strip()
-    history = history + [message, reply]
+
+    # Append new message pair as tuple (user, bot)
+    history.append((message, reply))
     return history, history
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Chatapt")
+    gr.Markdown("## 🤖 Chatapt")
+
     chatbot = gr.Chatbot()
     state = gr.State([])
-    msg = gr.Textbox(show_label=False, placeholder="Type a message and press Enter")
+
+    with gr.Row():
+        msg = gr.Textbox(show_label=False, placeholder="Type a message and press Enter...")
+
     msg.submit(chat, [state, msg], [state, chatbot])
 
 demo.launch()