Update app.py

app.py

@@ -8,8 +8,8 @@ import time
 import traceback
 from PIL import Image
 
-# 🔧 CONFIGURE: Replace with your Django server URL
-DJANGO_API_URL = "http://your-django-server.com/api/openai-prompt/"
+# 🔧 CONFIGURE: Your Flask API URL
+FLASK_API_URL = "http://3.14.153.135:5000"
 
 def query_database(question):
     if not question.strip():
@@ -24,15 +24,15 @@ def query_database(question):
         print("="*50)
         
         # Prepare the request
-        payload = {"prompt": question}
+        payload = {"question": question}
         
-        print(f"Request URL: {DJANGO_API_URL}")
+        print(f"Request URL: {FLASK_API_URL}/ask")
         print(f"Request payload: {payload}")
         
         # Make the request
         print("Sending request...")
         response = requests.post(
-            DJANGO_API_URL, 
+            f"{FLASK_API_URL}/ask", 
             json=payload,
             timeout=300
         )
@@ -61,18 +61,11 @@ def query_database(question):
         
         # Check HTTP status
         if response.status_code != 200:
-            error_msg = result.get("message", f"HTTP {response.status_code}")
+            error_msg = result.get("error", f"HTTP {response.status_code}")
             print(f"ERROR: HTTP status {response.status_code}: {error_msg}")
             return "", None, None, f"❌ Server error: {error_msg}", f"Request time: {elapsed_time:.2f}s", "HTTP error"
         
-        # Check for application error
-        if "status" in result and result["status"] == "error":
-            error_msg = result.get("message", "Unknown error")
-            print(f"ERROR: Application error: {error_msg}")
-            return "", None, None, f"❌ {error_msg}", f"Request time: {elapsed_time:.2f}s", "Application error"
-        
         # Extract data
-        conversational_response = result.get("response", "")
         sql = result.get("sql", "")
         rows = result.get("result", [])
         chart_data = result.get("chart")
@@ -106,7 +99,7 @@ def query_database(question):
         
         print(f"=== REQUEST COMPLETED SUCCESSFULLY ===")
         
-        return sql, df, chart_image, conversational_response, details, "Success"
+        return sql, df, chart_image, f"✅ Query completed successfully", details, "Success"
     
     except requests.exceptions.ConnectionError as e:
         error_msg = f"Connection failed: {str(e)}"
@@ -128,6 +121,29 @@ def query_database(question):
         print(f"Traceback: {traceback.format_exc()}")
         return "", None, None, f"🚨 {error_msg}", f"Error: {str(e)}", "Unexpected error"
 
+def check_health():
+    try:
+        print("Checking API health...")
+        response = requests.get(f"{FLASK_API_URL}/health", timeout=10)
+        
+        if response.status_code == 200:
+            health_data = response.json()
+            status = health_data.get('status', 'unknown')
+            model = health_data.get('model', 'unknown')
+            tables = health_data.get('tables', [])
+            data_rows = health_data.get('data_rows', 0)
+            
+            health_msg = f"✅ API Status: {status.upper()}\n"
+            health_msg += f"🤖 Model: {model}\n"
+            health_msg += f"📊 Tables: {', '.join(tables) if tables else 'None'}\n"
+            health_msg += f"📈 Data Rows: {data_rows:,}"
+            
+            return health_msg, "success"
+        else:
+            return f"❌ API returned status {response.status_code}", "error"
+    except Exception as e:
+        return f"❌ Health check failed: {str(e)}", "error"
+
 # 🎨 Theme: Enterprise Dark Blue
 theme = gr.themes.Default(
     primary_hue="blue",
@@ -233,6 +249,10 @@ with gr.Blocks(theme=theme, title="Enterprise SQL Assistant", css="""
                     example6 = gr.Button("Count of members by status", elem_classes=["example-btn"])
                     example7 = gr.Button("Show me the first 10 rows", elem_classes=["example-btn"])
                     example8 = gr.Button("What is the average age of members?", elem_classes=["example-btn"])
+            
+            gr.Markdown("### 🩺 API Health Check")
+            health_btn = gr.Button("Check API Health", variant="secondary")
+            health_output = gr.Markdown(label="API Status")
         
         with gr.Column(scale=2):
             gr.Markdown("### 🤖 AI Assistant Response")
@@ -271,6 +291,13 @@ with gr.Blocks(theme=theme, title="Enterprise SQL Assistant", css="""
         return example_text
     
     # Events
+    # Health check button
+    health_btn.click(
+        fn=check_health,
+        inputs=[],
+        outputs=[health_output, error_details]
+    )
+    
     # Example query buttons
     for example_btn in [example1, example2, example3, example4, example5, example6, example7, example8]:
         example_btn.click(