Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 import asyncio
 import websockets
 import uuid
@@ -29,7 +29,7 @@ def get_node_name():
     """🎲 Spins the wheel of fate to name our node - a random alias or user pick! 🏷️"""
     parser = argparse.ArgumentParser(description='Start a chat node with a specific name')
     parser.add_argument('--node-name', type=str, default=None, help='Name for this chat node')
-    parser.add_argument('--port', type=int, default=7860, help='Port to run the Gradio interface on')
+    parser.add_argument('--port', type=int, default=8501, help='Port to run the Streamlit interface on')  # Default Streamlit port
     args = parser.parse_args()
     return args.node_name or f"node-{uuid.uuid4().hex[:8]}", args.port
 
@@ -122,77 +122,49 @@ async def start_websocket_server(host='0.0.0.0', port=8765):
     return server
 
 # Chat interface maker - the stage builder for our chat extravaganza! 🎭🏟️
-def create_gradio_interface(initial_username):
+def create_streamlit_interface(initial_username):
     """🖌️ Sets up the chat stage with live updates and name-switching flair! 🌟"""
-    with gr.Blocks(title=f"Chat Node: {NODE_NAME}", css=".chat-box { font-family: monospace; background: #1e1e1e; color: #d4d4d4; padding: 10px; border-radius: 5px; }") as demo:
-        # Shared state - the magic memory box all clients peek into! 🪄📦
-        chat_state = gr.State(value=load_chat())
-
-        gr.Markdown(f"# Chat Node: {NODE_NAME}")
-        gr.Markdown("Chat live, switch names, and keep the party going! 🎉")
-
-        with gr.Row():
-            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,
-                    allow_custom_value=True  # Let the initial username stick! 🎯
-                )
-
-        with gr.Row():
-            username_input = gr.Textbox(label="Your Name", value=initial_username)
-            message_input = gr.Textbox(label="Message", placeholder="Type your epic line here! ✍️")
-            send_button = gr.Button("Send 🚀")
-
-        # Chat updater - the wizard keeping the chat spell alive! 🧙‍♂️✨
-        def update_chat(username, message, current_chat):
-            """🎩 Conjures a fresh chat view with your latest quip - abracadabra! 🎇"""
-            if message.strip():
-                save_chat_entry(username, message)
-            new_chat = load_chat()
-            return new_chat, get_user_list(new_chat)
-
-        # Name switcher - the shapeshifter swapping your chat persona! 🦸‍♂️🎭
-        def switch_user(new_username, current_chat):
-            """🦄 Transforms you into a new chat hero - new name, same game! 🏆"""
-            return new_username, current_chat, get_user_list(current_chat)
-
-        # Live chat persistence - the timekeeper syncing the chat clock! ⏰🔄
-        async def persist_chat():
-            """⏳ Keeps the chat tome in sync, scribbling updates every few ticks! 📝"""
-            while True:
-                await asyncio.sleep(5)  # Persist every 5 seconds
-                with open(CHAT_FILE, 'r') as f:
-                    current = f.read()
-                chat_state.value = current
-
-        # Hook up the magic! 🎣
-        send_button.click(
-            fn=update_chat,
-            inputs=[username_input, message_input, chat_state],
-            outputs=[chat_display, user_list]
-        )
-        message_input.submit(
-            fn=update_chat,
-            inputs=[username_input, message_input, chat_state],
-            outputs=[chat_display, user_list]
-        )
-        user_list.change(
-            fn=switch_user,
-            inputs=[user_list, chat_state],
-            outputs=[username_input, chat_display, user_list]
-        )
-
-        # Start with the latest chat vibes! 🎵
-        demo.load(lambda: (load_chat(), get_user_list(load_chat())), None, [chat_display, user_list])
-
-        # Kick off the persistence dance! 💃
-        asyncio.create_task(persist_chat())
-
-    return demo
+    # Custom CSS for a sleek chat box
+    st.markdown("""
+        <style>
+        .chat-box {
+            font-family: monospace;
+            background: #1e1e1e;
+            color: #d4d4d4;
+            padding: 10px;
+            border-radius: 5px;
+            height: 400px;
+            overflow-y: auto;
+        }
+        </style>
+    """, unsafe_allow_html=True)
+
+    # Title and intro
+    st.title(f"Chat Node: {NODE_NAME}")
+    st.markdown("Chat live, switch names, and keep the party going! 🎉")
+
+    # Session state to persist username
+    if 'username' not in st.session_state:
+        st.session_state.username = initial_username
+
+    # Chat display
+    chat_content = load_chat()
+    st.markdown(f"<div class='chat-box'>{chat_content}</div>", unsafe_allow_html=True)
+
+    # User switcher
+    user_list = get_user_list(chat_content)
+    new_username = st.selectbox("Switch User", user_list + [st.session_state.username], index=len(user_list))
+    if new_username != st.session_state.username:
+        st.session_state.username = new_username
+
+    # Message input and send
+    message = st.text_input("Message", placeholder="Type your epic line here! ✍️")
+    if st.button("Send 🚀") and message.strip():
+        save_chat_entry(st.session_state.username, message)
+        st.rerun()  # Trigger a rerun to refresh the chat
+
+    # Auto-refresh every second for live updates
+    st.markdown("<script>window.setTimeout(() => window.location.reload(), 1000);</script>", unsafe_allow_html=True)
 
 # Main event - the ringmaster kicking off the chat circus! 🎪🤡
 async def main():
@@ -202,27 +174,11 @@ async def main():
     await start_websocket_server()
 
     # Grab username from URL or roll the dice! 🎲
-    initial_username = os.environ.get("QUERY_STRING", "").split("username=")[-1] if "username=" in os.environ.get("QUERY_STRING", "") else random.choice(FUN_USERNAMES)
+    query_params = st.query_params if hasattr(st, 'query_params') else {}  # Streamlit 1.20+ syntax
+    initial_username = query_params.get("username", random.choice(FUN_USERNAMES)) if query_params else random.choice(FUN_USERNAMES)
     print(f"Welcoming {initial_username} to the chat bash!")
 
-    interface = create_gradio_interface(initial_username)
-    
-    # URL param magic - the gatekeeper checking your VIP pass! 🎟️
-    from starlette.middleware.base import BaseHTTPMiddleware
-    class UsernameMiddleware(BaseHTTPMiddleware):
-        async def dispatch(self, request, call_next):
-            """🔑 Snags your name from the URL VIP list - exclusive entry! 🎩"""
-            query_params = dict(request.query_params)
-            if "username" in query_params and query_params["username"] in FUN_USERNAMES:
-                global initial_username
-                initial_username = query_params["username"]
-            return await call_next(request)
-
-    app = gr.routes.App.create_app(interface)
-    app.add_middleware(UsernameMiddleware)
-    
-    import uvicorn
-    await uvicorn.Server(uvicorn.Config(app, host="0.0.0.0", port=port)).serve()
+    create_streamlit_interface(initial_username)
 
 if __name__ == "__main__":
     asyncio.run(main())