Error Fix

app.py

@@ -107,18 +107,31 @@ def process_text_query(query, history):
     if not query:
         return "", history
     
-    # Check if this looks like an SQL query for CSV data
-    if any(keyword in query.lower() for keyword in ['sql', 'query', 'table', 'select', 'from', 'where', 'group by', 'data', 'csv', 'average', 'count', 'sum', 'max', 'min']):
-        try:
-            # Connect to the SQLite database
-            conn = sqlite3.connect(DB_PATH)
-            cursor = conn.cursor()
-            
-            # Get list of tables
-            cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
-            tables = [row[0] for row in cursor.fetchall()]
-            
-            if tables:
+    # More specific SQL detection - look for actual SQL-like patterns or explicit SQL requests
+    sql_keywords = ['select', 'from', 'where', 'group by', 'order by', 'having', 'join']
+    data_analysis_keywords = ['average', 'count', 'sum', 'maximum', 'minimum', 'mean', 'analyze', 'calculate']
+    
+    # Check if this is explicitly about the CSV/database data
+    is_sql_query = (
+        any(keyword in query.lower() for keyword in sql_keywords) or
+        ('csv' in query.lower() and any(keyword in query.lower() for keyword in data_analysis_keywords)) or
+        'database' in query.lower() or
+        'table' in query.lower()
+    )
+    
+    try:
+        # Connect to the SQLite database to check if we have any tables
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
+        cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
+        tables = [row[0] for row in cursor.fetchall()]
+        conn.close()
+        
+        if is_sql_query and tables:
+            try:
+                conn = sqlite3.connect(DB_PATH)
+                cursor = conn.cursor()
+                
                 # Build context with table information
                 table_info = []
                 for table in tables:
@@ -129,9 +142,13 @@ def process_text_query(query, history):
                 # Create question with context
                 question_with_context = f"The database contains the following tables:\n{chr(10).join(table_info)}\n\n{query}"
                 
-                # Generate SQL query using the existing query engine
+                # Generate SQL query using the query engine
                 sql_query = query_engine.generate_response(query_prompt.format(question=question_with_context))
                 
+                # Verify the response is actually a SQL query
+                if not any(keyword in sql_query.lower() for keyword in ['select', 'from']):
+                    raise ValueError("Generated response is not a valid SQL query")
+                
                 try:
                     # Execute the query
                     result_df = pd.read_sql_query(sql_query, conn)
@@ -142,15 +159,6 @@ def process_text_query(query, history):
                     else:
                         data_str = result_df.to_string()
                     
-                    # Interpret the results using the existing query engine
-                    interpretation = query_engine.generate_response(
-                        interpret_prompt.format(
-                            question=query,
-                            sql_query=sql_query,
-                            data=data_str
-                        )
-                    )
-                    
                     # Create the response
                     response = f"**SQL Query:**\n```sql\n{sql_query}\n```\n\n"
                     
@@ -159,19 +167,20 @@ def process_text_query(query, history):
                     else:
                         response += "**No results found.**\n\n"
                     
-                    response += f"**Analysis:**\n{interpretation}"
-                    
                 except Exception as e:
                     response = f"**SQL Query:**\n```sql\n{sql_query}\n```\n\n**Error executing query:** {str(e)}"
-            else:
-                response = "No CSV data has been uploaded yet. Please upload a CSV file first."
-            
-            conn.close()
-        except Exception as e:
-            # Fall back to regular document query
+                
+                conn.close()
+                
+            except Exception as e:
+                # If there's an error with SQL processing, fall back to document query
+                response = document_assistant.process_query(query)
+        else:
+            # Process regular document query
             response = document_assistant.process_query(query)
-    else:
-        # Process regular document query
+            
+    except Exception as e:
+        # If there's any database connection error, fall back to document query
         response = document_assistant.process_query(query)
     
     # Update history with message format