v2

README.md

@@ -1,12 +1,29 @@
 ---
-title: ChatbotAgentGraphSQLgemini Etheroji
-emoji: 🤖
+title: Chatbot SQL Agent with Gemini
+emoji: 🔍
 colorFrom: blue
-colorTo: indigo
+colorTo: purple
 sdk: gradio
-sdk_version: "4.47.0"
+sdk_version: "3.50.2"
 app_file: app.py
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Chatbot SQL Agent with Gemini
+
+A powerful chatbot that can answer questions by querying your SQL database using Google's Gemini model.
+
+## Features
+
+- Natural language to SQL query conversion
+- Interactive chat interface
+- Direct database connectivity
+- Powered by Google's Gemini AI
+
+## Setup
+
+1. Set up your environment variables in `.env` file
+2. Install dependencies: `pip install -r requirements.txt`
+3. Run the app: `python app.py`
+
+Check out the [configuration reference](https://huggingface.co/docs/hub/spaces-config-reference) for more options.

app.py

@@ -208,7 +208,9 @@ async def stream_agent_response(question: str, chat_history: List) -> Tuple[List
             f"3. El modelo de lenguaje esté disponible\n\n"
             f"Error: {agent_error}"
         )
-        return chat_history + [[question, error_msg]], gr.update(visible=False)
+        chat_history = chat_history + [[question, error_msg]]
+        yield chat_history, gr.update(visible=False)
+        return
     
     try:
         # Agregar un mensaje de "pensando"
@@ -234,11 +236,12 @@ async def stream_agent_response(question: str, chat_history: List) -> Tuple[List
         
         # Actualizar el historial con la respuesta completa
         chat_history[-1][1] = response_text
-        return chat_history, gr.update(visible=False)
+        yield chat_history, gr.update(visible=False)
         
     except Exception as e:
         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)
         return chat_history, gr.update(visible=False)
 
 # Custom CSS for the app