Update app.py

app.py

@@ -68,28 +68,31 @@ def chat_interface(room_name, rooms, message, history, system_message, max_token
 
 def main():
     rooms = load_rooms()
+    room_names = list(rooms.keys())
     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)
+                room_name = gr.Dropdown(room_names, label="会話部屋", value=room_names[0] if room_names 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())
+                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(create_room(r, name).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="メッセージ")