Error Fix

app.py

@@ -715,53 +715,34 @@ def process_file_upload(files):
 
 def list_documents():
     """List all indexed documents"""
-    info_list = []
-    
-    # Check for CSV data
     try:
-        conn = sqlite3.connect(DB_PATH)
-        cursor = conn.cursor()
-        cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
-        tables = cursor.fetchall()
+        docs = document_assistant.get_all_documents()
+        if not docs:
+            return "No documents indexed yet."
         
-        if tables:
-            info_list.append("📊 CSV Data Tables:")
-            for table in tables:
-                # Get column info
-                cursor.execute(f"PRAGMA table_info({table[0]});")
-                columns = [col[1] for col in cursor.fetchall()]
-                
-                # Get row count
-                cursor.execute(f"SELECT COUNT(*) FROM {table[0]};")
-                row_count = cursor.fetchone()[0]
-                
-                info_list.append(f"- {table[0]} ({row_count:,} rows, {len(columns)} columns)")
+        result = "Indexed Documents:\n\n"
+        for doc in docs:
+            result += f"- {doc['filename']} ({doc['file_type']})\n"
         
-        conn.close()
+        return result
     except Exception as e:
-        info_list.append(f"Error accessing CSV data: {str(e)}")
-    
-    # Check for indexed documents
-    docs = document_assistant.get_all_documents()
-    if docs:
-        info_list.append("\n📑 Indexed Documents:")
-        for doc in docs:
-            info_list.append(f"- {doc['filename']} (ID: {doc['id']})")
-    
-    if not info_list:
-        return "No data or documents loaded yet"
-    
-    return "\n".join(info_list)
+        return f"Error listing documents: {str(e)}"
 
 def clear_context():
-    """Clear the current context and chat history"""
+    """Clear the current context"""
     global current_context
-    current_context = {
-        "file_type": None,
-        "file_name": None,
-        "table_name": None
-    }
-    return None
+    
+    try:
+        # Reset the context
+        current_context = {
+            "file_type": None,
+            "file_name": None,
+            "table_name": None
+        }
+        
+        return [{"role": "assistant", "content": "Context cleared. You can now upload new documents or CSV files."}]
+    except Exception as e:
+        return [{"role": "assistant", "content": f"Error clearing context: {str(e)}"}]
 
 def process_voice_input(audio_path):
     """Process voice input and return transcribed text"""
@@ -780,7 +761,7 @@ def process_voice_input(audio_path):
             # Recognize speech using Google Speech Recognition
             text = r.recognize_google(audio_data)
             
-    return text
+            return text
     except sr.UnknownValueError:
         return "Could not understand audio"
     except sr.RequestError as e:
@@ -825,39 +806,47 @@ def text_to_speech_output(text):
 
 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
+    try:
+        # 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
+    except Exception as e:
+        print(f"Error creating test visualization: {str(e)}")
+        return None
 
 def create_test_html_visualization():
     """Create a test HTML 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')
-    
-    # Convert to HTML with CDN-hosted plotly.js
-    html = pio.to_html(fig, full_html=False, include_plotlyjs='cdn')
-    
-    return html
+    try:
+        # 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')
+        
+        # Convert to HTML
+        html = pio.to_html(fig, full_html=False)
+        
+        return html
+    except Exception as e:
+        print(f"Error creating test HTML visualization: {str(e)}")
+        return None
 
 def flush_databases():
     """Flush ChromaDB and SQLite databases"""
@@ -885,8 +874,11 @@ def flush_databases():
     
     # Flush ChromaDB by resetting the document assistant
     try:
-        document_assistant.reset_database()
-        result.append("✅ ChromaDB cleared successfully")
+        success = document_assistant.reset_database()
+        if success:
+            result.append("✅ ChromaDB cleared successfully")
+        else:
+            result.append("⚠️ ChromaDB reset may not have been complete")
     except Exception as e:
         result.append(f"❌ Error clearing ChromaDB: {str(e)}")
     
@@ -1035,8 +1027,11 @@ with gr.Blocks(title="AI Document Analysis & Voice Assistant") as demo:
         save_btn = gr.Button("Save Settings")
         
         def save_settings(key):
-            os.environ["GROQ_API_KEY"] = key
-            return "Settings saved!"
+            try:
+                os.environ["GROQ_API_KEY"] = key
+                return "Settings saved!"
+            except Exception as e:
+                return f"Error saving settings: {str(e)}"
         
         save_btn.click(
             save_settings,