Update app.py

app.py

@@ -7,7 +7,6 @@ import json
 import time
 import traceback
 from PIL import Image
-
 # 🔧 CONFIGURE: Your Flask API URL (Public URL)
 FLASK_API_URL = "http://52.15.175.95:5000"
 
@@ -95,6 +94,7 @@ def query_database(question, dashboard_mode=False, chart_type=None):
         print(f"SQL: {sql[:100] if sql else 'None'}...")
         print(f"Rows count: {row_count}")
         print(f"Visualization received: {'Yes' if visualization else 'No'}")
+        print(f"Visualization details: {visualization}")
         
         # Create DataFrame
         df = pd.DataFrame(rows) if rows else pd.DataFrame()
@@ -102,9 +102,11 @@ def query_database(question, dashboard_mode=False, chart_type=None):
         
         # Process visualization
         chart_image = None
+        chart_html = None
         chart_title = ""
         chart_type_result = ""
         chart_error = None
+        chart_format = ""
         
         if visualization:
             try:
@@ -124,15 +126,24 @@ def query_database(question, dashboard_mode=False, chart_type=None):
                                 image_bytes = base64.b64decode(chart_image_b64)
                                 chart_image = Image.open(BytesIO(image_bytes))
                                 print("Chart decoded successfully")
+                                chart_format = "png"
                             except Exception as e:
                                 print(f"Error decoding chart: {e}")
                                 chart_error = f"Chart decoding error: {str(e)}"
                         
+                        # If no image, check for HTML fallback
+                        if not chart_image and "html" in visualization:
+                            chart_html = visualization["html"]
+                            print("HTML chart fallback available")
+                            chart_format = "html"
+                        
                         # Extract chart metadata
                         chart_title = visualization.get("title", "")
                         chart_type_result = visualization.get("type", "")
+                        chart_format = visualization.get("format", chart_format)
                         print(f"Chart type: {chart_type_result}")
                         print(f"Chart title: {chart_title}")
+                        print(f"Chart format: {chart_format}")
                 elif isinstance(visualization, str):
                     # Fallback for backward compatibility
                     try:
@@ -142,6 +153,7 @@ def query_database(question, dashboard_mode=False, chart_type=None):
                         image_bytes = base64.b64decode(visualization)
                         chart_image = Image.open(BytesIO(image_bytes))
                         print("Chart decoded successfully (fallback)")
+                        chart_format = "png"
                     except Exception as e:
                         print(f"Error decoding chart (fallback): {e}")
                         chart_error = f"Chart decoding error: {str(e)}"
@@ -158,33 +170,35 @@ def query_database(question, dashboard_mode=False, chart_type=None):
             details += f"Chart type: {chart_type_result}\n"
         if chart_title:
             details += f"Chart title: {chart_title}\n"
+        if chart_format:
+            details += f"Chart format: {chart_format}\n"
         if dashboard_mode:
             details += f"Dashboard mode: Enabled\n"
         details += f"Response size: {len(response.text)} bytes"
         
         print(f"=== REQUEST COMPLETED SUCCESSFULLY ===")
         
-        return sql, df, chart_image, f"✅ Query completed successfully", details, "Success", chart_type_result, chart_title, chart_error
+        return sql, df, chart_image, chart_html, f"✅ Query completed successfully", details, "Success", chart_type_result, chart_title, chart_error
     
     except requests.exceptions.ConnectionError as e:
         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", "", "", ""
+        return "", None, None, None, f"❌ {error_msg}", "Connection error", "Connection error", "", "", ""
     except requests.exceptions.Timeout:
         error_msg = "Request timed out after 300 seconds"
         print(f"TIMEOUT ERROR: {error_msg}")
-        return "", None, None, f"⏱️ {error_msg}", "Timeout error", "Timeout error", "", "", ""
+        return "", None, 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", "", "", ""
+        return "", None, None, None, f"❌ {error_msg}", "Request error", "Request error", "", "", ""
     except Exception as 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", "", "", ""
+        return "", None, None, None, f"🚨 {error_msg}", f"Error: {str(e)}", "Unexpected error", "", "", ""
 
 def check_health():
     try:
@@ -462,10 +476,16 @@ with gr.Blocks(theme=theme, title="Enterprise SQL Assistant", css="""
                                 height=400,
                                 elem_classes="chart-container"
                             )
+                            # Add HTML component for fallback
+                            html_output = gr.HTML(
+                                label="Interactive Chart",
+                                visible=False
+                            )
                         with gr.Column(scale=1):
                             gr.Markdown("### 📊 Chart Information")
                             chart_type_output = gr.Markdown(label="Chart Type", elem_classes="chart-metadata")
                             chart_title_output = gr.Markdown(label="Chart Title", elem_classes="chart-metadata")
+                            chart_format_output = gr.Markdown(label="Chart Format", elem_classes="chart-metadata")
                             chart_error_output = gr.Markdown(label="Chart Error", elem_classes="chart-metadata")
             
             gr.Markdown("### 📊 Request Details")
@@ -505,7 +525,7 @@ with gr.Blocks(theme=theme, title="Enterprise SQL Assistant", css="""
     submit_btn.click(
         fn=query_database,
         inputs=[question_input, dashboard_checkbox, chart_type_dropdown],
-        outputs=[sql_output, results_output, chart_output, chatbot_output, request_details, error_details, chart_type_output, chart_title_output, chart_error_output]
+        outputs=[sql_output, results_output, chart_output, html_output, chatbot_output, request_details, error_details, chart_type_output, chart_title_output, chart_format_output, chart_error_output]
     )
 
 # Launch