Update app.py

app.py

@@ -1,16 +1,23 @@
 import streamlit as st
 from openai import OpenAI
+import os
 
 # 1. Configuration
 st.set_page_config(page_title="High School Writing Coach", layout="wide")
 
-# Initialize client (ensure you have your API key in .streamlit/secrets.toml)
-# or set it directly for testing: client = OpenAI(api_key="sk-...")
-if "OPENAI_API_KEY" in st.secrets:
-    client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
-else:
-    st.error("Please set your OpenAI API Key in .streamlit/secrets.toml")
-    st.stop()
+# Get the API key from Environment Variables (HF Spaces) OR st.secrets (Local)
+api_key = os.environ.get("OPENAI_API_KEY")
+
+if not api_key:
+    try:
+        # Fallback for local development if using secrets.toml
+        api_key = st.secrets["OPENAI_API_KEY"]
+    except (FileNotFoundError, KeyError):
+        st.error("OpenAI API Key not found. Please add 'OPENAI_API_KEY' to your Hugging Face Space secrets.")
+        st.stop()
+
+# Initialize client with the found key
+client = OpenAI(api_key=api_key)
 
 # 2. System Prompt
 # This enforces the "No Rewrite" rule.