Update app.py

app.py

@@ -240,60 +240,11 @@ def save_victim_info(victim: dict):
 # -----------------------
 # OpenAI integration
 # -----------------------
-DEFAULT_MODEL = os.environ.get("OPENAI_MODEL", "gpt-4o-mini")
-
-# Initialize session state for temporary UI key
-if "ui_api_key" not in st.session_state:
-    st.session_state.ui_api_key = "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"
+OPENAI_api_key = st.secrets["OPENAI_API_KEY"]
+OPENAI_model   = st.secrets["OPENAI_MODEL"]
 
 def get_openai_config_simple():
-    """
-    Returns a small config dict:
-      {
-        "OPENAI_apiKey": <string or None>,
-        "OPENAI_model": <string or None>
-      }
-    Priority: env var -> st.secrets['openai'] -> session UI key
-    """
-    cfg = {
-        "OPENAI_apiKey": os.environ.get("OPENAI_API_KEY"),
-        "OPENAI_model": os.environ.get("OPENAI_MODEL"),
-    }
-
-    # Try reading .streamlit/secrets.toml [openai] table if present
-    try:
-        if hasattr(st, "secrets"):
-            openai_sec = st.secrets.get("openai")
-            if isinstance(openai_sec, dict):
-                cfg["OPENAI_apiKey"] = openai_sec.get("OPENAI_API_KEY") or openai_sec.get("api_key") or cfg["OPENAI_apiKey"]
-                cfg["OPENAI_model"]  = openai_sec.get("OPENAI_MODEL")  or openai_sec.get("model")   or cfg["OPENAI_model"]
-    except Exception:
-        pass
-
-    # fallback to the temporary UI key if nothing else
-    if not cfg["OPENAI_apiKey"]:
-        cfg["OPENAI_apiKey"] = st.session_state.get("ui_api_key")
-
-    return cfg
-
-def get_effective_model():
-    # Prefer env var
-    env_model = os.environ.get("OPENAI_MODEL")
-    if env_model:
-        return env_model
-    
-    # Try Streamlit secrets (handle nested or flat)
-    secret_model = None
-    if hasattr(st, "secrets"):
-        try:
-            # Nested under [openai]
-            secret_model = st.secrets["openai"]["OPENAI_MODEL"]
-        except KeyError:
-            # Flat key
-            secret_model = st.secrets.get("OPENAI_MODEL")
-    
-    # Fallback to default
-    return secret_model or DEFAULT_MODEL
+    return {"OPENAI_apiKey": OPENAI_api_key, "OPENAI_model": OPENAI_model}
 
 def make_openai_client(api_key):
     if not api_key: