Update app.py

app.py

@@ -2,15 +2,16 @@ import os
 import streamlit as st
 from streamlit.errors import StreamlitSecretNotFoundError
 
-hf_token = os.getenv("HF_TOKEN")  # default from env
+hf_token = os.getenv("HF_TOKEN")  # fallback to environment variable
 
 try:
-    # only works if secrets.toml exists
-    if "HF_TOKEN" in st.secrets:
-        hf_token = st.secrets["HF_TOKEN"]
+    hf_token = st.secrets.get("HF_TOKEN", hf_token)  # only works if secrets exist
 except StreamlitSecretNotFoundError:
-    # no secrets configured; keep env value
-    pass
+    pass  # no secrets configured, keep env token
+
+if not hf_token:
+    st.error("HF_TOKEN not found. Add it to Streamlit Secrets or set it as an env variable.")
+    st.stop()
 
 st.set_page_config(page_title="Multi-Agent Codegen (HF)", layout="wide")