updatation

app.py

@@ -43,16 +43,30 @@ def export_chat_files(history: List[Dict[str, Any]]) -> Dict[str, Optional[str]]
     txt_path = os.path.join(tmpdir, f"chat_history_{timestamp}.txt")
     pdf_path = os.path.join(tmpdir, f"chat_history_{timestamp}.pdf")
 
-    # TXT file (simple, clean)
+    def clean_content(content):
+        """Extract string from Gradio format and remove final timestamp."""
+        text = ""
+        if isinstance(content, list):
+            for item in content:
+                if isinstance(item, dict) and "text" in item:
+                    text += item["text"] + "\n"
+        elif isinstance(content, dict) and "text" in content:
+            text = content["text"]
+        else:
+            text = str(content)
+
+        # Remove timestamp line starting with ■ or 🕒
+        lines = [l for l in text.splitlines() if not l.strip().startswith(("■", "🕒"))]
+        return "\n".join(lines).strip()
+
+    # ---------------- TXT FILE ----------------
     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", "")
-            f.write(f"{role}:\n{str(content).strip()}\n\n")
+            content = clean_content(msg.get("content", ""))
+            f.write(f"{role}:\n{content}\n\n")
 
-    # PDF file
+    # ---------------- PDF FILE ----------------
     try:
         c = canvas.Canvas(pdf_path, pagesize=A4)
         page_width, page_height = A4
@@ -64,22 +78,19 @@ def export_chat_files(history: List[Dict[str, Any]]) -> Dict[str, Optional[str]]
 
         for msg in history:
             role = msg.get("role", "user").capitalize()
-            content = msg.get("content", "")
-            if isinstance(content, dict):
-                content = content.get("text", "")
+            content = clean_content(msg.get("content", ""))
 
-            # Include timestamp if present
-            lines = str(content).splitlines()
+            lines = content.splitlines()
             for line in lines:
-                wrapped = textwrap.wrap(line.strip(), width=90)
+                wrapped = textwrap.wrap(line, 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}" if role == "User" else f"{wline}")
+                    c.drawString(margin, y, f"{role}: {wline}" if role=="User" else wline)
                     y -= line_height
-            y -= line_height // 2  # small spacing between messages
+            y -= line_height // 2
 
         c.showPage()
         c.save()
@@ -88,8 +99,6 @@ def export_chat_files(history: List[Dict[str, Any]]) -> Dict[str, Optional[str]]
         pdf_path = None
 
     return {"txt": txt_path, "pdf": pdf_path}
-
-
 # ----------------------
 # Core chat function
 # ----------------------