updatation

app.py

@@ -42,13 +42,9 @@ def export_chat_files(history: List[Dict[str, Any]]) -> Dict[str, Optional[str]]
             "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?",
-            "Are there any specific industries or areas you'd like to explore in more detail?" ,
-            "How can I help you better?" ,
-            "What did you like about our interaction?",
-            "Do you have any feedback on your experience?" ,
- 
-            "would you like to explore", "need clarification"
+            "feel free", "happy to help",
+            "how can i assist", "do you have",
+            "would you like to", "need clarification"
         ]
         if not lines:
             return lines
@@ -59,19 +55,28 @@ 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:
+            # Ensure always a dict
+            if not isinstance(msg, dict):
+                continue
+
             content = msg.get("content", "")
+            if not isinstance(content, str):
+                content = str(content)
+
             lines = content.splitlines()
 
+            # CLEANUP
             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 for l in clean if l.strip() != ""]
             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")
 
+            if clean:
+                f.write("\n".join(clean).strip() + "\n")
+                f.write("-" * 60 + "\n")
+
+    # PDF
     pdf_path = None
     try:
         from reportlab.lib.pagesizes import A4
@@ -81,12 +86,15 @@ def export_chat_files(history: List[Dict[str, Any]]) -> Dict[str, Optional[str]]
         c = canvas.Canvas(pdf_path, pagesize=A4)
         width, height = A4
         margin = 40
+
         textobject = c.beginText(margin, height - margin)
         textobject.setFont("Helvetica", 10)
+
         with open(txt_path, "r", encoding="utf-8") as fh:
             for line in fh:
                 for wrapped in textwrap.wrap(line.rstrip(), 100):
                     textobject.textLine(wrapped)
+
         c.drawText(textobject)
         c.showPage()
         c.save()
@@ -287,31 +295,17 @@ with gr.Blocks(title="Tayyab — Chatbot") as demo:
 
     new_chat_btn.click(new_chat, outputs=[chatbot])
 
-    # def export_handler(history):
-    #     files = export_chat_files(history or [])
-    #     return (
-    #         gr.update(value=files.get("txt"), visible=True),
-    #         gr.update(value=files.get("pdf"), visible=bool(files.get("pdf")))
-    #     )
-
-    # export_btn.click(export_handler, inputs=[chatbot], outputs=[file_txt, file_pdf])
-
-
 
     def export_handler(history):
-        fixed_history = []
+        # history is already list of dicts: [{"role": "...", "content": "..."}]
+        files = export_chat_files(history or [])
 
-        for user_msg, bot_msg in history:
-            fixed_history.append({"role": "user", "content": user_msg})
-            fixed_history.append({"role": "assistant", "content": bot_msg})
-
-        files = export_chat_files(fixed_history)
-        
         return (
             gr.update(value=files.get("txt"), visible=True),
             gr.update(value=files.get("pdf"), visible=bool(files.get("pdf")))
         )
 
+
     export_btn.click(export_handler, inputs=[chatbot], outputs=[file_txt, file_pdf])