s

app.py

@@ -11,7 +11,7 @@ def create_character(name, age, gender, personality, backstory):
         "personality": personality,
         "backstory": backstory,
     }
-    return f"Character {name} created successfully!"
+    return f"Character {name} created successfully!", list(characters.keys())
 
 # Function to handle personality-based responses
 def respond_with_personality(character_name, message, history):
@@ -36,31 +36,6 @@ def respond_with_personality(character_name, message, history):
     history.append((message, response))
     return history, response
 
-# Function to get available characters for switching
-def available_characters():
-    return list(characters.keys())
-
-# Gradio interface to create characters
-def create_character_interface():
-    with gr.Row():
-        name_input = gr.Textbox(label="Character Name", placeholder="Enter character name")
-        age_input = gr.Number(label="Character Age", value=25, step=1)
-        gender_input = gr.Dropdown(choices=["Male", "Female", "Other"], label="Character Gender")
-        personality_input = gr.Dropdown(
-            choices=["Heroic", "Sarcastic", "Optimistic", "Empathetic"],
-            label="Character Personality"
-        )
-        backstory_input = gr.Textbox(label="Character Backstory", placeholder="Enter backstory")
-
-        create_button = gr.Button("Create Character")
-        create_output = gr.Textbox(label="Output", interactive=False)
-
-    create_button.click(create_character, 
-                        inputs=[name_input, age_input, gender_input, personality_input, backstory_input], 
-                        outputs=[create_output])
-
-    return name_input, age_input, gender_input, personality_input, backstory_input, create_button, create_output
-
 # Main chat function that handles conversation with multiple characters
 def chat(character_name, message, history):
     if character_name not in characters:
@@ -77,11 +52,32 @@ def build_interface():
 
         # Character creation UI
         with gr.Tab("Create Character"):
-            create_character_interface()
+            name_input = gr.Textbox(label="Character Name", placeholder="Enter character name")
+            age_input = gr.Number(label="Character Age", value=25, step=1)
+            gender_input = gr.Dropdown(choices=["Male", "Female", "Other"], label="Character Gender")
+            personality_input = gr.Dropdown(
+                choices=["Heroic", "Sarcastic", "Optimistic", "Empathetic"],
+                label="Character Personality"
+            )
+            backstory_input = gr.Textbox(label="Character Backstory", placeholder="Enter backstory")
+
+            create_button = gr.Button("Create Character")
+            create_output = gr.Textbox(label="Output", interactive=False)
+            character_selector = gr.Dropdown(choices=[], label="Select Character")
+
+            # Update the dropdown after character creation
+            def create_and_update(name, age, gender, personality, backstory):
+                output_message, updated_character_list = create_character(name, age, gender, personality, backstory)
+                return output_message, gr.update(choices=updated_character_list)
+
+            create_button.click(
+                create_and_update, 
+                inputs=[name_input, age_input, gender_input, personality_input, backstory_input], 
+                outputs=[create_output, character_selector]
+            )
 
         # Chat UI
         with gr.Tab("Chat"):
-            character_selector = gr.Dropdown(choices=available_characters(), label="Select Character")
             chat_box = gr.Chatbot(label="Chat History")
             message_input = gr.Textbox(label="Your Message", placeholder="Type a message...")