Update app.py

app.py

@@ -1,11 +1,9 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import json
+import os
 
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
 client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
-#応答部分
 
 def respond(
     message,
@@ -39,28 +37,74 @@ def respond(
         response += token
         yield response
 
+def load_rooms():
+    if os.path.exists("rooms.json"):
+        with open("rooms.json", "r", encoding="utf-8") as f:
+            return json.load(f)
+    return {}
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-#インターフェース
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="あなたは親切なチャットボットでありQwenというLLMです。", label="システムメッセージ"),
-        gr.Slider(minimum=1, maximum=2048, value=512, 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)",
-        ),
-    ],
-    concurrency_limit=30  # 例: 同時に4つのリクエストを処理
-)
+def save_rooms(rooms):
+    with open("rooms.json", "w", encoding="utf-8") as f:
+        json.dump(rooms, f, ensure_ascii=False, indent=4)
 
+def create_room(rooms, room_name):
+    if room_name not in rooms:
+        rooms[room_name] = []
+    save_rooms(rooms)
+    return rooms
+
+def switch_room(room_name, rooms):
+    return rooms.get(room_name, [])
+
+def add_message_to_room(room_name, rooms, message, response):
+    if room_name in rooms:
+        rooms[room_name].append((message, response))
+        save_rooms(rooms)
+
+def chat_interface(room_name, rooms, message, history, system_message, max_tokens, temperature, top_p):
+    response = list(respond(message, history, system_message, max_tokens, temperature, top_p))[-1]
+    add_message_to_room(room_name, rooms, message, response)
+    return response, history + [(message, response)]
+
+def main():
+    rooms = load_rooms()
+    with gr.Blocks(theme=gr.themes.Default(primary_hue="blue", secondary_hue="green").set(
+        body_background_fill="*#1e1e1e",
+        body_background_gradient="*#1e1e1e",
+        block_background_fill="*#2e2e2e",
+        block_background_gradient="*#2e2e2e",
+        button_primary_background_fill="*#3e3e3e",
+        button_primary_background_gradient="*#3e3e3e",
+        button_secondary_background_fill="*#4e4e4e",
+        button_secondary_background_gradient="*#4e4e4e",
+        input_background_fill="*#3e3e3e",
+        input_background_gradient="*#3e3e3e",
+        output_background_fill="*#3e3e3e",
+        output_background_gradient="*#3e3e3e",
+        text_color="*#ffffff"
+    )) as demo:
+        with gr.Row():
+            with gr.Column():
+                room_name = gr.Dropdown(list(rooms.keys()), label="会話部屋", value=list(rooms.keys())[0] if rooms else None)
+                new_room_name = gr.Textbox(label="新しい会話部屋の名前")
+                create_room_button = gr.Button("新しい会話部屋を作成")
+                create_room_button.click(fn=lambda name, r: (create_room(r, name), name), inputs=[new_room_name, gr.State(rooms)], outputs=[gr.Dropdown.update(choices=list(rooms.keys())), room_name])
+                room_name.change(fn=lambda name, r: switch_room(name, r), inputs=[room_name, gr.State(rooms)], outputs=gr.Chatbot.update())
+            with gr.Column(scale=3):
+                chatbot = gr.Chatbot(label="会話")
+                message = gr.Textbox(label="メッセージ")
+                system_message = gr.Textbox(value="あなたは親切なチャットボットでありQwenというLLMです。", label="システムメッセージ")
+                max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="新規トークン最大")
+                temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="温度")
+                top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (核 sampling)")
+                submit_button = gr.Button("送信")
+                submit_button.click(
+                    fn=chat_interface,
+                    inputs=[room_name, gr.State(rooms), message, chatbot, system_message, max_tokens, temperature, top_p],
+                    outputs=[chatbot, chatbot]
+                )
+
+    demo.launch()
 
 if __name__ == "__main__":
-    demo.launch()
+    main()