Update app.py

app.py

@@ -14,6 +14,8 @@ import threading
 import asyncio
 import csv
 from utils.dl_utils import dl_guff_model
+import io
+import shutil
 
 # 定数
 DEFAULT_INI_FILE = 'settings.ini'
@@ -35,6 +37,11 @@ model_path = os.path.join("models", model_filename)
 if not os.path.exists(model_path):
     dl_guff_model("models", f"https://huggingface.co/MCZK/EZO-Common-9B-gemma-2-it-GGUF/resolve/main/{model_filename}")
 
+
+
+
+
+
 class ConfigManager:
     @staticmethod
     def load_settings(filename):
@@ -442,20 +449,16 @@ class CharacterMaker:
  
     def make_prompt(self, input_str: str):
         prompt_template = """{{chat_author_description}}
-
 {{chat_instructions}}
-
 ・キャラクターの回答例
 {% for qa in example_qa %}
 {{qa}}
 {% endfor %}
-
 ・会話履歴
 {% for history in histories %}
 user: {{history.user}}
 assistant: {{history.assistant}}
 {% endfor %}
-
 user: {{input_str}}
 assistant:"""
         
@@ -509,6 +512,12 @@ def generate_text_wrapper(text, gen_characters, gen_token_multiplier, instructio
     return chat_or_gen(text=text, gen_characters=gen_characters, gen_token_multiplier=gen_token_multiplier, instruction=instruction, mode="gen")
 
 # ログ関連関数
+def delete_all_logs():
+    logs_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "logs")
+    if os.path.exists(logs_dir):
+        shutil.rmtree(logs_dir)
+    os.makedirs(logs_dir)
+
 def list_log_files():
     logs_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "logs")
     if not os.path.exists(logs_dir):
@@ -533,7 +542,8 @@ def load_chat_log(file_name):
                         chat_history.append([None, message])
     return chat_history
 
-def save_chat_log(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")
@@ -550,7 +560,17 @@ def save_chat_log(chat_history):
             if assistant_message:
                 writer.writerow(["assistant", assistant_message])
     
-    return f"チャットログが {file_path} に保存されました。"
+    # ダウンロード用のファイルを生成
+    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])
+    
+    return gr.File.update(value=output.getvalue(), visible=True, filename=filename)
 
 def resume_chat_from_log(chat_history):
     # チャットボットのUIを更新
@@ -692,16 +712,18 @@ def build_gradio_interface():
                 )
                 
                 with gr.Row():
-                    save_log_button = gr.Button("チャットログを保存")
+                    download_log_button = gr.Button("チャットログをダウンロード")
                 
-                save_log_output = gr.Textbox(label="保存状態")
+                download_log_output = gr.File(label="ダウンロード")
                 
-                save_log_button.click(
-                    save_chat_log,
+                download_log_button.click(
+                    save_and_download_chat_log,
                     inputs=[chatbot],
-                    outputs=[save_log_output]
+                    outputs=[download_log_output]
                 )
-
+                
+                save_log_output = gr.Textbox(label="保存状態")
+                
             with gr.Tab("文章生成"):
                 with gr.Row():
                     with gr.Column(scale=2):
@@ -871,6 +893,10 @@ async def start_gradio():
     print(f"{DEFAULT_INI_FILE} をデフォルト設定で上書きします。")
     Settings.create_default_ini(DEFAULT_INI_FILE)
 
+    # 起動時にログを削除
+    delete_all_logs()
+    print("全てのログファイルが削除されました。")
+
     config = ConfigManager.load_settings(DEFAULT_INI_FILE)
     settings = Settings._parse_config(config)