Update app.py

app.py

@@ -1816,11 +1816,14 @@ def load_user_memory(user_id):
         st.session_state["vector_store"] = {}
 
 def get_document_content(doc_name=None):
+    """
+    Retrieve content of an uploaded document from Streamlit session state.
+    """
     documents = st.session_state.get("documents", {})
     if not documents:
         return None, "No documents have been uploaded."
 
-    # If a specific document name is provided
+    # Fetch specific document by name
     if doc_name:
         for doc_id, doc_data in documents.items():
             if doc_data.get("name", "").lower() == doc_name.lower():
@@ -1831,7 +1834,7 @@ def get_document_content(doc_name=None):
                     return None, f"Document '{doc_name}' does not contain any content."
         return None, f"Document '{doc_name}' not found."
 
-    # Default to the most recent document
+    # Default to the most recently uploaded document
     last_doc = list(documents.values())[-1]
     content = last_doc.get("content")
     if content:
@@ -1842,21 +1845,18 @@ def get_document_content(doc_name=None):
 
 def handle_document_query(query):
     """
-    Handle queries related to uploaded documents for response generation.
+    Handle user queries related to uploaded documents.
     """
-    # Extract specific document name if mentioned
+    # Extract document name from the 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:
         return error
 
-    # Generate AI response with document context
+    # Generate AI response using document context
     full_prompt = f"Document Content:\n{doc_content}\n\nUser Query: {query}\n\nResponse:"
     try:
         llm = ChatOpenAI(model="gpt-4", temperature=0.5, api_key=openai_api_key)
@@ -1865,7 +1865,6 @@ def handle_document_query(query):
     except Exception as e:
         logger.error(f"Error generating response using the document: {e}")
         return f"Error generating response using the document: {e}"
-