Update app.py

app.py

@@ -32,24 +32,26 @@ generator = pipeline(
 def format_prompt(chat_history, user_message, system_message, response_style):
     # Start with system message
     prompt = system_message + "\n\n"
-    
-    # Append previous conversation content only
+
+    # Add only user messages (optional: you can also add last assistant reply if needed)
     for turn in chat_history:
-        prompt += f"{turn['content']}\n"
-    
-    # Append the new user message
-    prompt += user_message
+        if turn['role'] == 'user':
+            prompt += f"{turn['content']}\n"
     
-    # Optional: add response style instruction
+    # Add the new user message
+    prompt += f"{user_message}\n"
+
+    # Optionally instruct for explanation style
     if response_style == "No explanation":
-        prompt += " Provide only the answer."
+        prompt += " Answer concisely with no explanation."
     elif response_style == "Short explanation":
-        prompt += " Provide a short one-sentence explanation."
+        prompt += " Answer briefly with a one-sentence explanation."
     elif response_style == "Detailed explanation":
-        prompt += " Provide a detailed explanation."
+        prompt += " Answer in detail with reasoning and examples."
     
     return prompt
 
+
 # ---------------- CHAT FUNCTION ----------------
 def chat(user_message, chat_history, system_message, max_tokens, temperature, top_p, response_style):
     chat_history = chat_history or []