Your commit message

app.py

@@ -9,7 +9,7 @@ from langchain_core.prompts import ChatPromptTemplate
 # Add parent directory to path to import backend modules
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from backend.main import process_query, upload_document, process_voice
+from backend.main import process_query, upload_document, process_voice, DocumentAssistant
 from backend.db import SQLiteDB
 from backend.vector_db import ChromaVectorDB
 from backend.query_engine import QueryEngine
@@ -27,6 +27,9 @@ voice_assistant = VoiceAssistant()
 document_parser = DocumentParser()
 document_agents = DocumentAgents()
 
+# Initialize DocumentAssistant
+document_assistant = DocumentAssistant()
+
 # Define the prompt with examples
 query_prompt = ChatPromptTemplate.from_messages(
     [
@@ -92,11 +95,8 @@ def process_text_query(query, history):
     # Log query to database
     sqlite_db.log_query(query)
     
-    # Get relevant documents
-    relevant_docs = vector_db.search(query)
-    
-    # Generate response
-    response = query_engine.generate_response(query, relevant_docs)
+    # Use DocumentAssistant to process the query
+    response = document_assistant.process_query(query)
     
     # Update database with response
     sqlite_db.log_query(query, response)