Update app.py

app.py

@@ -10,22 +10,18 @@ import io
 import yaml
 
 def load_openai_key():
-    """Load OpenAI API key from environment variables or Streamlit secrets"""
+    """Load OpenAI API key from HuggingFace secrets"""
     try:
-        # First try to get from environment variable
-        api_key = os.getenv('OPENAI_API_KEY')
+        # Try to get from HuggingFace secrets via environment variable
+        api_key = os.getenv('OPENAI_API_KEY_HF')
         
-        # If not in env, try to get from Streamlit secrets
-        if not api_key and hasattr(st, 'secrets'):
-            api_key = st.secrets.get('OPENAI_API_KEY')
-            
         if not api_key:
-            st.error("OpenAI API key not found. Please set it in environment variables or Streamlit secrets.")
+            st.error("OpenAI API key not found in HuggingFace secrets. Please ensure the secret is set as OPENAI_API_KEY_HF in your HuggingFace space.")
             return None
             
         return api_key
     except Exception as e:
-        st.error(f"Error loading OpenAI API key: {str(e)}")
+        st.error(f"Error loading OpenAI API key from HuggingFace secrets: {str(e)}")
         return None
 
 def initialize_openai_client():
@@ -145,6 +141,10 @@ def main():
     if 'chat_history' not in st.session_state:
         st.session_state.chat_history = []
     
+    # Initialize variables
+    uploaded_file = None
+    screenshot_taken = False
+
     # Sidebar
     with st.sidebar:
         st.title("🚀 Chart Analysis AI")
@@ -157,8 +157,10 @@ def main():
         if upload_option == "Upload Image":
             uploaded_file = st.file_uploader("Upload your chart", type=["png", "jpg", "jpeg"])
         elif upload_option == "Take Screenshot":
-            st.button("Open Screenshot Tool", key="screenshot")
-            # Screenshot functionality would be implemented here
+            if st.button("Take Screenshot", key="screenshot"):
+                st.info("Feature coming soon! For now, please use the Upload Image option.")
+                # Future screenshot implementation will go here
+                screenshot_taken = False
         
         # Analysis Options
         st.subheader("Analysis Options")