Update app.py

app.py

@@ -30,29 +30,26 @@ generator = pipeline(
 
 # ---------------- HELPERS ----------------
 def format_prompt(chat_history, user_message, system_message, response_style):
+    # Start with system message
     prompt = system_message + "\n\n"
     
-    # Include past conversation without "User/Assistant" prefixes
+    # Append previous conversation content only
     for turn in chat_history:
-        if turn["role"] == "user":
-            prompt += f"Question: {turn['content']}\n"
-        else:
-            prompt += f"Answer: {turn['content']}\n"
+        prompt += f"{turn['content']}\n"
     
-    # Add the current message
-    prompt += f"Question: {user_message}\nAnswer:"
+    # Append the new user message
+    prompt += user_message
     
-    # Style control
+    # Optional: add response style instruction
     if response_style == "No explanation":
-        prompt += " Give only the direct answer, no explanation."
+        prompt += " Provide only the answer."
     elif response_style == "Short explanation":
-        prompt += " Give a short one-sentence explanation."
+        prompt += " Provide a short one-sentence explanation."
     elif response_style == "Detailed explanation":
-        prompt += " Give a detailed answer with reasoning and examples."
+        prompt += " Provide a detailed explanation."
     
     return prompt
 
-
 # ---------------- CHAT FUNCTION ----------------
 def chat(user_message, chat_history, system_message, max_tokens, temperature, top_p, response_style):
     chat_history = chat_history or []
@@ -66,8 +63,10 @@ def chat(user_message, chat_history, system_message, max_tokens, temperature, to
         top_p=top_p,
     )[0]['generated_text']
     
-    response = output.strip()
+    # Sometimes Flan-T5 outputs the question in the result, so strip it
+    response = output.replace(prompt, "").strip()
     
+    # Save user and assistant content without labels
     chat_history.append({"role": "user", "content": user_message})
     chat_history.append({"role": "assistant", "content": response})