Update app.py

app.py

@@ -2,13 +2,9 @@ import streamlit as st
 from docx import Document
 import requests
 import io
-import os
-
-HF_API_KEY = os.getenv("HF_API_KEY")
 
 # Hugging Face API details
 API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-large"
-HEADERS = {"Authorization": "Bearer HF_API_KEY"}
 
 # Department and Role mapping
 DEPARTMENT_ROLES = {
@@ -56,11 +52,22 @@ def generate_jd_with_hf(department, role):
         }
     }
     
-    response = requests.post(API_URL, headers=HEADERS, json=payload)
-    if response.status_code == 200:
-        return response.json()[0]['generated_text']
-    else:
-        raise Exception(f"API Error: {response.status_code} - {response.text}")
+    try:
+        # Get API key from Streamlit secrets (for Hugging Face Spaces) or environment variable
+        api_key = st.secrets.get("HUGGINGFACE_API_KEY", "YOUR_HUGGINGFACE_API_KEY")
+        headers = {"Authorization": f"Bearer {api_key}"}
+        
+        response = requests.post(API_URL, headers=headers, json=payload)
+        
+        # Check for API errors
+        if response.status_code == 200:
+            return response.json()[0]['generated_text']
+        else:
+            st.error(f"API Error: {response.status_code} - {response.text}")
+            return None
+    except Exception as e:
+        st.error(f"Failed to connect to Hugging Face API: {str(e)}")
+        return None
 
 # Function to create Word document
 def create_word_document(role, content):
@@ -111,10 +118,10 @@ with col2:
 # Generate JD button
 if st.button("✨ Generate Job Description"):
     with st.spinner("Generating professional job description using AI..."):
-        try:
-            # Generate JD content
-            jd_content = generate_jd_with_hf(department, role)
-            
+        # Generate JD content
+        jd_content = generate_jd_with_hf(department, role)
+        
+        if jd_content:
             # Create Word document
             doc_buffer = create_word_document(role, jd_content)
             
@@ -131,9 +138,6 @@ if st.button("✨ Generate Job Description"):
                 mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
                 help="Download the job description in Word format"
             )
-            
-        except Exception as e:
-            st.error(f"Error generating job description: {str(e)}")
 
 # Instructions
 st.markdown("""