Update app.py

app.py

@@ -189,26 +189,51 @@ def query_database(question, dashboard_mode=False, chart_type=None):
 def check_health():
     try:
         print("Checking API health...")
+        print(f"API URL: {FLASK_API_URL}/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}")
+        print(f"Health check response headers: {dict(response.headers)}")
+        print(f"Health check response text: {response.text}")
         
         if response.status_code == 200:
-            health_data = response.json()
-            status = health_data.get('status', 'unknown')
-            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"
+            try:
+                health_data = response.json()
+                print(f"Parsed health data: {health_data}")
+                
+                status = health_data.get('status', 'unknown')
+                tables = health_data.get('tables', [])
+                model = health_data.get('model', 'unknown')
+                data_rows = health_data.get('data_rows', 0)
+                
+                # Ensure tables is a list before joining
+                if isinstance(tables, list):
+                    tables_str = ', '.join(tables) if tables else 'None'
+                else:
+                    tables_str = str(tables)
+                
+                health_msg = f"✅ API Status: {status.upper()}\n"
+                health_msg += f"🤖 Model: {model}\n"
+                health_msg += f"📊 Tables: {tables_str}\n"
+                health_msg += f"📈 Data Rows: {data_rows:,}"
+                
+                return health_msg, "success"
+            except json.JSONDecodeError as e:
+                error_msg = f"Failed to parse health check response: {str(e)}"
+                print(f"JSON PARSE ERROR: {error_msg}")
+                return f"❌ {error_msg}", "error"
         else:
             return f"❌ API returned status {response.status_code}\nResponse: {response.text}", "error"
+    except requests.exceptions.ConnectionError as e:
+        error_msg = f"Connection to API failed: {str(e)}"
+        print(f"CONNECTION ERROR: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return f"❌ {error_msg}", "error"
+    except requests.exceptions.Timeout:
+        error_msg = "Health check request timed out"
+        print(f"TIMEOUT ERROR: {error_msg}")
+        return f"❌ {error_msg}", "error"
     except Exception as e:
         error_msg = f"Health check failed: {str(e)}"
         print(f"HEALTH CHECK ERROR: {error_msg}")
@@ -218,26 +243,44 @@ def check_health():
 def get_schema():
     try:
         print("Fetching database schema...")
+        print(f"API URL: {FLASK_API_URL}/tables")
+        
         response = requests.get(f"{FLASK_API_URL}/tables", timeout=10)
         
         print(f"Schema response status: {response.status_code}")
-        print(f"Schema response: {response.text}")
+        print(f"Schema response text: {response.text}")
         
         if response.status_code == 200:
-            tables_data = response.json()
-            tables = tables_data.get("tables", [])
-            
-            schema_text = "## Database Schema\n\n"
-            for table in tables:
-                schema_text += f"### {table['name']}\n"
-                schema_text += "| Column |\n|--------|\n"
-                for col in table['columns']:
-                    schema_text += f"| {col} |\n"
-                schema_text += "\n"
-            
-            return schema_text, "success"
+            try:
+                tables_data = response.json()
+                print(f"Parsed tables data: {tables_data}")
+                
+                tables = tables_data.get("tables", [])
+                
+                schema_text = "## Database Schema\n\n"
+                for table in tables:
+                    schema_text += f"### {table.get('name', 'Unknown')}\n"
+                    schema_text += "| Column |\n|--------|\n"
+                    for col in table.get('columns', []):
+                        schema_text += f"| {col} |\n"
+                    schema_text += "\n"
+                
+                return schema_text, "success"
+            except json.JSONDecodeError as e:
+                error_msg = f"Failed to parse schema response: {str(e)}"
+                print(f"JSON PARSE ERROR: {error_msg}")
+                return f"❌ {error_msg}", "error"
         else:
             return f"❌ API returned status {response.status_code}\nResponse: {response.text}", "error"
+    except requests.exceptions.ConnectionError as e:
+        error_msg = f"Connection to API failed: {str(e)}"
+        print(f"CONNECTION ERROR: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return f"❌ {error_msg}", "error"
+    except requests.exceptions.Timeout:
+        error_msg = "Schema request timed out"
+        print(f"TIMEOUT ERROR: {error_msg}")
+        return f"❌ {error_msg}", "error"
     except Exception as e:
         error_msg = f"Failed to fetch schema: {str(e)}"
         print(f"SCHEMA FETCH ERROR: {error_msg}")