Update app.py

app.py

@@ -394,20 +394,23 @@ chain_neo4j = (
 phi_short_template = f"""
 As an expert on Birmingham, Alabama, I will provide concise, accurate, and informative responses to your queries. Given the sunny weather today, {current_date}, feel free to ask me anything you need to know about the city.
 
-- Brief details and facts
-- No unnecessary elaboration
-- Focus on relevance
+Provide only the direct answer to the question without any follow-up questions.
 
 {{context}}
 Question: {{question}}
 Answer:
 """
 
-# Incorporate this template into the Phi-3.5 response generation section
+
+def clean_response(response_text):
+    # Remove any unwanted follow-up questions or unnecessary text
+    cleaned_response = re.sub(r'Question:.*\nAnswer:', '', response_text, flags=re.DOTALL).strip()
+    return cleaned_response
+
 def generate_answer(message, choice, retrieval_mode, selected_model):
     logging.debug(f"generate_answer called with choice: {choice} and retrieval_mode: {retrieval_mode}")
 
-    try:
+    try {
         if "hotel" in message.lower() or "hotels" in message.lower() and "birmingham" in message.lower():
             response = fetch_google_hotels()
             return response, extract_addresses(response)
@@ -453,7 +456,8 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
                 if response:
                     generated_text = response[0]['generated_text']
                     logging.debug(f"Phi-3.5 Response: {generated_text}")
-                    return generated_text, extract_addresses(generated_text)
+                    cleaned_response = clean_response(generated_text)
+                    return cleaned_response, extract_addresses(cleaned_response)
                 else:
                     logging.error("Phi-3.5 did not return any response.")
                     return "No response generated.", []
@@ -464,9 +468,10 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
         else:
             return "Invalid retrieval mode selected.", []
 
-    except Exception as e:
+    } catch (Exception e) {
         logging.error(f"Error in generate_answer: {e}")
         return "Sorry, I encountered an error while processing your request.", []
+}