Update app.py

app.py

@@ -371,7 +371,21 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
         # Use different retrievers based on the selected model
         if selected_model == phi_pipe:
             retriever = phi_retriever
-            prompt = f"Based on the provided documents, {message}"
+            # Retrieve context from vector store
+            context = retriever.get_relevant_documents(message)
+            # Construct a prompt with clear formatting instructions
+            prompt = f"""
+            Based on the provided documents, generate a well-formatted response with the following details:
+            {context}
+
+            Please format the output as follows:
+            Name: [Event Name]
+            Location: [Location]
+            Date and Time: [Date and Time]
+            Description: [Event Description]
+
+            Question: {message}
+            """
         else:
             retriever = gpt_retriever
             prompt_template = QA_CHAIN_PROMPT_1 if choice == "Details" else QA_CHAIN_PROMPT_2
@@ -392,9 +406,7 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
 
             elif selected_model == phi_pipe:
                 # Use Phi-3.5 directly with the formatted prompt and specific vector store
-                context = retriever.get_relevant_documents(message)
-                full_prompt = f"{context}\n\n{prompt}"
-                response = selected_model(full_prompt, **{
+                response = selected_model(prompt, **{
                     "max_new_tokens": 300,
                     "return_full_text": False,
                     "temperature": 0.5,