Update app.py

app.py

@@ -476,6 +476,14 @@ def clean_response(response_text):
 
     return cleaned_response
 
+# Define a new template specifically for GPT-4o-mini in VDB Details mode
+gpt4o_mini_template_details = f"""
+As a highly specialized assistant, I provide precise, detailed, and informative responses. On this bright day of {current_date}, I'm equipped to assist with all your queries about Birmingham, Alabama, offering detailed insights tailored to your needs.
+Given your request, here is the detailed information you're seeking:
+{{context}}
+Question: {{question}}
+Detailed Answer:
+"""
 
 
 
@@ -489,9 +497,20 @@ def generate_answer(message, choice, retrieval_mode, selected_model):
         choice = None
         retrieval_mode = None
 
+    # try:
+    #     # Select the appropriate template based on the choice
+    #     if choice == "Details":
+    #         prompt_template = QA_CHAIN_PROMPT_1
+    #     elif choice == "Conversational":
+    #         prompt_template = QA_CHAIN_PROMPT_2
+    #     else:
+    #         prompt_template = QA_CHAIN_PROMPT_1  # Fallback to template1
+
     try:
-        # Select the appropriate template based on the choice
-        if choice == "Details":
+        # Select the appropriate template based on the choice and model
+        if choice == "Details" and selected_model == chat_model1:  # GPT-4o-mini
+            prompt_template = PromptTemplate(input_variables=["context", "question"], template=gpt4o_mini_template_details)
+        elif choice == "Details":
             prompt_template = QA_CHAIN_PROMPT_1
         elif choice == "Conversational":
             prompt_template = QA_CHAIN_PROMPT_2