Update app.py

app.py

@@ -133,7 +133,7 @@ def get_response_from_pdf(query):
     if os.path.exists("faiss_database"):
         database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
     else:
-        return "No documents available. Please upload PDF documents to answer questions.", ""
+        return "No documents available. Please upload PDF documents to answer questions."
 
     retriever = database.as_retriever()
     relevant_docs = retriever.get_relevant_documents(query)
@@ -142,20 +142,15 @@ def get_response_from_pdf(query):
     prompt = f"""<s>[INST] Using the following context from the PDF documents:
 {context_str}
 Write a detailed and complete response that answers the following user question: '{query}'
-After writing the response, please provide a list of sources used (document names) in your answer. [/INST]"""
+Do not include a list of sources in your response. [/INST]"""
 
     generated_text = generate_chunked_response(prompt)
 
-    # Remove all instruction-related content
+    # Clean the response
     clean_text = re.sub(r'<s>\[INST\].*?\[/INST\]\s*', '', generated_text, flags=re.DOTALL)
     clean_text = clean_text.replace("Using the following context from the PDF documents:", "").strip()
 
-    # Split the content and sources
-    parts = clean_text.split("Sources:", 1)
-    main_content = parts[0].strip()
-    sources = parts[1].strip() if len(parts) > 1 else ""
-
-    return main_content, sources
+    return clean_text
 
 def get_response_with_search(query):
     search_results = duckduckgo_search(query)
@@ -169,7 +164,7 @@ After writing the document, please provide a list of sources used in your respon
     
     generated_text = generate_chunked_response(prompt)
     
-    # Remove all instruction-related content
+    # Clean the response
     clean_text = re.sub(r'<s>\[INST\].*?\[/INST\]\s*', '', generated_text, flags=re.DOTALL)
     clean_text = clean_text.replace("Using the following context:", "").strip()
     
@@ -183,10 +178,11 @@ After writing the document, please provide a list of sources used in your respon
 def chatbot_interface(message, history, use_web_search):
     if use_web_search:
         main_content, sources = get_response_with_search(message)
+        formatted_response = f"{main_content}\n\nSources:\n{sources}"
     else:
-        main_content, sources = get_response_from_pdf(message)
-    
-    formatted_response = f"{main_content}\n\nSources:\n{sources}"
+        response = get_response_from_pdf(message)
+        formatted_response = response  # No sources for PDF responses
+
     history.append((message, formatted_response))
     return history