v11

app.py

@@ -199,6 +199,7 @@ def generate_plot(data, x_col, y_col, title, x_label, y_label):
 
 async def stream_agent_response(question: str, chat_history: List) -> Tuple[List, Dict]:
     """Procesa la pregunta del usuario y devuelve la respuesta del agente."""
+
     if not agent:
         error_msg = (
             "## ⚠️ Error: Agente no inicializado\n\n"
@@ -229,8 +230,11 @@ async def stream_agent_response(question: str, chat_history: List) -> Tuple[List
             if sql_query:
                 # Ejecutar la consulta y actualizar la respuesta
                 db_connection, _ = setup_database_connection()
-                query_result = execute_sql_query(sql_query, db_connection)
-                response_text += f"\n\n### 🔍 Resultado de la consulta:\n```sql\n{sql_query}\n```\n\n{query_result}"
+                if db_connection:
+                    query_result = execute_sql_query(sql_query, db_connection)
+                    response_text += f"\n\n### 🔍 Resultado de la consulta:\n```sql\n{sql_query}\n```\n\n{query_result}"
+                else:
+                    response_text += "\n\n⚠️ No se pudo conectar a la base de datos para ejecutar la consulta."
         else:
             response_text = "Error: No se recibió respuesta del agente."
         
@@ -242,6 +246,7 @@ async def stream_agent_response(question: str, chat_history: List) -> Tuple[List
         error_msg = f"## ❌ Error\n\nOcurrió un error al procesar tu solicitud:\n\n```\n{str(e)}\n```"
         chat_history[-1][1] = error_msg
         yield chat_history, gr.update(visible=False)
+        yield chat_history, gr.update(visible=False)
 
 # Custom CSS for the app
 custom_css = """
@@ -376,11 +381,29 @@ def create_ui():
                     ```
                     """)
                 else:
-                    gr.Markdown("""
-                    ## ✅ Sistema listo
-                    
-                    El asistente está listo para responder tus preguntas sobre la base de datos.
-                    """)
+                    if os.getenv('SPACE_ID'):
+                        # Modo demo en Hugging Face Spaces
+                        gr.Markdown("""
+                        ## 🚀 Modo Demo
+                        
+                        Esta es una demostración del asistente de base de datos SQL. Para usar la versión completa con conexión a base de datos:
+                        
+                        1. Clona este espacio en tu cuenta de Hugging Face
+                        2. Configura las variables de entorno en la configuración del espacio:
+                           - `DB_USER`: Tu usuario de base de datos
+                           - `DB_PASSWORD`: Tu contraseña de base de datos
+                           - `DB_HOST`: La dirección del servidor de base de datos
+                           - `DB_NAME`: El nombre de la base de datos
+                           - `GOOGLE_API_KEY`: Tu clave de API de Google Gemini
+                        
+                        **Nota:** Actualmente estás en modo de solo demostración.
+                        """)
+                    else:
+                        gr.Markdown("""
+                        ## ✅ Sistema listo
+                        
+                        El asistente está listo para responder tus preguntas sobre la base de datos.
+                        """)
         
         # Interfaz de chat - usando el nuevo formato de mensajes
         chatbot = gr.Chatbot(