Update app.py

app.py

@@ -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=8501, help='Port to run the Streamlit interface on')  # Default Streamlit port
+    parser.add_argument('--port', type=int, default=8501, help='Port to run the Streamlit interface on')
     args = parser.parse_args()
     return args.node_name or f"node-{uuid.uuid4().hex[:8]}", args.port
 
@@ -123,8 +123,8 @@ async def start_websocket_server(host='0.0.0.0', port=8765):
 
 # Chat interface maker - the stage builder for our chat extravaganza! 🎭🏟️
 def create_streamlit_interface(initial_username):
-    """🖌️ Sets up the chat stage with live updates and name-switching flair! 🌟"""
-    # Custom CSS for a sleek chat box
+    """🖌️ Sets up the chat stage with live updates, timers, and refresh rate flair! 🌟"""
+    # Custom CSS for a sleek chat box and timer
     st.markdown("""
         <style>
         .chat-box {
@@ -136,16 +136,25 @@ def create_streamlit_interface(initial_username):
             height: 400px;
             overflow-y: auto;
         }
+        .timer {
+            font-size: 18px;
+            color: #ffcc00;
+            text-align: center;
+        }
         </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! 🎉")
+    st.markdown("Chat live, switch names, and tweak the refresh rate! 🎉")
 
-    # Session state to persist username
+    # Session state for username and refresh rate
     if 'username' not in st.session_state:
         st.session_state.username = initial_username
+    if 'refresh_rate' not in st.session_state:
+        st.session_state.refresh_rate = 5  # Default 5 seconds
+    if 'last_refresh' not in st.session_state:
+        st.session_state.last_refresh = time.time()
 
     # Chat display
     chat_content = load_chat()
@@ -161,10 +170,49 @@ def create_streamlit_interface(initial_username):
     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)
+        st.session_state.last_refresh = time.time()  # Reset timer on send
+        st.rerun()
+
+    # Refresh rate controls
+    st.subheader("Set Refresh Rate ⏳")
+    refresh_rate = st.slider(
+        "Refresh Rate (seconds)",
+        min_value=1,
+        max_value=300,  # 5 minutes = 300 seconds
+        value=st.session_state.refresh_rate,
+        step=1
+    )
+    st.session_state.refresh_rate = refresh_rate
+
+    # Emoji buttons for quick settings
+    col1, col2, col3 = st.columns(3)
+    with col1:
+        if st.button("🐇 Small (1s)"):
+            st.session_state.refresh_rate = 1
+            st.session_state.last_refresh = time.time()
+            st.rerun()
+    with col2:
+        if st.button("🐢 Medium (5s)"):
+            st.session_state.refresh_rate = 5
+            st.session_state.last_refresh = time.time()
+            st.rerun()
+    with col3:
+        if st.button("🐘 Large (5m)"):
+            st.session_state.refresh_rate = 300
+            st.session_state.last_refresh = time.time()
+            st.rerun()
+
+    # Countdown timer
+    elapsed = time.time() - st.session_state.last_refresh
+    remaining = max(0, st.session_state.refresh_rate - int(elapsed))
+    st.markdown(f"<p class='timer'>Next refresh in: {remaining} seconds</p>", unsafe_allow_html=True)
+
+    # Auto-refresh logic
+    if elapsed >= st.session_state.refresh_rate:
+        st.session_state.last_refresh = time.time()
+        st.rerun()
+    else:
+        st.markdown(f"<script>window.setTimeout(() => window.location.reload(), {int((st.session_state.refresh_rate - elapsed) * 1000)});</script>", unsafe_allow_html=True)
 
 # Main event - the ringmaster kicking off the chat circus! 🎪🤡
 async def main():
@@ -174,7 +222,7 @@ async def main():
     await start_websocket_server()
 
     # Grab username from URL or roll the dice! 🎲
-    query_params = st.query_params if hasattr(st, 'query_params') else {}  # Streamlit 1.20+ syntax
+    query_params = st.query_params if hasattr(st, 'query_params') else {}
     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!")