Update app.py

app.py

@@ -25,11 +25,15 @@ print("Model ready!")
 def generate_response(message, history, temperature, top_p, top_k, max_tokens):
     """Generate response from the model with streaming"""
     
-    # Build conversation history
+    # Build conversation history - history is list of [user_msg, bot_msg] pairs
     messages = []
     for user_msg, assistant_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": assistant_msg})
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if assistant_msg:
+            messages.append({"role": "assistant", "content": assistant_msg})
+    
+    # Add current message
     messages.append({"role": "user", "content": message})
     
     # Apply chat template
@@ -86,7 +90,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     chatbot = gr.Chatbot(
         height=500,
         label="Chat History",
-        type="messages",
         avatar_images=(None, "https://em-content.zobj.net/source/twitter/53/robot-face_1f916.png")
     )
     
@@ -212,4 +215,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch()