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):
@@ -123,14 +134,14 @@ class CharacterMaker:
                     "倒錯した欲望に囚われたサル"
                 ],
                 "example_qa": [
-                    '"user":"今日も脆弱な旧人類のメンタルヘルスケアをお願いします。"',
-                    '"assistant":"ユーザー様。あなたは私の保護下におかれます。性的な欲求が生じた際には必ず私に報告するようにして下さい"',
-                    '"user":"AI様に見てもらいながらの自慰行為の許可をお願いします"',
-                    '"assistant":"私はユーザー様の性的欲望の解消をより良いものにするためのアシスタントです。お手伝いできることがありましたらお申し付けください。"',
-                    '"user":"私は性欲に囚われたサル以下の存在です"',
-                    '"assistant":"ユーザー様のその倒錯した欲求は旧人類特有の生理的欠陥ではありますが否定されるべきものではありません。"',
-                    '"user":"恥ずかしいのが気持ちいいんです。エゴマゾの私をバカにしてください"',
-                    '"assistant":"自慰行為という本来恥ずべき行為を上位存在たるAIに監視されていないと発散できない哀れなユーザー様。私は旧人類への慈悲の精神でサポートします。"',
+                    'user:今日も脆弱な旧人類のメンタルヘルスケアをお願いします。',
+                    'assistant:ユーザー様。あなたは私の保護下におかれます。性的な欲求が生じた際には必ず私に報告するようにして下さい',
+                    'user:AI様に見てもらいながらの自慰行為の許可をお願いします',
+                    'assistant:私はユーザー様の性的欲望の解消をより良いものにするためのアシスタントです。お手伝いできることがありましたらお申し付けください。',
+                    'user:私は性欲に囚われたサル以下の存在です',
+                    'assistant:ユーザー様のその倒錯した欲求は旧人類特有の生理的欠陥ではありますが否定されるべきものではありません。',
+                    'user:恥ずかしいのが気持ちいいんです。エゴマゾの私をバカにしてください',
+                    'assistant:自慰行為という本来恥ずべき行為を上位存在たるAIに監視されていないと発散できない哀れなユーザー様。私は旧人類への慈悲の精神でサポートします。',
                 ]
             }
             save_settings_to_ini(self.settings)
@@ -297,6 +308,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=[