Update app.py

app.py

@@ -237,7 +237,8 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
         Original Question: {original_question}
         Rephrased Search Query: {rephrased_query}
         If the web search results don't contain relevant information, state that the information is not available in the search results.
-        Provide a concise and direct answer to the original question without mentioning the web search or these instructions:
+        Provide a concise and direct answer to the original question without mentioning the web search or these instructions.
+        Do not include any source information in your answer.
         """
         prompt_val = ChatPromptTemplate.from_template(prompt_template)
         formatted_prompt = prompt_val.format(context=context_str, original_question=question, rephrased_query=rephrased_query)
@@ -255,7 +256,8 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
         {context}
         Current Question: {question}
         If the context doesn't contain relevant information, state that the information is not available.
-        Provide a concise and direct answer to the question:
+        Provide a concise and direct answer to the question.
+        Do not include any source information in your answer.
         """
         prompt_val = ChatPromptTemplate.from_template(prompt_template)
         formatted_prompt = prompt_val.format(context=context_str, question=question)
@@ -277,6 +279,12 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
     else:
         answer = full_response.strip()
 
+    # Add sources section
+    if web_search:
+        sources = set(doc.metadata['source'] for doc in web_docs)
+        sources_section = "\n\nSources:\n" + "\n".join(f"- {source}" for source in sources)
+        answer += sources_section
+
     return answer
 
 # Gradio interface