Update src/streamlit_app.py

src/streamlit_app.py

@@ -88,15 +88,6 @@ def calculate_experience(work_history):
 
     return round(total_experience, 1)
 
-def get_api_key():
-    """Safely get API key from secrets or user input."""
-    try:
-        # Try to get from Streamlit secrets
-        return st.secrets["GEMINI_API_KEY"]
-    except (KeyError, FileNotFoundError, st.errors.StreamlitSecretNotFoundError):
-        # If secrets don't exist or key is not found, ask user for input
-        return st.text_input("Enter Gemini API Key", type="password")
-
 def parse_resume(file_uploaded, api_key):
     """Parse resume and extract information."""
     genai.configure(api_key=api_key)
@@ -195,12 +186,12 @@ def main():
     st.title("Resume Parser")
     st.write("Upload a resume (PDF, DOCX, or Image) to extract information")
 
-    # Get API key from secrets only
-    api_key = get_api_key()
+    # 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")
 
     uploaded_file = st.file_uploader("Choose a resume file", type=["pdf", "docx", "doc", "jpg", "jpeg", "png"])
 
-    if uploaded_file:
+    if uploaded_file and api_key:
         with st.spinner('Analyzing resume...'):
             result = parse_resume(uploaded_file, api_key)