Error Fix

app.py

@@ -627,8 +627,8 @@ def generate_visualization(result_df, query):
         print("Visualization requested, attempting to create plot...")
         
         # Set common figure parameters
-        fig_width = 1000
-        fig_height = 700
+        fig_width = 1200
+        fig_height = 650
         
         # Determine visualization type from query
         viz_type = 'bar'  # Default
@@ -695,7 +695,14 @@ def generate_visualization(result_df, query):
             
             if y_cols:
                 fig = px.line(result_df, x=x_col, y=y_cols[0])
-                fig.update_layout(title_text=f'Line Chart of {y_cols[0]} over {x_col}')
+                fig.update_layout(
+                    title_text=f'Line Chart of {y_cols[0]} over {x_col}',
+                    xaxis=dict(
+                        tickangle=-45,
+                        tickmode='auto',
+                        nticks=20
+                    )
+                )
             else:
                 # Fallback to bar chart
                 viz_type = 'bar'
@@ -707,24 +714,31 @@ def generate_visualization(result_df, query):
             y_col = numeric_cols[0] if numeric_cols else result_df.columns[1] if len(result_df.columns) > 1 else None
             
             if y_col:
-                fig = px.bar(result_df, x=x_col, y=y_col)
-                fig.update_layout(title_text=f'Bar Chart of {y_col} by {x_col}')
+                fig = px.bar(result_df, x=x_col, y=y_col, width=900)
+                fig.update_layout(
+                    title_text=f'Bar Chart of {y_col} by {x_col}',
+                    bargap=0.2
+                )
             else:
-                fig = px.bar(result_df, x=x_col)
-                fig.update_layout(title_text=f'Bar Chart of {x_col}')
+                fig = px.bar(result_df, x=x_col, width=900)
+                fig.update_layout(
+                    title_text=f'Bar Chart of {x_col}',
+                    bargap=0.2
+                )
         
         # Set common layout properties
         fig.update_layout(
             width=fig_width,
             height=fig_height,
-            template="plotly_white"
+            template="plotly_white",
+            margin=dict(l=40, r=40, t=80, b=80, pad=4),
+            autosize=True
         )
         
         print(f"Created figure with width={fig_width}, height={fig_height}")
         
         # Convert to image
         print("Converting figure to image...")
-        import plotly.io as pio
         img_bytes = pio.to_image(fig, format="png", width=fig_width, height=fig_height, scale=2)
         print("Image conversion successful")
         
@@ -736,7 +750,7 @@ def generate_visualization(result_df, query):
         print("HTML conversion successful")
         
         # Return the HTML img tag
-        return f"<img src='{img_src}' width='100%' style='min-height:700px;' />"
+        return f"<img src='{img_src}' width='100%' style='max-width:1200px; min-height:650px; display:block; margin:0 auto;' />"
         
     except Exception as e:
         import traceback