Update app.py

app.py

@@ -14,9 +14,23 @@ base_model = AutoModelForCausalLM.from_pretrained(
 
 model = PeftModel.from_pretrained(base_model, "Rustamshry/Plantinga-RL").to("cpu")
 
+system = """
+Respond in the following format:
+<reasoning>
+Explain your reasoning here step-by-step.
+</reasoning>
+<answer>
+Give the final answer here.
+</answer>
+"""
 
 # --- Chatbot logic ---
 def generate_response(user_input, chat_history):
+
+    if not chat_history:
+        # Add the system message only once, at the beginning
+        chat_history = [{"role": "system", "content": system}]
+    
     if not user_input.strip():
         return chat_history, chat_history