Error Fix

app.py

@@ -10,6 +10,7 @@ from langchain_groq import ChatGroq
 import plotly.express as px
 import time
 import plotly.io as pio
+import traceback
 
 # Load environment variables
 load_dotenv()
@@ -212,25 +213,37 @@ def process_text_query(query, history):
                                 x_col = result_df.columns[0]
                                 y_cols = numeric_cols[:3]  # Use up to 3 numeric columns
                                 
+                                print(f"Creating plot with x={x_col}, y={y_cols}")
+                                
                                 # Create appropriate plot based on data characteristics
-                                if 'month' in result_df.columns or 'date' in result_df.columns or 'year' in result_df.columns:
+                                if 'month' in result_df.columns or 'date' in result_df.columns or 'year' in result_df.columns or any('date' in col.lower() for col in result_df.columns):
                                     # Time series data - use line chart
                                     fig = px.line(result_df, x=x_col, y=y_cols, title="Time Series Analysis")
                                 else:
                                     # Regular data - use bar chart
                                     fig = px.bar(result_df, x=x_col, y=y_cols[0], title="Data Visualization")
                                 
+                                # Ensure the figure is properly configured
+                                fig.update_layout(
+                                    autosize=True,
+                                    width=800,
+                                    height=500,
+                                    margin=dict(l=50, r=50, b=100, t=100, pad=4)
+                                )
+                                
                                 # Store the figure for display
                                 current_plot = fig
+                                print(f"Plot created successfully: {current_plot is not None}")
                                 
                                 # Add note about visualization
                                 response += "\n\n**A visualization has been generated and is displayed below.**"
-                        
-                        # After creating the plot
-                        print(f"Plot created: {current_plot is not None}")
-                        
+                            else:
+                                print("Not enough numeric columns or data points for visualization")
+                        else:
+                            print("Not enough columns for visualization")
                     except Exception as viz_error:
                         print(f"Visualization error: {str(viz_error)}")
+                        traceback.print_exc()  # Add this to get a full traceback
                         # Continue without visualization if there's an error
                 
             except Exception as e:
@@ -417,6 +430,26 @@ def text_to_speech_output(text):
     # Since we don't have VoiceAssistant, return None
     return None
 
+def create_test_visualization():
+    """Create a test visualization to verify plotting works"""
+    # Create sample data
+    data = pd.DataFrame({
+        'Month': ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun'],
+        'Value': [10, 15, 13, 17, 20, 25]
+    })
+    
+    # Create a simple bar chart
+    fig = px.bar(data, x='Month', y='Value', title='Test Visualization')
+    
+    # Configure the figure
+    fig.update_layout(
+        autosize=True,
+        width=800,
+        height=500
+    )
+    
+    return fig
+
 # Create Gradio interface
 with gr.Blocks(title="AI Document Analysis & Voice Assistant") as demo:
     gr.Markdown("# 🤖 AI Document Analysis & Voice Assistant")
@@ -425,8 +458,14 @@ with gr.Blocks(title="AI Document Analysis & Voice Assistant") as demo:
     with gr.Tab("Chat"):
         chatbot = gr.Chatbot(height=400, type="messages")
         
-        # Add a plot component
-        plot_output = gr.Plot(label="Data Visualization", visible=False)
+        # Add a plot component with explicit sizing
+        plot_output = gr.Plot(
+            label="Data Visualization", 
+            visible=False,
+            elem_id="visualization_plot",  # Add an element ID for easier debugging
+            container=True,  # Ensure it has a container
+            min_width=700    # Set minimum width
+        )
         
         with gr.Row():
             with gr.Column(scale=8):
@@ -532,6 +571,16 @@ with gr.Blocks(title="AI Document Analysis & Voice Assistant") as demo:
             inputs=[api_key],
             outputs=[gr.Textbox(label="Status")]
         )
+        
+        gr.Markdown("## Debugging")
+        test_viz_btn = gr.Button("Test Visualization")
+        test_viz_output = gr.Plot(label="Test Visualization")
+        
+        test_viz_btn.click(
+            create_test_visualization,
+            inputs=[],
+            outputs=[test_viz_output]
+        )
 
 # Launch the app
 if __name__ == "__main__":