Update app.py

app.py

@@ -4,6 +4,7 @@ from openai import OpenAI
 import json
 import requests
 import datetime
+import tempfile
 
 openai_api_key = os.getenv("OPENROUTER_API_KEY")
 openai_base_url = os.getenv("OPENAI_BASE_URL")
@@ -215,17 +216,18 @@ def export_chat(history):
     # Convert to JSONL format (each line is a JSON object)
     jsonl_content = "\n".join([json.dumps(msg) for msg in messages])
     
-    # Create a file for download
-    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
-    file_name = f"chat_export_{timestamp}.jsonl"
+    # Create a temporary file for download
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jsonl", mode="w", encoding="utf-8")
+    temp_file.write(jsonl_content)
+    temp_file.close()
     
-    return (file_name, jsonl_content)
+    return temp_file.name  # Return the file path
 
 def import_chat(file):
     """Import chat history from a JSONL file."""
     try:
-        content = file.decode('utf-8')
-        lines = [line.strip() for line in content.split('\n') if line.strip()]
+        content = file.read()  # Read the file directly as a string
+        lines = [line.strip() for line in content.split("\n") if line.strip()]
         
         messages = [json.loads(line) for line in lines]
         
@@ -242,7 +244,7 @@ def import_chat(file):
                 i += 2
             else:
                 i += 1
-        
+
         return new_history
     except Exception as e:
         raise gr.Error(f"Error importing chat: {str(e)}")
@@ -282,10 +284,10 @@ with gr.Blocks(css=css) as demo:
         import_btn = gr.UploadButton("Import Chat", file_types=[".jsonl"])
     
     # Connect buttons to functions
-    export_btn.click(fn=export_chat, inputs=chatbot, outputs=gr.File())
+    export_btn.click(fn=export_chat, inputs=chatbot, outputs=gr.File(label="Download Chat History"))
     import_btn.upload(fn=import_chat, inputs=[import_btn], outputs=chatbot)
 
     gr.Markdown(LICENSE)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)