Fixing-1

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):
+def batch_generate_sql(questions_text, table_headers, progress=gr.Progress()):
     """Generate SQL for multiple questions."""
     print(f"batch_generate_sql called with: {questions_text}, {table_headers}")
     
@@ -113,29 +113,32 @@ def batch_generate_sql(questions_text, table_headers):
     try:
         # Parse questions
         questions = [q.strip() for q in questions_text.split("\n") if q.strip()]
+        total_questions = len(questions)
         
         output = f"**Batch Results:**\n"
-        output += f"Total Queries: {len(questions)}\n"
+        output += f"Total Queries: {total_questions}\n\n"
         successful_count = 0
         
         for i, question in enumerate(questions):
+            progress(i / total_questions, desc=f"Processing query {i+1}/{total_questions}")
+            
             try:
                 start_time = time.time()
                 result = sql_generator.generate_sql(question, table_headers)
                 processing_time = time.time() - start_time
                 
-                output += f"\n**Query {i+1}:** {question}\n"
+                output += f"**Query {i+1}:** {question}\n"
                 output += f"```sql\n{result['sql_query']}\n```\n"
-                output += f"Model: {result['model_used']} | Time: {processing_time:.2f}s\n"
+                output += f"Model: {result['model_used']} | Time: {processing_time:.2f}s\n\n"
                 
                 if result['status'] == 'success':
                     successful_count += 1
                     
             except Exception as e:
-                output += f"\n**Query {i+1}:** {question}\n"
-                output += f"❌ Error: {str(e)}\n"
+                output += f"**Query {i+1}:** {question}\n"
+                output += f"❌ Error: {str(e)}\n\n"
         
-        output += f"\n**Successful:** {successful_count}/{len(questions)}"
+        output += f"**Summary:** {successful_count}/{total_questions} queries successful"
         return output
         
     except Exception as e:
@@ -177,15 +180,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,7 +203,7 @@ 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"
                 )
-                generate_btn = gr.Button("🚀 Generate SQL", variant="primary", size="lg")
+                generate_btn = gr.Button("Generate SQL", variant="primary", size="lg")
             
             with gr.Column(scale=1):
                 output = gr.Markdown(label="Result")
@@ -228,7 +231,7 @@ with gr.Blocks(title="Text-to-SQL RAG with CodeLlama", theme=gr.themes.Soft()) a
             health_btn = gr.Button("Check System Health", variant="secondary", size="lg")
             health_output = gr.Markdown(label="Health Status")
     
-    # Event handlers - Fixed with explicit function calls
+    # Event handlers - Fixed with proper function binding
     generate_btn.click(
         fn=generate_sql,
         inputs=[question_input, table_headers_input],
@@ -245,7 +248,7 @@ with gr.Blocks(title="Text-to-SQL RAG with CodeLlama", theme=gr.themes.Soft()) a
     
     health_btn.click(
         fn=check_system_health,
-        inputs=None,
+        inputs=[],
         outputs=health_output,
         api_name="check_health"
     )