Update src/streamlit_app.py

src/streamlit_app.py

@@ -112,24 +112,29 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # Function to safely get OpenAI API key
+# Function to safely get OpenAI API key (UPDATED)
 def get_openai_api_key():
     """Safely retrieve OpenAI API key from various sources"""
     api_key = None
     
-    # Method 1: Try from Streamlit secrets
-    try:
-        if hasattr(st, 'secrets') and 'OPENAI_API_KEY' in st.secrets:
-            api_key = st.secrets["OPENAI_API_KEY"]
-    except Exception:
-        pass
+    # Method 1: Try from environment variables first (for HF Spaces)
+    api_key = os.getenv('OPENAI_API_KEY')
     
-    # Method 2: Try from environment variables
+    # Method 2: Try alternative environment variable names
     if not api_key:
-        api_key = os.getenv('OPENAI_API_KEY')
+        api_key = os.getenv('OPENAI_API_TOKEN')
     
-    # Method 3: Try from Hugging Face Spaces environment
+    # Method 3: Try from Streamlit secrets (only if secrets exist)
     if not api_key:
-        api_key = os.getenv('OPENAI_API_TOKEN')
+        try:
+            # Check if secrets file exists before accessing
+            import streamlit as st
+            if hasattr(st, 'secrets') and st.secrets:
+                if 'OPENAI_API_KEY' in st.secrets:
+                    api_key = st.secrets["OPENAI_API_KEY"]
+        except (FileNotFoundError, Exception):
+            # Silently continue if secrets file doesn't exist
+            pass
     
     return api_key