Error Fix

app.py

@@ -638,8 +638,8 @@ def generate_visualization(result_df, query):
         print("Visualization requested, attempting to create plot...")
         
         # Set common figure parameters
-        fig_width = 1200
-        fig_height = 650
+        fig_width = 900  # Adjusted for a more square shape
+        fig_height = 800  # Increased to make it more square
         
         # Determine visualization type from query
         viz_type = 'bar'  # Default
@@ -724,26 +724,60 @@ def generate_visualization(result_df, query):
             x_col = result_df.columns[0]
             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, width=900)
-                fig.update_layout(
-                    title_text=f'Bar Chart of {y_col} by {x_col}',
-                    bargap=0.2
-                )
+            # Check if we have many categories (more than 10)
+            if len(result_df) > 10:
+                # Use horizontal bar chart for many categories
+                if y_col:
+                    fig = px.bar(
+                        result_df, 
+                        y=x_col,  # Swap x and y for horizontal orientation
+                        x=y_col,
+                        orientation='h',  # Horizontal orientation
+                        title=f'Bar Chart of {y_col} by {x_col}'
+                    )
+                else:
+                    fig = px.bar(
+                        result_df,
+                        y=x_col,  # Swap x and y for horizontal orientation
+                        orientation='h',  # Horizontal orientation
+                        title=f'Bar Chart of {x_col}'
+                    )
             else:
-                fig = px.bar(result_df, x=x_col, width=900)
-                fig.update_layout(
-                    title_text=f'Bar Chart of {x_col}',
-                    bargap=0.2
-                )
+                # Use vertical bar chart for fewer categories
+                if y_col:
+                    fig = px.bar(
+                        result_df,
+                        x=x_col,
+                        y=y_col,
+                        title=f'Bar Chart of {y_col} by {x_col}',
+                        width=900,
+                        height=800
+                    )
+                else:
+                    fig = px.bar(
+                        result_df,
+                        x=x_col,
+                        title=f'Bar Chart of {x_col}',
+                        width=900,
+                        height=800
+                    )
+            
+            # Improve bar chart layout
+            fig.update_layout(
+                bargap=0.2,  # Increase gap between bars
+                uniformtext_minsize=8,  # Minimum text size
+                uniformtext_mode='hide'  # Hide text if it doesn't fit
+            )
         
         # Set common layout properties
         fig.update_layout(
             width=fig_width,
             height=fig_height,
             template="plotly_white",
-            margin=dict(l=40, r=40, t=80, b=80, pad=4),
-            autosize=True
+            margin=dict(l=40, r=40, t=80, b=80, pad=4),  # Balanced margins
+            autosize=True,  # Allow the plot to resize with the container
+            plot_bgcolor='rgba(240,240,240,0.2)',  # Light gray background
+            paper_bgcolor='white'
         )
         
         print(f"Created figure with width={fig_width}, height={fig_height}")
@@ -761,7 +795,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='max-width:1200px; min-height:650px; display:block; margin:0 auto;' />"
+        return f"<img src='{img_src}' width='100%' style='max-width:900px; height:800px; object-fit:contain; display:block; margin:0 auto;' />"
         
     except Exception as e:
         import traceback