Update app.py

app.py

@@ -160,14 +160,10 @@ def create_chat_interface():
             history.append([message, "❌ Please select an AI model first"])
             return history, ""
         
-        # Show typing indicator
-        history.append([message, "🤔 Thinking..."])
-        yield history, ""
-        
         # Convert gradio history to API format
         conversation_history = []
-        for i, (user_msg, ai_msg) in enumerate(history[:-1]):  # Exclude the current "thinking" message
-            if user_msg and ai_msg and ai_msg != "🤔 Thinking...":
+        for user_msg, ai_msg in history:
+            if user_msg and ai_msg:
                 conversation_history.append({"role": "user", "content": user_msg})
                 conversation_history.append({"role": "assistant", "content": ai_msg})
         
@@ -175,13 +171,13 @@ def create_chat_interface():
         result = chat_ai.send_message(selected_model, message, conversation_history)
         
         if result["success"]:
-            # Update the last message with the real response
-            history[-1] = [message, result["response"]]
-            yield history, ""
+            # Add the new conversation to history
+            history.append([message, result["response"]])
+            return history, ""
         else:
-            # Update with error message
-            history[-1] = [message, f"❌ Error: {result['error']}"]
-            yield history, ""
+            # Add error message to history
+            history.append([message, f"❌ Error: {result['error']}"])
+            return history, ""
     
     def clear_chat():
         """Clear the chat history"""
@@ -285,7 +281,8 @@ Ready to chat! Type your message below.
         
         # Chat functionality
         def submit_message(message, history, model):
-            return chat_with_ai(message, history, model)
+            new_history, cleared_input = chat_with_ai(message, history, model)
+            return new_history, ""  # Return updated history and clear the input
         
         # Send message on button click or enter
         send_btn.click(