Update src/app.py

src/app.py

@@ -69,7 +69,6 @@ with st.sidebar:
         st.header("🛡️ Admin Tools")
         admin_metric_placeholder = st.empty()
         
-        # FIX: Point to the correct persistence path
         log_path = tracker.get_log_path()
         if log_path.exists():
             with open(log_path, "r") as f:
@@ -101,30 +100,40 @@ with st.sidebar:
     model_options = list(model_map.keys())
     model_captions = ["Slower for now, but free and private" for _ in model_options]
 
-    if "admin" in st.session_state.roles:
+    # 2. CHECK FOR GPT-4o ACCESS (Admin OR User Key)
+    # We moved the input UP so the user can unlock the option immediately
+    
+    # Check if user is admin
+    is_admin = "admin" in st.session_state.roles
+    
+    # Input for Non-Admins
+    user_api_key = None
+    if not is_admin:
+        user_api_key = st.text_input(
+            "🔓 Unlock GPT-4o (Enter API Key)", 
+            type="password", 
+            help="Enter your OpenAI API key to access GPT-4o. Press Enter to apply."
+        )
+        if user_api_key:
+            st.session_state.user_openai_key = user_api_key
+            st.caption("✅ Key Active")
+        else:
+            st.session_state.user_openai_key = None
+
+    # 3. DYNAMICALLY ADD GPT-4o TO THE LIST
+    # If Admin OR if they just entered a key, show the option
+    if is_admin or st.session_state.get("user_openai_key"):
         model_options.append("GPT-4o (Omni)")
         model_captions.append("Fast, smart, sends data to OpenAI")
 
+    # 4. RENDER THE SELECTOR
     model_choice = st.radio(
         "Choose your Intelligence:",
         model_options,
         captions=model_captions,
-        key="model_selector_radio" # Unique Key Added for Safety
+        key="model_selector_radio" 
     )
     st.info(f"Connected to: **{model_choice}**")
-
-    # --- REGULAR USER OPENAI API KEY ---
-    user_api_key = st.text_input(
-        "OpenAI API Key (Optional)", 
-        type="password", 
-        help="Enter your own key to bypass limits. Press Enter to apply."
-    )
-    
-    if user_api_key:
-        st.session_state.user_openai_key = user_api_key
-        st.caption("✅ Key Active")  # Visual confirmation
-    else:
-        st.session_state.user_openai_key = None
     
     st.divider()
     st.header("⚙️ Controls")