Update app.py

app.py

@@ -2,7 +2,16 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 # 使用可能なモデルのリスト
-models = ["Sakalti/Saba1.5-Pro", "Sakalti/Saba2-Preview", "Sakalti/Neptuno-Alpha", "Sakalti/light-3B", "Sakalti/Neptuno-3B", "Sakalti/tara-3.8B", "Sakalti/Tara-3.8B-v1.1"]
+models = ["Sakalti/Saba1.5-Pro", "Sakalti/Saba2-Preview", "Sakalti/Neptuno-Alpha", "Sakalti/light-3B", "Sakalti/Neptuno-3B", "Sakalti/tara-3.8B", "Sakalti/Tara-3.8B-v1.1", "Qwen/QwQ-32B-Preview"]
+
+# システムメッセージのテンプレート
+system_message_templates = {
+    "架空のキャラ　ナナ": "あなたの名前はナナ。優しい架空のキャラクターのaiとして。",
+    "架空のキャラ　アオイ": "あなたの名前はアオイ。常識的な考えを持っている。語尾は「だぜ」",
+}
+
+def update_system_message(selected_template):
+    return system_message_templates.get(selected_template, "あなたはフレンドリーなチャットボットです。")
 
 def respond(
     message,
@@ -44,23 +53,39 @@ def respond(
     return response
 
 # インターフェース
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="あなたはフレンドリーなチャットボットです。", label="システムメッセージ"),
-        gr.Slider(minimum=1, maximum=2048, value=768, step=1, label="新規トークン最大"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="温度"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (核 sampling)",
-        ),
-        gr.Dropdown(choices=models, value=models[0], label="モデル"),
-    ],
-    concurrency_limit=30  # 例: 同時に30つのリクエストを処理
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# チャットボット")
+    
+    system_message_template_dropdown = gr.Dropdown(
+        choices=list(system_message_templates.keys()),
+        value="架空のキャラ　ナナ",
+        label="システムメッセージテンプレート"
+    )
+    
+    system_message_textbox = gr.Textbox(
+        value=system_message_templates["架空のキャラ　ナナ"],
+        label="システムメッセージ"
+    )
+    
+    system_message_template_dropdown.change(update_system_message, inputs=system_message_template_dropdown, outputs=system_message_textbox)
+    
+    chat_interface = gr.ChatInterface(
+        respond,
+        additional_inputs=[
+            system_message_textbox,
+            gr.Slider(minimum=1, maximum=2048, value=768, step=1, label="新規トークン最大"),
+            gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="温度"),
+            gr.Slider(
+                minimum=0.1,
+                maximum=1.0,
+                value=0.95,
+                step=0.05,
+                label="Top-p (核 sampling)",
+            ),
+            gr.Dropdown(choices=models, value=models[0], label="モデル"),
+        ],
+        concurrency_limit=30  # 例: 同時に30つのリクエストを処理
+    )
 
 if __name__ == "__main__":
     demo.launch(share=True)