Update app.py

app.py

@@ -75,10 +75,9 @@ def predict_chat(message: str, history: list):
         yield "Error: Model or tokenizer failed to load. Please check the Space logs for details."
         return
 
-    messages = [{"role": "system", "content": "You are a friendly chatbot."}]
-    for human_msg, ai_msg in history:
-        messages.append({"role": "user", "content": human_msg})
-        messages.append({"role": "assistant", "content": ai_msg})
+    # history is already in the 'messages' format if type='messages' is set on chatbot
+    # It contains dictionaries with 'role' and 'content'
+    messages = [{"role": "system", "content": "You are a friendly chatbot."}] + history 
     messages.append({"role": "user", "content": message})
 
     generated_text = ""
@@ -131,18 +130,16 @@ def predict_chat(message: str, history: list):
 if __name__ == "__main__":
     load_model_for_zerocpu()
 
-    initial_chatbot_message = (
+    # Initial message for the chatbot in the 'messages' format
+    initial_messages = [{"role": "assistant", "content": 
         "Hello! I'm an AI assistant. I'm currently running in a CPU-only "
         "environment for efficient demonstration. How can I help you today?"
-    )
+    }]
 
-    # Use gr.ChatInterface directly without gr.Blocks wrapper for simplicity
-    # This often works better when ChatInterface is the sole component
     demo = gr.ChatInterface(
         fn=predict_chat,
-        # Define the chatbot here, with type='messages'
-        chatbot=gr.Chatbot(height=500, type='messages',
-                           value=[[None, initial_chatbot_message]]), # Set initial message directly here
+        # Define the chatbot here, with type='messages' and initial value in the correct format
+        chatbot=gr.Chatbot(height=500, type='messages', value=initial_messages),
         textbox=gr.Textbox(
             placeholder="Ask me a question...",
             container=False,
@@ -162,9 +159,7 @@ if __name__ == "__main__":
             ["What's the best way to stay motivated?"],
         ],
         cache_examples=False,
-        # Gradio 4.x has `clear_btn` directly on ChatInterface again
-        # but if this causes issues, you might need to revert to a gr.ClearButton() below
-        clear_btn="Clear Chat" # Re-added clear_btn as it seems to be supported again in latest Gradio versions
+        clear_btn="Clear Chat" 
     )
 
     demo.launch()