Update app.py

app.py

@@ -85,7 +85,6 @@ async def websocket_handler(websocket, path):
 
         async for message in websocket:
             try:
-                # Parse message as simple text (no JSON)
                 parts = message.split('|', 1)
                 if len(parts) == 2:
                     username, content = parts
@@ -136,7 +135,12 @@ def create_gradio_interface(initial_username):
             with gr.Column(scale=3):
                 chat_display = gr.Code(label="Chat History", language="markdown", lines=15, elem_classes=["chat-box"])
             with gr.Column(scale=1):
-                user_list = gr.Dropdown(label="Switch User", choices=get_user_list(chat_state.value), value=initial_username)
+                user_list = gr.Dropdown(
+                    label="Switch User",
+                    choices=get_user_list(chat_state.value),
+                    value=initial_username,
+                    allow_custom_value=True  # Let the initial username stick! 🎯
+                )
 
         with gr.Row():
             username_input = gr.Textbox(label="Your Name", value=initial_username)
@@ -169,8 +173,7 @@ def create_gradio_interface(initial_username):
         send_button.click(
             fn=update_chat,
             inputs=[username_input, message_input, chat_state],
-            outputs=[chat_display, user_list],
-            _js="() => ['', document.getElementById('message_input').value, document.getElementById('chat_display').value]"
+            outputs=[chat_display, user_list]
         )
         message_input.submit(
             fn=update_chat,