Update app.py

app.py

@@ -32,10 +32,19 @@ def load_model():
 with st.spinner("Loading transformer model..."):
     model = load_model()
 
-# Sidebar for API key
+# Get API key from environment variables (Hugging Face Spaces secrets)
+api_key = os.getenv("OPENROUTER_API_KEY")
+
+# Sidebar for configuration
 with st.sidebar:
     st.header("⚙️ Configuration")
-    api_key = st.text_input("OpenRouter API Key", type="password", help="Get your API key from https://openrouter.ai/keys")
+    
+    if api_key:
+        st.success("✅ API Key loaded from environment")
+    else:
+        st.error("❌ API Key not found in environment")
+        api_key = st.text_input("OpenRouter API Key", type="password", help="Get your API key from https://openrouter.ai/keys")
+        st.info("💡 Tip: Set OPENROUTER_API_KEY in Hugging Face Spaces secrets to avoid typing it every time")
     
     st.markdown("---")
     st.markdown("""