Update app.py

app.py

@@ -284,40 +284,34 @@ with gr.Blocks(css=css) as demo:
         outputs=model_status
     )
     
-    # Handle message submission
-    def user_message_submitted(message, chat_history):
-        """Handle user submitted message"""
-        # Ensure chat_history is a list
-        if chat_history is None:
-            chat_history = []
-            
+    # Bind message submission
+    def process_message(message, chat_history):
+        """Process user message and generate response"""
         if message.strip() == "":
             return "", chat_history
         
         # Add user message to chat history
         chat_history = list(chat_history)
         chat_history.append((message, None))
+        
+        # Generate response
+        response = generate_response_non_streaming(message, model_name.value, temperature.value, max_tokens.value)
+        
+        # Add response to chat history
+        chat_history[-1] = (message, response)
+        
         return "", chat_history
-    
-    # Bind message submission
+
     submit_event = user_input.submit(
-        fn=user_message_submitted,
+        fn=process_message,
         inputs=[user_input, chatbot],
         outputs=[user_input, chatbot]
-    ).then(
-        fn=update_chat_with_response,
-        inputs=[chatbot, user_input, model_name, temperature, max_tokens],
-        outputs=chatbot
     )
-    
+
     submit_button.click(
-        fn=user_message_submitted,
+        fn=process_message,
         inputs=[user_input, chatbot],
         outputs=[user_input, chatbot]
-    ).then(
-        fn=update_chat_with_response,
-        inputs=[chatbot, user_input, model_name, temperature, max_tokens],
-        outputs=chatbot
     )
     
     # Clear chat
@@ -326,6 +320,48 @@ with gr.Blocks(css=css) as demo:
         outputs=chatbot
     )
     
+    # # Handle message submission
+    # def user_message_submitted(message, chat_history):
+    #     """Handle user submitted message"""
+    #     # Ensure chat_history is a list
+    #     if chat_history is None:
+    #         chat_history = []
+            
+    #     if message.strip() == "":
+    #         return "", chat_history
+        
+    #     # Add user message to chat history
+    #     chat_history = list(chat_history)
+    #     chat_history.append((message, None))
+    #     return "", chat_history
+    
+    # # Bind message submission
+    # submit_event = user_input.submit(
+    #     fn=user_message_submitted,
+    #     inputs=[user_input, chatbot],
+    #     outputs=[user_input, chatbot]
+    # ).then(
+    #     fn=update_chat_with_response,
+    #     inputs=[chatbot, user_input, model_name, temperature, max_tokens],
+    #     outputs=chatbot
+    # )
+    
+    # submit_button.click(
+    #     fn=user_message_submitted,
+    #     inputs=[user_input, chatbot],
+    #     outputs=[user_input, chatbot]
+    # ).then(
+    #     fn=update_chat_with_response,
+    #     inputs=[chatbot, user_input, model_name, temperature, max_tokens],
+    #     outputs=chatbot
+    # )
+    
+    # # Clear chat
+    # clear_button.click(
+    #     fn=lambda: [],
+    #     outputs=chatbot
+    # )
+    
     examples = [
         ["Últimamente tengo la tensión un poco alta, ¿cómo debo adaptar mis hábitos?"],
         ["What are the common side effects of metformin?"],