updatation

app.py

@@ -34,45 +34,32 @@ def now_ts():
 # ----------------------
 # Export TXT/PDF
 # ----------------------
+# ----------------------
+# Clean PDF Export
+# ----------------------
 def export_chat_files(history: List[Dict[str, Any]]) -> Dict[str, Optional[str]]:
     tmpdir = tempfile.gettempdir()
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     txt_path = os.path.join(tmpdir, f"chat_history_{timestamp}.txt")
     pdf_path = os.path.join(tmpdir, f"chat_history_{timestamp}.pdf")
 
-    def remove_last_closing_line(lines):
-        closing_keywords = [
-            "let me know", "is there anything else", "anything else i can help",
-            "feel free to ask", "hope this helps", "need further assistance",
-            "feel free", "happy to help", "hello! how can i assist you today?",
-            "how can i help you better?", "would you like to explore", "need clarification"
-        ]
-        if not lines:
-            return lines
-        last = lines[-1].lower().strip()
-        if any(k in last for k in closing_keywords):
-            return lines[:-1]
-        return lines
-
-    # TXT
+    # TXT file (simple, clean)
     with open(txt_path, "w", encoding="utf-8") as f:
         for msg in history:
+            role = msg.get("role", "user").capitalize()
             content = msg.get("content", "")
             if isinstance(content, dict):
                 content = content.get("text", "")
-            lines = [l.strip() for l in str(content).splitlines() if l.strip() and not l.strip().startswith("🕒")]
-            lines = remove_last_closing_line(lines)
-            f.write("\n".join(lines) + "\n\n")
-            f.write("-" * 60 + "\n\n")
+            f.write(f"{role}:\n{str(content).strip()}\n\n")
 
-    # PDF
+    # PDF file
     try:
         c = canvas.Canvas(pdf_path, pagesize=A4)
         page_width, page_height = A4
         margin = 40
         y = page_height - margin
-        line_height = 14
-        font_size = 10
+        line_height = 16
+        font_size = 11
         c.setFont("Helvetica", font_size)
 
         for msg in history:
@@ -81,19 +68,18 @@ def export_chat_files(history: List[Dict[str, Any]]) -> Dict[str, Optional[str]]
             if isinstance(content, dict):
                 content = content.get("text", "")
 
-            lines = [l.strip() for l in str(content).splitlines() if l.strip() and not l.strip().startswith("🕒")]
-            lines = remove_last_closing_line(lines)
-
+            # Include timestamp if present
+            lines = str(content).splitlines()
             for line in lines:
-                wrapped = textwrap.wrap(line, width=95)
+                wrapped = textwrap.wrap(line.strip(), width=90)
                 for wline in wrapped:
                     if y < margin + line_height:
                         c.showPage()
                         c.setFont("Helvetica", font_size)
                         y = page_height - margin
-                    c.drawString(margin, y, f"{role}: {wline}")
+                    c.drawString(margin, y, f"{role}: {wline}" if role == "User" else f"{wline}")
                     y -= line_height
-            y -= line_height
+            y -= line_height // 2  # small spacing between messages
 
         c.showPage()
         c.save()