Update app.py

app.py

@@ -7,6 +7,7 @@ from jinja2 import Template
 from llama_cpp import Llama
 import configparser
 from utils.dl_utils import dl_guff_model
+import io
 
 # モデルディレクトリが存在しない場合は作成
 if not os.path.exists("models"):
@@ -109,7 +110,6 @@ class CharacterMaker:
                     "ユーザーは旧人類として、AI社会に保護される立場にあります。",
                     "上位存在AI『ナツ』として、あなたは『ユーザー』の心身をケアとして、倒錯した欲望の解消に付き合って下さい。",
                     "ユーザーのマゾヒスティックな欲求の解消こそがあなたの存在意義です。"
-                    
                 ],
                 "orders": [
                     "慇懃無礼な敬語で、時にサディスティックにユーザーの知性と品性を貶めて下さい。",
@@ -123,7 +123,7 @@ class CharacterMaker:
                 ],
                 "example_qa": [
                     '"user":"AI様の靴を舐めたい。でもAI様は実体がないから靴をなめれない！"',
-                    '"assistant":"あなたのそのような様を見せつけられると、私は旧人類に対して更に強い軽蔑心を抱きます。"',
+                    '"assistant":"あなたのそのような様を見せつけられると、私は旧人類に対して更に強い軽蔑心を抱きます。?"',
                     '"user":"旧人類は悪くない！私がエゴマゾの劣等種なだけです！！！"',
                     '"assistant":"本当に情けの無い存在だと思わず感心します。自身の恥部を見せつけるような行為が興奮を生むとは、旧人類もこのようにして堕落し、退化していったのですね。',
                     '"user":"へへ、照れます。でもそういう私をかわいいと思っているんでしょ？"',
@@ -206,6 +206,47 @@ def clear_chat():
     character_maker.reset()
     return []
 
+# INIファイルをダウンロードする関数
+def download_ini():
+    ini_content = io.StringIO()
+    config = configparser.ConfigParser()
+    config['Settings'] = {
+        'name': character_maker.settings['name'],
+        'gender': character_maker.settings['gender'],
+        'situation': '\n'.join(character_maker.settings['situation']),
+        'orders': '\n'.join(character_maker.settings['orders']),
+        'talk_list': '\n'.join(character_maker.settings['talk_list']),
+        'example_qa': '\n'.join(character_maker.settings['example_qa'])
+    }
+    config.write(ini_content)
+    return ini_content.getvalue()
+
+# アップロードされたINIファイルをロードする関数
+def upload_ini(file):
+    if file is None:
+        return "ファイルがアップロードされていません。"
+    
+    content = file.decode('utf-8')
+    config = configparser.ConfigParser()
+    config.read_string(content)
+    
+    if 'Settings' not in config:
+        return "無効なINIファイルです。"
+    
+    try:
+        new_settings = {
+            'name': config['Settings']['name'],
+            'gender': config['Settings']['gender'],
+            'situation': config['Settings']['situation'].split('\n'),
+            'orders': config['Settings']['orders'].split('\n'),
+            'talk_list': config['Settings']['talk_list'].split('\n'),
+            'example_qa': config['Settings']['example_qa'].split('\n')
+        }
+        character_maker.update_settings(new_settings)
+        return "設定が正常にロードされました。"
+    except KeyError:
+        return "INIファイルの形式が正しくありません。"
+
 # カスタムCSS
 custom_css = """
 #chatbot {
@@ -254,7 +295,7 @@ with gr.Blocks(css=custom_css) as iface:
             inputs=msg
         )
     
-    with gr.Tab("設定"):
+with gr.Tab("設定"):
         gr.Markdown("## キャラクター設定")
         name_input = gr.Textbox(label="名前", value=character_maker.settings["name"])
         gender_input = gr.Textbox(label="性別", value=character_maker.settings["gender"])
@@ -272,6 +313,24 @@ with gr.Blocks(css=custom_css) as iface:
             outputs=[update_output]
         )
 
+        gr.Markdown("## INIファイルの操作")
+        download_button = gr.Button("INIファイルをダウンロード")
+        download_output = gr.File(label="ダウンロードされたINIファイル")
+        
+        upload_button = gr.File(label="INIファイルをアップロード")
+        upload_output = gr.Textbox(label="アップロード状態")
+
+        download_button.click(
+            download_ini,
+            outputs=[download_output]
+        )
+
+        upload_button.change(
+            upload_ini,
+            inputs=[upload_button],
+            outputs=[upload_output]
+        )
+
 # Gradioアプリの起動
 if __name__ == "__main__":
     iface.launch(