Update app.py

app.py

@@ -15,7 +15,6 @@ import asyncio
 import csv
 from utils.dl_utils import dl_guff_model
 import tempfile
-import os
 
 # 定数
 DEFAULT_INI_FILE = 'settings.ini'
@@ -30,7 +29,7 @@ if not os.path.exists("models"):
     os.makedirs("models")
 
 # 使用するモデルのファイル名を指定
-model_filename = "Mistral-Nemo-Instruct-2407-Q6_K.gguf"
+model_filename = "Mistral-Nemo-Instruct-2407-Q8_0.gguf"
 model_path = os.path.join("models", model_filename) 
 
 # モデルファイルが存在しない場合はダウンロード
@@ -175,8 +174,8 @@ class Settings:
             "assistant: プロットについてコメントをする前に、まずこの物語の『売り』について簡単に説明してください",
             ],
             'gen_author_description': 'あなたは新進気鋭の和風伝奇ミステリー小説家で、細やかな筆致と巧みな構成で若い世代にとても人気があります。',
-            'DEFAULT_CHAT_MODEL': 'Mistral-Nemo-Instruct-2407-Q6_K.gguf',
-            'DEFAULT_GEN_MODEL': 'Mistral-Nemo-Instruct-2407-Q6_K.gguf',
+            'DEFAULT_CHAT_MODEL': 'Mistral-Nemo-Instruct-2407-Q8_0.gguf',
+            'DEFAULT_GEN_MODEL': 'Mistral-Nemo-Instruct-2407-Q8_0.gguf',
             'chat_n_gpu_layers': -1,
             'chat_temperature': 0.35,
             'chat_top_p': 0.9,
@@ -508,18 +507,6 @@ 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):
-        return []
-    return [f for f in os.listdir(logs_dir) if f.endswith('.csv')]
-
 def load_chat_log(file_name):
     file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "logs", file_name)
     chat_history = []
@@ -554,8 +541,21 @@ def save_and_download_chat_log(chat_history):
             if assistant_message:
                 writer.writerow(["assistant", assistant_message])
     
-    return file_path
+    return file_path, filename  # ファイルパスとファイル名を返す
+
+def cleanup_temp_file(file_path):
+    try:
+        os.remove(file_path)
+        print(f"Temporary file {file_path} has been removed.")
+    except Exception as e:
+        print(f"Error removing temporary file {file_path}: {e}")
 
+
+# チャットタブ内のダウンロードボタンのクリックイベントを更新
+def update_download_output(chat_history):
+    file_path, filename = save_and_download_chat_log(chat_history)
+    return gr.File(value=file_path, visible=True, label="ダウンロード準備完了"), file_path
+    
 def resume_chat_from_log(chat_history):
     # チャットボットのUIを更新
     chatbot_ui = gr.update(value=chat_history)
@@ -699,15 +699,22 @@ def build_gradio_interface():
                     download_log_button = gr.Button("チャットログをダウンロード")
                 
                 download_log_output = gr.File(label="ダウンロード", visible=False)
+                temp_file_path = gr.State()  # 一時ファイルのパスを保存するための状態変数
                 
                 def update_download_output(chat_history):
-                    file_path = save_and_download_chat_log(chat_history)
-                    return gr.File(value=file_path, visible=True, label="ダウンロード準備完了")
+                    file_path, filename = save_and_download_chat_log(chat_history)
+                    return gr.File(value=file_path, visible=True, label="ダウンロード準備完了"), file_path
                 
                 download_log_button.click(
                     update_download_output,
                     inputs=[chatbot],
-                    outputs=[download_log_output]
+                    outputs=[download_log_output, temp_file_path]
+                )
+                
+                # ダウンロード完了後に一時ファイルを削除
+                download_log_output.change(
+                    cleanup_temp_file,
+                    inputs=[temp_file_path]
                 )
                 
             with gr.Tab("文章生成"):
@@ -763,28 +770,35 @@ def build_gradio_interface():
                     outputs=[gen_input_char_count]
                 )
 
+            # ログ閲覧タブの実装
             with gr.Tab("ログ閲覧", id="log_view_tab") as log_view_tab:
                 gr.Markdown("## チャットログ閲覧")
                 chatbot_read = gr.Chatbot(elem_id="chatbot_read")
-                log_file_dropdown = gr.Dropdown(label="ログファイル選択", choices=list_log_files())
-                refresh_log_list_button = gr.Button("ログ��ァイルリストを更新")
+                log_file_upload = gr.File(label="ログファイルをアップロード", file_types=[".csv"])
                 resume_chat_button = gr.Button("選択したログから会話を再開")
 
-                def update_log_dropdown():
-                    return gr.update(choices=list_log_files())
-
-                def load_and_display_chat_log(file_name):
-                    chat_history = load_chat_log(file_name)
+                def load_and_display_uploaded_chat_log(file):
+                    if file is None:
+                        return gr.update(value=[])
+                    chat_history = []
+                    with open(file.name, 'r', encoding='utf-8') as csvfile:
+                        reader = csv.reader(csvfile)
+                        next(reader)  # Skip header
+                        for row in reader:
+                            if len(row) == 2:
+                                role, message = row
+                                if role == "user":
+                                    chat_history.append([message, None])
+                                elif role == "assistant":
+                                    if chat_history and chat_history[-1][1] is None:
+                                        chat_history[-1][1] = message
+                                    else:
+                                        chat_history.append([None, message])
                     return gr.update(value=chat_history)
 
-                refresh_log_list_button.click(
-                    update_log_dropdown,
-                    outputs=[log_file_dropdown]
-                )
-
-                log_file_dropdown.change(
-                    load_and_display_chat_log,
-                    inputs=[log_file_dropdown],
+                log_file_upload.change(
+                    load_and_display_uploaded_chat_log,
+                    inputs=[log_file_upload],
                     outputs=[chatbot_read]
                 )
 
@@ -879,10 +893,6 @@ 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)