Update app.py

app.py

@@ -15,7 +15,7 @@ def load_data(file):
         elif file.name.endswith(".json"):
             return pd.read_json(file.name)
     except Exception as e:
-        print("Load error:", e)
+        print("File load error:", e)
         return pd.DataFrame()
 
 def load_url(url):
@@ -37,30 +37,39 @@ def process_inputs(file, url):
     df = load_url(url) if url else load_data(file)
     if df.empty:
         return pd.DataFrame(), gr.update(choices=[], value=None), gr.update(choices=[], value=None)
+    
     df_global = df
-
     all_cols = list(df.columns)
     num_cols = list(df.select_dtypes(include="number").columns)
 
+    print("Columns in data:", all_cols)
+    print("Numeric columns:", num_cols)
+
     default_x = all_cols[0] if all_cols else None
-    default_y = num_cols[0] if num_cols else None
+    default_y = num_cols[0] if num_cols else (all_cols[0] if all_cols else None)
 
     return (
         df.head(),
         gr.update(choices=all_cols, value=default_x),
-        gr.update(choices=num_cols, value=default_y)
+        gr.update(choices=num_cols if num_cols else all_cols, value=default_y)
     )
 
 def smart_update(chart_type, x, y):
     global df_global
+    print(f"Selected chart_type: {chart_type}, X: {x}, Y: {y}")
+
     if df_global.empty:
-        return None, gr.Textbox.update(value="No data loaded", visible=True)
+        return None, gr.Textbox.update(value="No data loaded.", visible=True)
 
     try:
         if chart_type == "Summary":
             summary = df_global.describe(include="all").to_string()
             return None, gr.Textbox.update(value=summary, visible=True)
-        elif chart_type == "Bar Chart":
+
+        if not x or not y:
+            return None, gr.Textbox.update(value="Please select both X and Y axes.", visible=True)
+
+        if chart_type == "Bar Chart":
             return px.bar(df_global, x=x, y=y), gr.Textbox.update(visible=False)
         elif chart_type == "Line Chart":
             return px.line(df_global, x=x, y=y), gr.Textbox.update(visible=False)
@@ -70,12 +79,15 @@ def smart_update(chart_type, x, y):
             return px.pie(df_global, names=x, values=y), gr.Textbox.update(visible=False)
         elif chart_type == "Box Plot":
             return px.box(df_global, x=x, y=y), gr.Textbox.update(visible=False)
+        else:
+            return None, gr.Textbox.update(value="Unsupported chart type.", visible=True)
+
     except Exception as e:
-        print("Plot error:", e)
+        print("Plotting error:", e)
         return None, gr.Textbox.update(value=str(e), visible=True)
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 📊 Data Visualizer with Summary")
+    gr.Markdown("## 📊 Data Visualizer with Chart & Summary")
 
     with gr.Row():
         file_input = gr.File(file_types=[".csv", ".xlsx", ".json"], label="Upload File")
@@ -84,17 +96,20 @@ with gr.Blocks() as demo:
     df_preview = gr.Dataframe(label="Preview", interactive=False)
     x_dropdown = gr.Dropdown(label="X-axis")
     y_dropdown = gr.Dropdown(label="Y-axis")
-    chart_type = gr.Dropdown(["Summary", "Bar Chart", "Line Chart", "Scatter Plot", "Pie Chart", "Box Plot"], label="Chart Type")
-
+    chart_type = gr.Dropdown(
+        ["Summary", "Bar Chart", "Line Chart", "Scatter Plot", "Pie Chart", "Box Plot"],
+        label="Chart Type"
+    )
     generate_button = gr.Button("Generate")
 
     plot_area = gr.Plot(label="Chart", visible=True)
     summary_box = gr.Textbox(label="Summary", visible=False, lines=15)
 
-    # Auto-load on file or URL change
+    # Automatically load and populate dropdowns on upload or URL change
     file_input.change(process_inputs, inputs=[file_input, url_input], outputs=[df_preview, x_dropdown, y_dropdown])
     url_input.change(process_inputs, inputs=[file_input, url_input], outputs=[df_preview, x_dropdown, y_dropdown])
 
+    # Generate plot or summary
     generate_button.click(
         smart_update,
         inputs=[chart_type, x_dropdown, y_dropdown],