v8

app.py

@@ -382,10 +382,11 @@ def create_ui():
                     El asistente está listo para responder tus preguntas sobre la base de datos.
                     """)
         
-        # Interfaz de chat - versión simplificada para compatibilidad
+        # Interfaz de chat - usando el nuevo formato de mensajes
         chatbot = gr.Chatbot(
             label="Chat",
-            height=500
+            height=500,
+            type="messages"  # Usando el nuevo formato de mensajes
         )
         
         # Área de entrada
@@ -437,48 +438,56 @@ def create_ui():
         
         return demo, chatbot, question_input, submit_button, streaming_output_display
 
-# Create the UI components
-demo, chatbot, question_input, submit_button, streaming_output_display = create_ui()
-
-def user_message(user_input: str, chat_history: List) -> Tuple[str, List]:
-    """Add user message to chat history and clear input."""
-    if not user_input.strip():
-        return "", chat_history
-    logger.info(f"User message: {user_input}")
-    return "", chat_history + [[user_input, None]]
-
-def bot_response(chat_history: List) -> Tuple[List, Dict]:
-    """Get bot response and update chat history."""
-    if not chat_history or not chat_history[-1][0]:
-        return chat_history, gr.update(visible=False)
+def create_application():
+    """Create and configure the Gradio application."""
+    # Create the UI components
+    demo, chatbot, question_input, submit_button, streaming_output_display = create_ui()
     
-    question = chat_history[-1][0]
-    logger.info(f"Processing question: {question}")
-    return stream_agent_response(question, chat_history[:-1])
-
-# Event handlers
-submit_click = submit_button.click(
-    fn=user_message,
-    inputs=[question_input, chatbot],
-    outputs=[question_input, chatbot],
-    queue=True
-).then(
-    fn=bot_response,
-    inputs=[chatbot],
-    outputs=[chatbot, streaming_output_display],
-    api_name="ask"
-)
+    def user_message(user_input: str, chat_history: List) -> Tuple[str, List]:
+        """Add user message to chat history and clear input."""
+        if not user_input.strip():
+            return "", chat_history
+        logger.info(f"User message: {user_input}")
+        return "", chat_history + [[user_input, None]]
+    
+    def bot_response(chat_history: List) -> Tuple[List, Dict]:
+        """Get bot response and update chat history."""
+        if not chat_history or not chat_history[-1][0]:
+            return chat_history, gr.update(visible=False)
+        
+        question = chat_history[-1][0]
+        logger.info(f"Processing question: {question}")
+        return stream_agent_response(question, chat_history[:-1])
+    
+    # Event handlers
+    with demo:
+        submit_click = submit_button.click(
+            fn=user_message,
+            inputs=[question_input, chatbot],
+            outputs=[question_input, chatbot],
+            queue=True
+        ).then(
+            fn=bot_response,
+            inputs=[chatbot],
+            outputs=[chatbot, streaming_output_display],
+            api_name="ask"
+        )
+        
+        question_input.submit(
+            fn=user_message,
+            inputs=[question_input, chatbot],
+            outputs=[question_input, chatbot],
+            queue=True
+        ).then(
+            fn=bot_response,
+            inputs=[chatbot],
+            outputs=[chatbot, streaming_output_display]
+        )
+    
+    return demo
 
-question_input.submit(
-    fn=user_message,
-    inputs=[question_input, chatbot],
-    outputs=[question_input, chatbot],
-    queue=True
-).then(
-    fn=bot_response,
-    inputs=[chatbot],
-    outputs=[chatbot, streaming_output_display]
-)
+# Create the application
+demo = create_application()
 
 # Configuración para Hugging Face Spaces
 def get_app():