Fixing-2

app.py

@@ -97,7 +97,7 @@ def generate_sql(question, table_headers):
         print(error_msg)
         return error_msg
 
-def batch_generate_sql(questions_text, table_headers, progress=gr.Progress()):
+def batch_generate_sql(questions_text, table_headers):
     """Generate SQL for multiple questions."""
     print(f"batch_generate_sql called with: {questions_text}, {table_headers}")
     
@@ -120,7 +120,7 @@ def batch_generate_sql(questions_text, table_headers, progress=gr.Progress()):
         successful_count = 0
         
         for i, question in enumerate(questions):
-            progress(i / total_questions, desc=f"Processing query {i+1}/{total_questions}")
+            print(f"Processing query {i+1}/{total_questions}: {question}")
             
             try:
                 start_time = time.time()
@@ -180,7 +180,7 @@ 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")
     
@@ -228,29 +228,25 @@ with gr.Blocks(title="Text-to-SQL RAG with CodeLlama", theme=gr.themes.Soft()) a
     
     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 - Fixed with proper function binding
+    # Event handlers - Simple and working
     generate_btn.click(
-        fn=generate_sql,
+        generate_sql,
         inputs=[question_input, table_headers_input],
-        outputs=output,
-        api_name="generate_sql"
+        outputs=output
     )
     
     batch_btn.click(
-        fn=batch_generate_sql,
+        batch_generate_sql,
         inputs=[batch_questions, batch_headers],
-        outputs=batch_output,
-        api_name="batch_generate_sql"
+        outputs=batch_output
     )
     
     health_btn.click(
-        fn=check_system_health,
-        inputs=[],
-        outputs=health_output,
-        api_name="check_health"
+        check_system_health,
+        outputs=health_output
     )
     
     gr.Markdown("---")
@@ -270,7 +266,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