Update main app with improved API key handling

README.md

@@ -5,7 +5,7 @@ colorFrom: blue
 colorTo: purple
 sdk: streamlit
 sdk_version: 1.31.1
-app_file: src/minimal_test.py
+app_file: src/streamlit_app.py
 pinned: false
 license: mit
 ---

src/streamlit_app.py

@@ -16,15 +16,27 @@ try:
     if os.path.exists(".env"):
         load_dotenv()
     
-    # Get API key from environment
-    api_key = os.getenv('ANTHROPIC_API_KEY')
+    # Try multiple ways to get the API key
+    api_key = None
+    
+    # Try direct environment variable
+    if 'ANTHROPIC_API_KEY' in os.environ:
+        api_key = os.environ['ANTHROPIC_API_KEY']
+    
+    # Try getenv
+    if not api_key:
+        api_key = os.getenv('ANTHROPIC_API_KEY')
+    
+    # Try Streamlit secrets
+    if not api_key and hasattr(st, 'secrets') and 'ANTHROPIC_API_KEY' in st.secrets:
+        api_key = st.secrets['ANTHROPIC_API_KEY']
+    
     if not api_key:
         st.error("""
-        ⚠️ Anthropic API Key not found!
-        
-        Please make sure you have set the ANTHROPIC_API_KEY:
-        - For local development: in your .env file
-        - For Hugging Face Spaces: in your Space's secrets
+        ⚠️ No API key found. Please make sure:
+        1. You've added the secret in Hugging Face Space settings
+        2. The secret is named exactly 'ANTHROPIC_API_KEY'
+        3. The Space has been rebuilt after adding the secret
         """)
         st.stop()