updatation

app.py

@@ -91,18 +91,36 @@ def export_chat_files(history: List[Dict[str, Any]]) -> Dict[str, Optional[str]]
     with open(txt_path, "w", encoding="utf-8") as f:
         for msg in history:
             content = msg.get("content", "")
+
+            # --- FIX: prevent splitlines() crash ---
+            if isinstance(content, list):
+                content = "\n".join([str(x) for x in content])
+            else:
+                content = str(content)
+            # ---------------------------------------
+
             lines = content.splitlines()
 
             clean = [l for l in lines if not l.strip().startswith("🕒")]
+
             if clean and clean[0].startswith("**"):
                 clean = clean[1:]
+
             clean = [l for l in clean if not set(l) == {"-"}]
-            clean = [l.replace("USER:", "").replace("ASSISTANT:", "").strip() for l in clean]
+
+            clean = [
+                l.replace("USER:", "")
+                .replace("ASSISTANT:", "")
+                .strip()
+                for l in clean
+            ]
 
             clean = remove_last_closing_line(clean)
+
             f.write("\n".join(clean).strip() + "\n")
             f.write("-" * 60 + "\n")
 
+
     pdf_path = None
     try:
         from reportlab.lib.pagesizes import A4

src/chatbot.py

@@ -28,13 +28,13 @@ class LocalChatbot:
             pass
 
         system_prompt = self._build_system_prompt(intent)
-        history_text = self.memory.get_formatted(separator=MSG_SEPARATOR) or ""
+        history_text = self.memory.get_formatted(separator=MSG_SEPARATOR)
 
         prompt = (
-            f"System: {system_prompt}\n"
+            f"<start_of_turn>system\n{system_prompt}<end_of_turn>\n"
             f"{history_text}"
-            f"User: {user_message}\n"
-            f"Assistant:"
+            f"<start_of_turn>user\n{user_message}<end_of_turn>\n"
+            f"<start_of_turn>model\n"
         )
         return prompt
 
@@ -55,7 +55,7 @@ class LocalChatbot:
             max_tokens=gen["max_tokens"],
             temperature=gen["temperature"],
             top_p=gen["top_p"],
-            stop=["User:", "System:", "Assistant:"]
+            stop=["User:", "\nUser:", "\nSystem:", "\nAssistant:"]
 
         )
 

src/conversation.py

@@ -29,10 +29,15 @@ class ConversationMemory:
     def get_formatted(self, separator: str = "\n") -> str:
         out_lines = []
         for msg in self.history:
-            role_label = "System" if msg["role"] == "system" else ("User" if msg["role"] == "user" else "Assistant")
-            out_lines.append(f"{role_label}: {msg['content']}")
+            if msg["role"] == "user":
+                out_lines.append(f"User says: {msg['content']}")
+            elif msg["role"] == "assistant":
+                out_lines.append(f"Assistant says: {msg['content']}")
+            else:
+                out_lines.append(f"System says: {msg['content']}")
         return separator.join(out_lines) + (separator if out_lines else "")
 
+
     def clear(self):
         self.history.clear()