Update app.py

app.py

@@ -9,6 +9,7 @@ import configparser
 from utils.dl_utils import dl_guff_model
 import io
 import tempfile
+import csv
 
 # モデルディレクトリが存在しない場合は作成
 if not os.path.exists("models"):
@@ -60,6 +61,16 @@ def load_settings_from_ini(filename='character_settings.ini'):
     except KeyError:
         return None
 
+def export_chat_history_to_csv(history):
+    with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.csv', newline='', encoding='utf-8') as temp_file:
+        writer = csv.writer(temp_file)
+        writer.writerow(["Role", "Message"])  # CSV header
+        for message in history:
+            writer.writerow(["User", message[0]])
+            writer.writerow(["Assistant", message[1]])
+        temp_file_path = temp_file.name
+    return temp_file_path
+
 # LlamaCppのラッパークラス
 class LlamaCppAdapter:
     def __init__(self, model_path, n_ctx=4096):
@@ -299,6 +310,14 @@ with gr.Blocks(css=custom_css) as iface:
             undo_btn="前のメッセージを取り消す",
             clear_btn="チャットをクリア",
         )
+        export_csv_button = gr.Button("チャット履歴をCSVでエクスポート")
+        export_csv_output = gr.File(label="エクスポートされたCSVファイル")
+        
+        export_csv_button.click(
+            export_chat_history_to_csv,
+            inputs=[chatbot],
+            outputs=[export_csv_output]
+        )      
         
         examples = gr.Examples(
             examples=[