Update app.py

app.py

@@ -5,81 +5,157 @@ import base64
 from io import BytesIO
 import json
 import time
+import traceback
 
 # 🔧 CONFIGURE: Replace with your EC2 public IP
-API_URL = "http://18.191.172.85:5000/ask"
-HEALTH_URL = "http://18.191.172.85:5000/health"
+API_URL = "http://3.14.153.135:5000/ask"
+HEALTH_URL = "http://3.14.153.135:5000/health"
+SIMPLE_TEST_URL = "http://3.14.153.135:5000/simple-test"
+DEBUG_URL = "http://3.14.153.135:5000/debug"
 
 def test_connection():
     try:
-        response = requests.get(HEALTH_URL, timeout=30)  # Increased to 30 seconds
+        print("=== TESTING CONNECTION ===")
+        response = requests.get(HEALTH_URL, timeout=30)
+        print(f"Connection test status: {response.status_code}")
+        print(f"Connection test response: {response.text}")
+        
         if response.status_code == 200:
             return f"✅ Connection successful! Status: {response.status_code}\nResponse: {response.text}"
         else:
             return f"⚠️ Connection returned status: {response.status_code}\nResponse: {response.text}"
-    except requests.exceptions.ConnectionError:
-        return "❌ Connection failed. The server might be down or not accessible from Hugging Face."
-    except requests.exceptions.Timeout:
-        return "⏱️ Connection timed out after 30 seconds."
     except Exception as e:
-        return f"🚨 Unexpected error: {str(e)}"
+        error_msg = f"Connection error: {str(e)}"
+        print(f"Connection error: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return f"❌ {error_msg}"
 
-def test_chart_generation():
+def test_simple_endpoint():
     try:
-        response = requests.get("http://18.191.172.85:5000/test-chart", timeout=60)  # Increased to 60 seconds
+        print("=== TESTING SIMPLE ENDPOINT ===")
+        response = requests.post(
+            SIMPLE_TEST_URL,
+            json={"question": "test question"},
+            timeout=30
+        )
+        
+        print(f"Simple test status: {response.status_code}")
+        print(f"Simple test response: {response.text}")
+        
         if response.status_code == 200:
             result = response.json()
-            if 'chart' in result:
-                image_bytes = base64.b64decode(result['chart'])
-                chart_image = BytesIO(image_bytes)
-                return "✅ Test chart generated successfully!", chart_image
-            else:
-                return f"❌ Error: {result.get('message', 'Unknown error')}", None
+            return f"✅ Simple test successful! SQL: {result.get('sql', 'N/A')}", None
         else:
-            return f"❌ HTTP Error: {response.status_code}", None
+            return f"❌ Simple test failed: {response.status_code} - {response.text}", None
+    except Exception as e:
+        error_msg = f"Simple test error: {str(e)}"
+        print(f"Simple test error: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return f"❌ {error_msg}", None
+
+def test_debug_endpoint():
+    try:
+        print("=== TESTING DEBUG ENDPOINT ===")
+        response = requests.post(
+            DEBUG_URL,
+            json={"test": "data"},
+            timeout=30
+        )
+        
+        print(f"Debug test status: {response.status_code}")
+        print(f"Debug test response: {response.text}")
+        
+        return f"Debug response: {response.text}", None
     except Exception as e:
-        return f"❌ Connection error: {str(e)}", None
+        error_msg = f"Debug test error: {str(e)}"
+        print(f"Debug test error: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return f"❌ {error_msg}", None
 
 def query_database(question):
     if not question.strip():
-        return "", None, None, "⚠️ Please enter a question.", ""
+        return "", None, None, "⚠️ Please enter a question.", "", ""
     
     try:
         start_time = time.time()
-        print(f"Sending request to: {API_URL}")
+        print("\n" + "="*50)
+        print("=== NEW QUERY REQUEST ===")
+        print(f"Time: {time.strftime('%Y-%m-%d %H:%M:%S')}")
         print(f"Question: {question}")
+        print("="*50)
+        
+        # Prepare the request
+        headers = {
+            'Content-Type': 'application/json',
+            'User-Agent': 'HuggingFace-Space/1.0'
+        }
+        payload = {"question": question}
         
+        print(f"Request URL: {API_URL}")
+        print(f"Request headers: {headers}")
+        print(f"Request payload: {payload}")
+        
+        # Make the request
+        print("Sending request...")
         response = requests.post(
             API_URL, 
-            json={"question": question}, 
-            timeout=300  # Increased timeout to 300 seconds (5 minutes)
+            json=payload, 
+            headers=headers,
+            timeout=300
         )
         
         elapsed_time = time.time() - start_time
-        print(f"Response status: {response.status_code}")
+        print(f"\n=== RESPONSE RECEIVED ===")
         print(f"Response time: {elapsed_time:.2f} seconds")
+        print(f"Response status: {response.status_code}")
+        print(f"Response headers: {dict(response.headers)}")
+        print(f"Response text length: {len(response.text)} characters")
+        print(f"Response text: {response.text}")
+        
+        # Check if response is empty
+        if not response.text:
+            error_msg = "Empty response from server"
+            print(f"ERROR: {error_msg}")
+            return "", None, None, f"❌ {error_msg}", f"Request time: {elapsed_time:.2f}s", "Empty response"
         
+        # Try to parse JSON response
         try:
             result = response.json()
-        except json.JSONDecodeError:
-            return "", None, None, f"❌ Invalid response from server: {response.text}", f"Request time: {elapsed_time:.2f}s"
+            print(f"Parsed JSON successfully: {type(result)}")
+        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]}")
+            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}")
-            return "", None, None, f"❌ Server error: {error_msg}", f"Request time: {elapsed_time:.2f}s"
+            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 "error" in result:
-            return "", None, None, f"❌ {result['error']}", f"Request time: {elapsed_time:.2f}s"
+            error_msg = result['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
         sql = result.get("sql", "N/A")
         rows = result.get("result", [])
         chart_data = result.get("chart")
         
+        print(f"\n=== EXTRACTED DATA ===")
+        print(f"SQL: {sql[:100] if sql else 'None'}...")
+        print(f"Rows count: {len(rows)}")
         print(f"Chart data received: {'Yes' if chart_data else 'No'}")
         print(f"Chart data length: {len(chart_data) if chart_data else 0}")
         
+        # Create DataFrame
         df = pd.DataFrame(rows) if rows else pd.DataFrame()
+        print(f"DataFrame shape: {df.shape}")
         
+        # Process chart
         chart_image = None
         if chart_data:
             try:
@@ -89,18 +165,36 @@ def query_database(question):
             except Exception as e:
                 print(f"Error decoding chart: {e}")
         
+        # Prepare details
         details = f"Request time: {elapsed_time:.2f}s\n"
+        details += f"Status code: {response.status_code}\n"
         details += f"Rows returned: {len(rows)}\n"
-        details += f"Chart generated: {'Yes' if chart_data else 'No'}"
+        details += f"Chart generated: {'Yes' if chart_data else 'No'}\n"
+        details += f"Response size: {len(response.text)} bytes"
+        
+        print(f"=== REQUEST COMPLETED SUCCESSFULLY ===")
         
-        return sql, df, chart_image, "✅ Query completed successfully!", details
+        return sql, df, chart_image, "✅ Query completed successfully!", details, "Success"
     
     except requests.exceptions.ConnectionError as e:
-        return "", None, None, f"❌ Connection failed: {str(e)}", "Connection error"
+        error_msg = f"Connection failed: {str(e)}"
+        print(f"CONNECTION ERROR: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return "", None, None, f"❌ {error_msg}", "Connection error", "Connection error"
     except requests.exceptions.Timeout:
-        return "", None, None, "⏱️ Request timed out after 300 seconds. Try a simpler question.", "Timeout error"
+        error_msg = "Request timed out after 300 seconds"
+        print(f"TIMEOUT ERROR: {error_msg}")
+        return "", None, None, f"⏱️ {error_msg}", "Timeout error", "Timeout error"
+    except requests.exceptions.RequestException as e:
+        error_msg = f"Request exception: {str(e)}"
+        print(f"REQUEST ERROR: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return "", None, None, f"❌ {error_msg}", "Request error", "Request error"
     except Exception as e:
-        return "", None, None, f"🚨 Unexpected error: {str(e)}", f"Error: {str(e)}"
+        error_msg = f"Unexpected error: {str(e)}"
+        print(f"UNEXPECTED ERROR: {error_msg}")
+        print(f"Traceback: {traceback.format_exc()}")
+        return "", None, None, f"🚨 {error_msg}", f"Error: {str(e)}", "Unexpected error"
 
 # 🎨 Theme: Enterprise Dark Blue
 theme = gr.themes.Default(
@@ -134,14 +228,16 @@ with gr.Blocks(theme=theme, title="Enterprise SQL Assistant") as demo:
             )
             submit_btn = gr.Button("🚀 Generate SQL & Results", variant="primary")
             
-            gr.Markdown("### 🔧 Connection Test")
-            test_btn = gr.Button("Test Connection to EC2")
-            connection_status = gr.Textbox(label="Connection Status", interactive=False)
-            
-            gr.Markdown("### 🎨 Chart Generation Test")
-            test_chart_btn = gr.Button("Test Chart Generation")
-            chart_test_status = gr.Textbox(label="Chart Test Status", interactive=False)
-            test_chart_output = gr.Image(label="Test Chart", type="pil")
+            gr.Markdown("### 🔧 Connection Tests")
+            with gr.Accordion("Connection Tests", open=True):
+                test_btn = gr.Button("Test Connection to EC2")
+                connection_status = gr.Textbox(label="Connection Status", interactive=False)
+                
+                simple_test_btn = gr.Button("Test Simple Endpoint")
+                simple_test_status = gr.Textbox(label="Simple Test Status", interactive=False)
+                
+                debug_test_btn = gr.Button("Test Debug Endpoint")
+                debug_test_status = gr.Textbox(label="Debug Test Status", interactive=False)
             
             gr.Markdown("### 💡 Example Queries")
             gr.Examples(
@@ -183,7 +279,10 @@ with gr.Blocks(theme=theme, title="Enterprise SQL Assistant") as demo:
                     )
             
             gr.Markdown("### 📊 Request Details")
-            request_details = gr.Textbox(label="Request Details", interactive=False, lines=4)
+            request_details = gr.Textbox(label="Request Details", interactive=False, lines=6)
+            
+            gr.Markdown("### 🔍 Error Details")
+            error_details = gr.Textbox(label="Error Details", interactive=False, lines=4)
     
     # Events
     test_btn.click(
@@ -191,15 +290,20 @@ with gr.Blocks(theme=theme, title="Enterprise SQL Assistant") as demo:
         outputs=[connection_status]
     )
     
-    test_chart_btn.click(
-        fn=test_chart_generation,
-        outputs=[chart_test_status, test_chart_output]
+    simple_test_btn.click(
+        fn=test_simple_endpoint,
+        outputs=[simple_test_status]
+    )
+    
+    debug_test_btn.click(
+        fn=test_debug_endpoint,
+        outputs=[debug_test_status]
     )
     
     submit_btn.click(
         fn=query_database,
         inputs=question_input,
-        outputs=[sql_output, results_output, chart_output, status, request_details]
+        outputs=[sql_output, results_output, chart_output, status, request_details, error_details]
     )
 
 # Launch