Update src/streamlit_app.py

src/streamlit_app.py

@@ -34,7 +34,7 @@ st.set_page_config(
 # ==========================================
 
 # 1. Bot Detector: The gold standard for catching GPT-written text
-MODEL_FAKE = "fakespot-ai/roberta-base-ai-text-detection-v1"
+MODEL_FAKE = "openai-community/roberta-base-openai-detector"
 
 # 2. Mood Scanner: Checks detailed sentiment
 MODEL_MOOD = "cardiffnlp/twitter-roberta-base-sentiment-latest"
@@ -43,21 +43,33 @@ MODEL_MOOD = "cardiffnlp/twitter-roberta-base-sentiment-latest"
 MODEL_GRAMMAR = "textattack/roberta-base-CoLA"
 
 # 4. Image Checker A: High Precision (Strict)
-MODEL_IMG_A = "dima806/ai_vs_real_image_detection"
+MODEL_IMG_A = "dima806/ai_generated_image_detection"
 
 # 5. Image Checker B: High Reliability (Broad)
 MODEL_IMG_B = "umm-maybe/AI-image-detector"
 
 # ==========================================
 
-# --- Secrets Management ---
-def get_token():
-    if key := os.environ.get("HF_TOKEN"): return key
-    if hasattr(st, "secrets") and "HF_TOKEN" in st.secrets: return st.secrets["HF_TOKEN"]
+# --- Robust Secrets Management ---
+def get_secret(key):
+    """Safe retrieval of secrets from Env Vars OR Streamlit Secrets"""
+    # 1. Check Environment Variables (Preferred for HF Spaces)
+    if key in os.environ:
+        return os.environ[key]
+    
+    # 2. Check Streamlit Secrets (Preferred for Local Dev)
+    # We wrap this in try/except because accessing st.secrets throws an error
+    # if the secrets.toml file does not exist.
+    try:
+        if key in st.secrets:
+            return st.secrets[key]
+    except Exception:
+        pass
+        
     return None
 
-HF_TOKEN = get_token()
-OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
+HF_TOKEN = get_secret("HF_TOKEN")
+OPENAI_API_KEY = get_secret("OPENAI_API_KEY")
 
 # --- Custom CSS ---
 def inject_custom_css():