Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,6 +9,7 @@ from datetime import datetime
 import re
 import pytesseract
 import io
+import os
 
 def extract_text_from_pdf(pdf_file):
     """Extract text from uploaded PDF file."""
@@ -182,16 +183,41 @@ def format_education(edu):
         parts.append(f"- GPA: {edu['gpa']}")
     return " ".join(parts)
 
+def get_api_key():
+    """Get API key from multiple sources."""
+    # Try environment variable first (Hugging Face Spaces)
+    api_key = os.getenv("GEMINI_API_KEY")
+    if api_key:
+        return api_key
+    
+    # Try Streamlit secrets (local development)
+    try:
+        if hasattr(st, 'secrets') and "GEMINI_API_KEY" in st.secrets:
+            return st.secrets["GEMINI_API_KEY"]
+    except:
+        pass
+    
+    # Return None if not found
+    return None
+
 def main():
     st.title("Resume Parser")
     st.write("Upload a resume (PDF, DOCX, or Image) to extract information")
 
-    # Get API key from secrets or user input
-    api_key = st.secrets["GEMINI_API_KEY"] if "GEMINI_API_KEY" in st.secrets else st.text_input("Enter Gemini API Key", type="password")
+    # Get API key from multiple sources
+    api_key = get_api_key()
+    
+    # If no API key found, ask user to input
+    if not api_key:
+        api_key = st.text_input("Enter Gemini API Key", type="password")
+        if not api_key:
+            st.warning("Please enter your Gemini API Key to proceed.")
+            st.info("You can get your API key from: https://makersuite.google.com/app/apikey")
+            return
 
     uploaded_file = st.file_uploader("Choose a resume file", type=["pdf", "docx", "doc", "jpg", "jpeg", "png"])
 
-    if uploaded_file and api_key:
+    if uploaded_file:
         with st.spinner('Analyzing resume...'):
             result = parse_resume(uploaded_file, api_key)