Update app.py

app.py

@@ -87,32 +87,29 @@ pipe = pipeline(
 
 # --- Chat Function ---
 def chat_fn(message, history):
-    # Convert history into text for prompt
-    history_text = ""
-    for msg in history[-10:]:
-        role = "User" if msg["role"] == "user" else "Assistant"
-        history_text += f"{role}: {msg['content']}\n"
-    history_text += f"User: {message}\nAssistant:"
-
-    # Generate response from model
+    # Use the same format as training data
+    prompt = f"### Question:\n{message}\n\n### Answer:\n"
+
+    # Generate response
     response = pipe(
-        history_text,
-        max_new_tokens=16,
+        prompt,
+        max_new_tokens=32,   # allow longer output
         do_sample=False,
         temperature=0.7,
         top_p=1.0
     )[0]["generated_text"]
 
-    # Extract assistant reply
-    reply = response.split("Assistant:")[-1].strip()
+    # Extract only the answer part
+    reply = response.split("### Answer:")[-1].strip()
 
-    # Format reply into bullet points
+    # Format into bullet points
     skills = [s.strip() for s in reply.replace(",", "\n").split("\n") if s.strip()]
     formatted_reply = "\n".join([f"- {s}" for s in skills])
 
     return formatted_reply
 
 
+
 # --- Gradio UI ---
 with gr.Blocks() as demo:
     gr.Markdown("## 🚀 Chat with My AI Skills Model")