Update app.py

app.py

@@ -397,6 +397,7 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search, c
     return "An unexpected error occurred. Please try again later."
 
 def extract_answer(full_response):
+    # First, try to split the response at common instruction phrases
     answer_patterns = [
         r"Provide a concise and direct answer to the question without mentioning the web search or these instructions:",
         r"Provide a concise and direct answer to the question:",
@@ -404,16 +405,26 @@ def extract_answer(full_response):
         r"Provide a summarized and direct answer to the question.",
         r"If the context doesn't contain relevant information, state that the information is not available in the document.",
         r"Provide a summarized and direct answer to the original question without mentioning the web search or these instructions:",
-        r"without mentioning the web search or these instructions.",
-        r"Do not include any source information in your answer:",
         r"Do not include any source information in your answer."     
     ]
 
     for pattern in answer_patterns:
         match = re.split(pattern, full_response, flags=re.IGNORECASE)
         if len(match) > 1:
-            return match[-1].strip()
-    return full_response.strip()
+            full_response = match[-1].strip()
+            break
+    
+    # Then, remove any remaining instruction-like phrases
+    cleanup_patterns = [
+        r"without mentioning the web search or these instructions\.",
+        r"Do not include any source information in your answer\.",
+        r"If the context doesn't contain relevant information, state that the information is not available in the document\."
+    ]
+
+    for pattern in cleanup_patterns:
+        full_response = re.sub(pattern, "", full_response, flags=re.IGNORECASE).strip()
+    
+    return full_response
 
 # Gradio interface
 with gr.Blocks() as demo: