Update app.py

app.py

@@ -571,6 +571,9 @@ Example: "Here's the setup: [IMAGE: double slit experiment diagram]"
 # -----------------------------------------------------------------------------
 # 6. SIDEBAR CONFIGURATION
 # -----------------------------------------------------------------------------
+# -----------------------------------------------------------------------------
+# 6. SIDEBAR CONFIGURATION
+# -----------------------------------------------------------------------------
 with st.sidebar:
     # Header
     col1, col2 = st.columns([1, 3])
@@ -606,7 +609,7 @@ with st.sidebar:
     
     enable_voice = st.toggle("🗣️ Text-to-Speech", value=False)
     
-    #st.divider()
+    st.divider()
     
     # Image Analysis Settings
     st.header("🖼️ Image Analysis")
@@ -618,39 +621,46 @@ with st.sidebar:
         help="Choose image analysis service"
     )
     
-    # Initialize variables
+    # Initialize variables - FIXED SECRETS CHECKING
     hf_token = None
     together_key = None
     openrouter_key = None
     
+    # Safe secrets checking function
+    def get_secret(key, default=None):
+        """Safely get a secret from st.secrets or environment."""
+        try:
+            # First try st.secrets (Hugging Face Spaces)
+            if hasattr(st, 'secrets'):
+                # Check if secrets is loaded and has the key
+                secrets_dict = dict(st.secrets)
+                if key in secrets_dict:
+                    return secrets_dict[key]
+        except:
+            pass
+        
+        # Fallback to environment variables
+        return os.environ.get(key, default)
+    
     if image_analysis_mode == "Hugging Face":
-        if hasattr(st, 'secrets') and "HF_TOKEN" in st.secrets:
-            hf_token = st.secrets["HF_TOKEN"]
-            st.success("✓ Hugging Face Token configured")
-        elif "HF_TOKEN" in os.environ:
-            hf_token = os.environ["HF_TOKEN"]
+        hf_token = get_secret("HF_TOKEN")
+        if hf_token:
             st.success("✓ Hugging Face Token configured")
         else:
             st.warning("Add HF_TOKEN to secrets")
             st.info("Get token: huggingface.co/settings/tokens")
     
     elif image_analysis_mode == "Together.ai":
-        if hasattr(st, 'secrets') and "TOGETHER_API_KEY" in st.secrets:
-            together_key = st.secrets["TOGETHER_API_KEY"]
-            st.success("✓ Together.ai API configured")
-        elif "TOGETHER_API_KEY" in os.environ:
-            together_key = os.environ["TOGETHER_API_KEY"]
+        together_key = get_secret("TOGETHER_API_KEY")
+        if together_key:
             st.success("✓ Together.ai API configured")
         else:
             st.warning("Add TOGETHER_API_KEY to secrets")
             st.info("Get key: together.ai (free credits available)")
     
     elif image_analysis_mode == "OpenRouter":
-        if hasattr(st, 'secrets') and "OPENROUTER_API_KEY" in st.secrets:
-            openrouter_key = st.secrets["OPENROUTER_API_KEY"]
-            st.success("✓ OpenRouter API configured")
-        elif "OPENROUTER_API_KEY" in os.environ:
-            openrouter_key = os.environ["OPENROUTER_API_KEY"]
+        openrouter_key = get_secret("OPENROUTER_API_KEY")
+        if openrouter_key:
             st.success("✓ OpenRouter API configured")
         else:
             st.warning("Add OPENROUTER_API_KEY to secrets")