Update app.py

app.py

@@ -2,8 +2,9 @@ import requests
 import os
 import json
 import streamlit as st
-from datetime import datetime
+from datetime import datetime, timedelta
 import time
+import uuid
 
 # Page configuration
 st.set_page_config(
@@ -43,6 +44,8 @@ st.markdown("""
 
 # File to store chat history
 HISTORY_FILE = "chat_history.json"
+# NEW: File to store online users
+USERS_FILE = "online_users.json"
 
 def load_chat_history():
     """Load chat history from file"""
@@ -71,6 +74,66 @@ def clear_chat_history():
     except Exception as e:
         st.error(f"Error clearing chat history: {e}")
 
+# NEW: User tracking functions
+def get_user_id():
+    """Get unique ID for this user session"""
+    if 'user_id' not in st.session_state:
+        st.session_state.user_id = str(uuid.uuid4())[:8]  # Short ID for family use
+    return st.session_state.user_id
+
+def update_online_users():
+    """Update that this user is online right now"""
+    try:
+        # Load current online users
+        users = {}
+        if os.path.exists(USERS_FILE):
+            with open(USERS_FILE, 'r') as f:
+                users = json.load(f)
+        
+        # Add/update this user
+        user_id = get_user_id()
+        users[user_id] = {
+            'last_seen': datetime.now().isoformat(),
+            'name': f'User-{user_id}'  # You can customize this
+        }
+        
+        # Remove users not seen in last 5 minutes
+        current_time = datetime.now()
+        active_users = {}
+        for uid, data in users.items():
+            last_seen = datetime.fromisoformat(data['last_seen'])
+            if current_time - last_seen < timedelta(minutes=5):
+                active_users[uid] = data
+        
+        # Save updated list
+        with open(USERS_FILE, 'w') as f:
+            json.dump(active_users, f, indent=2)
+        
+        return len(active_users)
+    except Exception:
+        return 1  # If error, assume at least you're online
+
+def get_online_count():
+    """Get number of people currently online"""
+    try:
+        if not os.path.exists(USERS_FILE):
+            return 0
+        
+        with open(USERS_FILE, 'r') as f:
+            users = json.load(f)
+        
+        # Check who's still active (last 5 minutes)
+        current_time = datetime.now()
+        active_count = 0
+        for data in users.values():
+            last_seen = datetime.fromisoformat(data['last_seen'])
+            if current_time - last_seen < timedelta(minutes=5):
+                active_count += 1
+        
+        return active_count
+    except Exception:
+        return 0
+
 # Initialize session state with saved history
 if "messages" not in st.session_state:
     st.session_state.messages = load_chat_history()
@@ -90,7 +153,6 @@ def check_api_status():
     except:
         return "Error"
 
-
 def get_ai_response(messages, model="openai/gpt-3.5-turbo"):
     if not OPENROUTER_API_KEY:
         return "No API key found. Please add OPENROUTER_API_KEY to environment variables."
@@ -183,6 +245,28 @@ with st.sidebar:
     
     st.divider()
     
+    # NEW: Live Users Section
+    st.header("👥 Who's Online")
+    
+    # Update that you're online
+    online_count = update_online_users()
+    
+    # Show live count
+    if online_count == 1:
+        st.info("🟢 Just you online")
+    else:
+        st.success(f"🟢 {online_count} people online")
+    
+    # Show your session
+    your_id = get_user_id()
+    st.caption(f"You: User-{your_id}")
+    
+    # Quick refresh button
+    if st.button("Refresh", use_container_width=True):
+        st.rerun()
+    
+    st.divider()
+    
     # All models including new ones
     models = [
         ("GPT-3.5 Turbo", "openai/gpt-3.5-turbo"),
@@ -286,6 +370,9 @@ for message in st.session_state.messages:
 
 # Chat input
 if prompt := st.chat_input("Ask anything..."):
+    # NEW: Update online status when user sends message
+    update_online_users()
+    
     # Add user message
     user_message = {"role": "user", "content": prompt}
     st.session_state.messages.append(user_message)