Update app.py

app.py

@@ -53,9 +53,19 @@ def generate_text(prompt, max_new_tokens=80, temperature=0.7):
 def chat_interface(message, history):
     """Formats the Gradio UI input to match your dataset's User/Bot structure."""
     prompt = ""
-    # Inject context memory
-    for user_msg, bot_msg in history:
-        prompt += f"User: {user_msg}\nBot: {bot_msg}\n"
+    
+    # Inject context memory (Handles both Gradio 4 and Gradio 5 formats)
+    for msg in history:
+        # Gradio 5 format: dictionaries with 'role' and 'content'
+        if isinstance(msg, dict):
+            if msg.get("role") == "user":
+                prompt += f"User: {msg.get('content')}\n"
+            elif msg.get("role") == "assistant":
+                prompt += f"Bot: {msg.get('content')}\n"
+                
+        # Fallback for Gradio 4 format: [user_msg, bot_msg]
+        elif isinstance(msg, (list, tuple)) and len(msg) == 2:
+            prompt += f"User: {msg[0]}\nBot: {msg[1]}\n"
     
     # Inject current message
     prompt += f"User: {message}\nBot:"