Error Fix

app.py

@@ -16,6 +16,7 @@ from io import BytesIO
 import speech_recognition as sr
 from gtts import gTTS
 import re
+import importlib.util
 
 # Load environment variables
 load_dotenv()
@@ -892,6 +893,23 @@ def flush_databases():
     
     return "\n".join(result)
 
+# At the beginning of app.py, after the imports
+# Add this code to monkey patch the vector_db module
+try:
+    from backend.vector_db import ChromaVectorDB
+except NameError as e:
+    if "response" in str(e):
+        # If the error is about 'response' not being defined, fix the module
+        import backend.vector_db
+        
+        # Remove the problematic code
+        if hasattr(backend.vector_db, 'response'):
+            delattr(backend.vector_db, 'response')
+        
+        # Reload the module
+        importlib.reload(backend.vector_db)
+        from backend.vector_db import ChromaVectorDB
+
 # Create Gradio interface
 with gr.Blocks(title="AI Document Analysis & Voice Assistant") as demo:
     gr.Markdown("# 🤖 AI Document Analysis & Voice Assistant")

backend/vector_db.py

@@ -2,11 +2,6 @@ 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"):
@@ -66,47 +61,4 @@ class ChromaVectorDB:
             return True
         except Exception as e:
             print(f"Error resetting collection: {str(e)}")
-            return False
-
-# 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>" 
+            return False