Fixing button

app.py

@@ -57,9 +57,17 @@ except Exception as e:
 
 def generate_sql(question, table_headers):
     """Generate SQL using the RAG system directly."""
+    print(f"generate_sql called with: {question}, {table_headers}")
+    
     if sql_generator is None:
         return "❌ Error: RAG system not initialized. Check the logs for initialization errors."
     
+    if not question or not question.strip():
+        return "❌ Error: Please enter a question."
+    
+    if not table_headers or not table_headers.strip():
+        return "❌ Error: Please enter table headers."
+    
     try:
         print(f"Generating SQL for: {question}")
         print(f"Table headers: {table_headers}")
@@ -91,9 +99,17 @@ def generate_sql(question, table_headers):
 
 def batch_generate_sql(questions_text, table_headers):
     """Generate SQL for multiple questions."""
+    print(f"batch_generate_sql called with: {questions_text}, {table_headers}")
+    
     if sql_generator is None:
         return "❌ Error: RAG system not initialized. Check the logs for initialization errors."
     
+    if not questions_text or not questions_text.strip():
+        return "❌ Error: Please enter questions."
+    
+    if not table_headers or not table_headers.strip():
+        return "❌ Error: Please enter table headers."
+    
     try:
         # Parse questions
         questions = [q.strip() for q in questions_text.split("\n") if q.strip()]
@@ -127,6 +143,8 @@ def batch_generate_sql(questions_text, table_headers):
 
 def check_system_health():
     """Check the health of the RAG system."""
+    print("check_system_health called")
+    
     try:
         if sql_generator is None:
             return "❌ System Status: RAG system not initialized\n\nCheck the logs above for initialization errors."
@@ -159,15 +177,15 @@ Check the console/logs above for detailed initialization information.
 
 # Create Gradio interface
 with gr.Blocks(title="Text-to-SQL RAG with CodeLlama", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🚀 Text-to-SQL RAG with CodeLlama")
+    gr.Markdown("#Text-to-SQL RAG with CodeLlama")
     gr.Markdown("Generate SQL queries from natural language using **RAG (Retrieval-Augmented Generation)** and **CodeLlama** models.")
     gr.Markdown("**Features:** RAG-enhanced generation, CodeLlama integration, Vector-based retrieval, Advanced prompt engineering")
     
     # Add initialization status
     if sql_generator is None:
-        gr.Markdown("⚠️ **Warning:** RAG system failed to initialize. Check the logs for errors.")
+        gr.Markdown("**Warning:** RAG system failed to initialize. Check the logs for errors.")
     else:
-        gr.Markdown("✅ **Status:** RAG system initialized successfully!")
+        gr.Markdown("**Status:** RAG system initialized successfully!")
     
     with gr.Tab("Single Query"):
         with gr.Row():
@@ -200,44 +218,48 @@ with gr.Blocks(title="Text-to-SQL RAG with CodeLlama", theme=gr.themes.Soft()) a
                     placeholder="e.g., id, name, salary, department",
                     value="id, name, salary, department"
                 )
-                batch_btn = gr.Button("🚀 Generate Batch SQL", variant="primary", size="lg")
+                batch_btn = gr.Button("Generate Batch SQL", variant="primary", size="lg")
             
             with gr.Column(scale=1):
                 batch_output = gr.Markdown(label="Batch Results")
     
     with gr.Tab("System Health"):
         with gr.Row():
-            health_btn = gr.Button("🔍 Check System Health", variant="secondary", size="lg")
+            health_btn = gr.Button("Check System Health", variant="secondary", size="lg")
             health_output = gr.Markdown(label="Health Status")
     
-    # Event handlers
+    # Event handlers - Fixed with explicit function calls
     generate_btn.click(
-        generate_sql,
+        fn=generate_sql,
         inputs=[question_input, table_headers_input],
-        outputs=output
+        outputs=output,
+        api_name="generate_sql"
     )
     
     batch_btn.click(
-        batch_generate_sql,
+        fn=batch_generate_sql,
         inputs=[batch_questions, batch_headers],
-        outputs=batch_output
+        outputs=batch_output,
+        api_name="batch_generate_sql"
     )
     
     health_btn.click(
-        check_system_health,
-        outputs=health_output
+        fn=check_system_health,
+        inputs=None,
+        outputs=health_output,
+        api_name="check_health"
     )
     
     gr.Markdown("---")
     gr.Markdown("""
-    ## 🎯 How It Works
+    ## How It Works
     
     1. **RAG System**: Retrieves relevant SQL examples from vector database
     2. **CodeLlama**: Generates SQL using retrieved examples as context
     3. **Vector Search**: Finds similar questions and their SQL solutions
     4. **Enhanced Generation**: Combines retrieval + generation for better accuracy
     
-    ## 🛠️ Technology Stack
+    ## Technology Stack
     
     - **Backend**: Direct RAG system integration
     - **LLM**: CodeLlama-7B-Python-GGUF (primary)
@@ -245,7 +267,7 @@ with gr.Blocks(title="Text-to-SQL RAG with CodeLlama", theme=gr.themes.Soft()) a
     - **Frontend**: Gradio interface
     - **Hosting**: Hugging Face Spaces
     
-    ## 📊 Performance
+    ## Performance
     
     - **Model**: CodeLlama-7B-Python-GGUF
     - **Response Time**: < 5 seconds
@@ -255,4 +277,4 @@ with gr.Blocks(title="Text-to-SQL RAG with CodeLlama", theme=gr.themes.Soft()) a
 
 # Launch the interface
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()