Update app.py

app.py

@@ -543,16 +543,9 @@ def load_chat_log(file_name):
     return chat_history
 
 def save_and_download_chat_log(chat_history):
-    # ログを保存
-    current_time = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    filename = f"{current_time}.csv"
-    logs_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "logs")
-    if not os.path.exists(logs_dir):
-        os.makedirs(logs_dir)
-    file_path = os.path.join(logs_dir, filename)
-    
-    with open(file_path, 'w', newline='', encoding='utf-8') as csvfile:
-        writer = csv.writer(csvfile)
+    # 一時ファイルを作成
+    with tempfile.NamedTemporaryFile(mode='w+', newline='', encoding='utf-8', suffix='.csv', delete=False) as temp_file:
+        writer = csv.writer(temp_file)
         writer.writerow(["Role", "Message"])
         for user_message, assistant_message in chat_history:
             if user_message:
@@ -560,17 +553,11 @@ def save_and_download_chat_log(chat_history):
             if assistant_message:
                 writer.writerow(["assistant", assistant_message])
     
-    # ダウンロード用のファイルを生成
-    output = io.StringIO()
-    writer = csv.writer(output)
-    writer.writerow(["Role", "Message"])
-    for user_message, assistant_message in chat_history:
-        if user_message:
-            writer.writerow(["user", user_message])
-        if assistant_message:
-            writer.writerow(["assistant", assistant_message])
+    # ファイル名を生成
+    current_time = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
+    filename = f"chat_log_{current_time}.csv"
     
-    return (output.getvalue(), filename)
+    return temp_file.name, filename
 
 def resume_chat_from_log(chat_history):
     # チャットボットのUIを更新
@@ -717,15 +704,14 @@ def build_gradio_interface():
                 download_log_output = gr.File(label="ダウンロード", visible=False)
                 
                 def update_download_output(chat_history):
-                    content, filename = save_and_download_chat_log(chat_history)
-                    return gr.File.update(value=content, visible=True, filename=filename)
+                    temp_file_path, filename = save_and_download_chat_log(chat_history)
+                    return gr.File(value=temp_file_path, visible=True, label="ダウンロード準備完了", filename=filename)
                 
                 download_log_button.click(
                     update_download_output,
                     inputs=[chatbot],
                     outputs=[download_log_output]
                 )
-                
                 save_log_output = gr.Textbox(label="保存状態")
                 
             with gr.Tab("文章生成"):