Update app.py

app.py

@@ -25,13 +25,17 @@ def query_database(question, dashboard_mode=False, chart_type=None):
         print(f"Chart type: {chart_type}")
         print("="*50)
         
-        # Prepare the request - FIXED: Use correct parameter names
+        # Prepare the request
         payload = {
             "question": question,
-            "visualize": True  # Always request visualization
+            "visualize": True
         }
         
-        # Add chart_type if specified - FIXED: Use correct parameter name
+        # Add dashboard_mode if specified
+        if dashboard_mode:
+            payload["dashboard_mode"] = dashboard_mode
+        
+        # Add chart_type if specified
         if chart_type and chart_type != "auto":
             payload["viz_type"] = chart_type
         
@@ -50,7 +54,7 @@ def query_database(question, dashboard_mode=False, chart_type=None):
         print(f"\n=== RESPONSE RECEIVED ===")
         print(f"Response time: {elapsed_time:.2f} seconds")
         print(f"Response status: {response.status_code}")
-        print(f"Response text: {response.text}")
+        print(f"Response headers: {dict(response.headers)}")
         
         # Check if response is empty
         if not response.text:
@@ -65,13 +69,20 @@ def query_database(question, dashboard_mode=False, chart_type=None):
         except json.JSONDecodeError as e:
             error_msg = f"Invalid JSON response: {str(e)}"
             print(f"ERROR: {error_msg}")
-            print(f"Response text (first 500 chars): {response.text[:500]}")
+            print(f"Response text (first 1000 chars): {response.text[:1000]}")
             return "", None, None, f"❌ {error_msg}", f"Request time: {elapsed_time:.2f}s", "JSON parse error", "", "", ""
         
         # Check HTTP status
         if response.status_code != 200:
             error_msg = result.get("error", f"HTTP {response.status_code}")
             print(f"ERROR: HTTP status {response.status_code}: {error_msg}")
+            
+            # For 500 errors, show more details
+            if response.status_code == 500:
+                error_details = f"Server Error (500): {error_msg}\n"
+                error_details += f"Response: {json.dumps(result, indent=2)}"
+                return "", None, None, f"❌ Server error: {error_msg}", f"Request time: {elapsed_time:.2f}s", error_details, "", "", ""
+            
             return "", None, None, f"❌ Server error: {error_msg}", f"Request time: {elapsed_time:.2f}s", "HTTP error", "", "", ""
         
         # Extract data
@@ -89,7 +100,7 @@ def query_database(question, dashboard_mode=False, chart_type=None):
         df = pd.DataFrame(rows) if rows else pd.DataFrame()
         print(f"DataFrame shape: {df.shape}")
         
-        # Process visualization - FIXED: Handle base64 prefix
+        # Process visualization
         chart_image = None
         chart_title = ""
         chart_type_result = ""
@@ -103,15 +114,19 @@ def query_database(question, dashboard_mode=False, chart_type=None):
                         chart_error = visualization["error"]
                         print(f"Chart error: {chart_error}")
                     else:
-                        # Extract the base64 image string from the structured visualization data
+                        # Extract the base64 image string
                         chart_image_b64 = visualization.get("image")
                         if chart_image_b64:
-                            # FIXED: Remove base64 prefix if present
+                            # Handle base64 prefix
                             if chart_image_b64.startswith("data:image/"):
                                 chart_image_b64 = chart_image_b64.split(",")[1]
-                            image_bytes = base64.b64decode(chart_image_b64)
-                            chart_image = Image.open(BytesIO(image_bytes))
-                            print("Chart decoded successfully")
+                            try:
+                                image_bytes = base64.b64decode(chart_image_b64)
+                                chart_image = Image.open(BytesIO(image_bytes))
+                                print("Chart decoded successfully")
+                            except Exception as e:
+                                print(f"Error decoding chart: {e}")
+                                chart_error = f"Chart decoding error: {str(e)}"
                         
                         # Extract chart metadata
                         chart_title = visualization.get("title", "")
@@ -121,7 +136,7 @@ def query_database(question, dashboard_mode=False, chart_type=None):
                 elif isinstance(visualization, str):
                     # Fallback for backward compatibility
                     try:
-                        # FIXED: Remove base64 prefix if present
+                        # Handle base64 prefix
                         if visualization.startswith("data:image/"):
                             visualization = visualization.split(",")[1]
                         image_bytes = base64.b64decode(visualization)
@@ -129,6 +144,7 @@ def query_database(question, dashboard_mode=False, chart_type=None):
                         print("Chart decoded successfully (fallback)")
                     except Exception as e:
                         print(f"Error decoding chart (fallback): {e}")
+                        chart_error = f"Chart decoding error: {str(e)}"
             except Exception as e:
                 print(f"Error processing visualization: {e}")
                 chart_error = f"Visualization processing error: {str(e)}"
@@ -175,26 +191,38 @@ def check_health():
         print("Checking API health...")
         response = requests.get(f"{FLASK_API_URL}/health", timeout=10)
         
+        print(f"Health check response status: {response.status_code}")
+        print(f"Health check response: {response.text}")
+        
         if response.status_code == 200:
             health_data = response.json()
             status = health_data.get('status', 'unknown')
-            tables = health_data.get('tables', [])  # FIXED: This is now correct
+            tables = health_data.get('tables', [])
+            model = health_data.get('model', 'unknown')
+            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"
+            return f"❌ API returned status {response.status_code}\nResponse: {response.text}", "error"
     except Exception as e:
-        return f"❌ Health check failed: {str(e)}", "error"
+        error_msg = f"Health check failed: {str(e)}"
+        print(f"HEALTH CHECK ERROR: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return f"❌ {error_msg}", "error"
 
 def get_schema():
     try:
         print("Fetching database schema...")
-        # FIXED: Use correct endpoint
         response = requests.get(f"{FLASK_API_URL}/tables", timeout=10)
         
+        print(f"Schema response status: {response.status_code}")
+        print(f"Schema response: {response.text}")
+        
         if response.status_code == 200:
             tables_data = response.json()
             tables = tables_data.get("tables", [])
@@ -209,9 +237,12 @@ def get_schema():
             
             return schema_text, "success"
         else:
-            return f"❌ API returned status {response.status_code}", "error"
+            return f"❌ API returned status {response.status_code}\nResponse: {response.text}", "error"
     except Exception as e:
-        return f"❌ Failed to fetch schema: {str(e)}", "error"
+        error_msg = f"Failed to fetch schema: {str(e)}"
+        print(f"SCHEMA FETCH ERROR: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return f"❌ {error_msg}", "error"
 
 # 🎨 Theme: Enterprise Dark Blue
 theme = gr.themes.Default(
@@ -331,7 +362,7 @@ with gr.Blocks(theme=theme, title="Enterprise SQL Assistant", css="""
                     label="Chart Type (Optional)",
                     choices=[
                         "auto", "bar", "line", "scatter", "pie", "histogram",
-                        "time_series", "correlation"  # FIXED: Added missing chart types
+                        "time_series", "correlation"
                     ],
                     value="auto",
                     info="Force a specific chart type"