Error Fix

backend/vector_db.py

@@ -2,6 +2,11 @@ import os
 import chromadb
 from typing import List, Dict, Any
 import hashlib
+import plotly.express as px
+import pandas as pd
+import base64
+import plotly.io as pio
+import traceback
 
 class ChromaVectorDB:
     def __init__(self, db_path: str = "./data/chroma_db"):
@@ -49,4 +54,47 @@ class ChromaVectorDB:
         )
         
         if results and results['ids']:
-            self.collection.delete(ids=results['ids']) 
+            self.collection.delete(ids=results['ids'])
+
+# 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
+)
+
+# Try both methods
+try:
+    # Method 1: Convert to image
+    print("Visualization requested, attempting to create plot...")
+    # ...
+    print(f"Created figure with width={fig.layout.width}, height={fig.layout.height}")
+    # ...
+    print("Converting figure to image...")
+    img_bytes = fig.to_image(format="png", width=fig.layout.width, height=fig.layout.height, scale=2)
+    print("Image conversion successful")
+    # ...
+except Exception as viz_error:
+    print(f"Visualization error: {str(viz_error)}")
+    traceback.print_exc()
+    response += f"\n\n**Visualization Error:** {str(viz_error)}"
+
+try:
+    # Method 2: Convert to HTML
+    html = pio.to_html(fig, full_html=False, include_plotlyjs='cdn')
+    print("HTML conversion successful")
+except Exception as e:
+    print(f"HTML conversion failed: {str(e)}")
+
+# Replace the image conversion with HTML
+html_plot = pio.to_html(fig, full_html=False, include_plotlyjs='cdn')
+response += f"\n\n<div style='width:100%; height:700px;'>{html_plot}</div>"