updatation

app.py

@@ -287,15 +287,33 @@ 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):
-        files = export_chat_files(history or [])
+        fixed_history = []
+
+        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])
-
-
+    
+    
 if __name__ == "__main__":
     demo.launch()