Update app.py

app.py

@@ -1074,31 +1074,30 @@ def google_search(query):
         return ["Error occurred during Google search"]
 
 
+
 def rag_response(query):
+    """
+    Handle RAG-based queries when uploaded document context is not mentioned.
+    """
     try:
-        # Check if the query references uploaded documents
         if "using uploaded document" in query.lower():
-            document_response = handle_document_query(query)  # Use your existing `handle_document_query` function
-            if document_response:
-                return document_response
-        else:
-        
-            # Proceed with the existing knowledge base logic if no uploaded document context is specified
-            retrieved_docs = search_knowledge_base(query)
-            context = "\n".join(doc.page_content for doc in retrieved_docs)
-            
-            # Prepare the prompt with the retrieved context
-            prompt = f"Context:\n{context}\n\nQuestion: {query}\nAnswer:"
-            llm = ChatOpenAI(model="gpt-4o", temperature=0.1, api_key=openai_api_key)
-            response = llm.invoke(prompt)
-            
-            # Replace terms in the final output as per your restrictions
-            response_content = response.content
-            
-            return response_content
+            # Handle document-specific queries
+            return handle_document_query(query)
+
+        # Proceed with the existing knowledge base logic
+        retrieved_docs = search_knowledge_base(query)  # Replace with actual KB search logic
+        if not retrieved_docs:
+            return "No relevant information found in the knowledge base."
+
+        context = "\n".join(doc.page_content for doc in retrieved_docs)
+        prompt = f"Context:\n{context}\n\nQuestion: {query}\nAnswer:"
+        llm = ChatOpenAI(model="gpt-4o", temperature=0.3, api_key=openai_api_key)
+        response = llm.invoke(prompt)
+
+        return response.content
     except Exception as e:
         logger.error(f"Error generating RAG response: {e}")
-        return "Error occurred during RAG response generation"
+        return "Error occurred during RAG response generation."
 
 
 # Define tools
@@ -1840,6 +1839,7 @@ def get_document_content(doc_name=None):
     else:
         return None, "The most recently uploaded document does not contain any content."
 
+
 def handle_document_query(query):
     """
     Handle queries related to uploaded documents for response generation.
@@ -1848,6 +1848,9 @@ def handle_document_query(query):
     doc_name_match = re.search(r"[\"']?([^\"']+\.(pdf|docx|doc|txt))[\"']?", query, re.IGNORECASE)
     doc_name = doc_name_match.group(1) if doc_name_match else None
 
+    if not doc_name:
+        return "No document mentioned in the query."
+
     # Fetch document content
     doc_content, error = get_document_content(doc_name)
     if error:
@@ -1856,13 +1859,13 @@ def handle_document_query(query):
     # Generate AI response with document context
     full_prompt = f"Document Content:\n{doc_content}\n\nUser Query: {query}\n\nResponse:"
     try:
-        llm = ChatOpenAI(model="gpt-4o", temperature=0.5, api_key=openai_api_key)
+        llm = ChatOpenAI(model="gpt-4", temperature=0.5, api_key=openai_api_key)
         response = llm.invoke(full_prompt)
         return response.content
     except Exception as e:
+        logger.error(f"Error generating response using the document: {e}")
         return f"Error generating response using the document: {e}"
 
-