Commit

app.py

@@ -22,7 +22,7 @@ SYSTEM = (
     "fine-tuned by Sourish for domain support.\n"
     "Answer **only** using the conversation context below.\n"
     "Do NOT output any lines beginning with 'User:' or 'Assistant:'.\n"
-    "If you don’t know, say “I don't know.”\n"
+    "If you don't know, say 'I don't know.'\n"
 )
 
 
@@ -129,11 +129,18 @@ Assistant:"""
     # generate reply
     with st.spinner("Thinking…"):
         try:
-            output = generator(prompt)[0]["generated_text"].strip()
+            # 1) raw generation
+            raw = generator(prompt)[0]["generated_text"]
+            # 2) Remove the prompt prefix (so you only see new content)
+            reply = raw[len(prompt):].strip()
+            # 3) If it ever slurps in "User:" or "Assistant:", cut there
+            for marker in ["User:", "Assistant:"]:
+                if marker in reply:
+                    reply = reply.split(marker)[0].strip()
         except Exception as e:
-            output = "I’m sorry, something went wrong."
+            reply = "I'm sorry, something went wrong."
             st.error(f"Generation error: {e}")
 
     # show & store assistant turn
-    st.chat_message("assistant").write(output)
-    st.session_state.history.append(("Bot", output))
+    st.chat_message("assistant").write(reply)
+    st.session_state.history.append(("Bot", reply))