Update app.py

app.py

@@ -11,14 +11,13 @@ JSON_URL = "https://file.notion.so/f/f/f86ed84d-b33c-4dfb-b0e0-97c5661516a3/3ed5
 
 MODEL_ID = "HuggingFaceH4/zephyr-7b-beta"
 
-# Access Hugging Face secret in Spaces
 import os
 HF_API_TOKEN = os.environ.get("HF_API_TOKEN")
 if not HF_API_TOKEN:
     raise ValueError("HF_API_TOKEN not found in environment. Add it in Space Secrets.")
 
 # ----------------------------
-# CATEGORIES DEFINED BY JOB TITLES
+# CATEGORIES
 # ----------------------------
 CATEGORIES = {
     "AI": [
@@ -98,14 +97,16 @@ Respond only 'Yes' or 'No'.
                 "Skills": ", ".join(person.get("skills", [])),
                 "Salary": person.get("annual_salary_expectation", {}).get("full-time", "N/A")
             })
+
     if len(candidates) == 0:
         return pd.DataFrame(), None
+
     df = pd.DataFrame(candidates)
-    # Convert to CSV for download
-    csv_buffer = io.StringIO()
+    # Convert to CSV as BytesIO for Gradio
+    csv_buffer = io.BytesIO()
     df.to_csv(csv_buffer, index=False)
-    csv_data = csv_buffer.getvalue()
-    return df, csv_data
+    csv_buffer.seek(0)
+    return df, csv_buffer
 
 # ----------------------------
 # GRADIO INTERFACE
@@ -113,10 +114,10 @@ Respond only 'Yes' or 'No'.
 def run_dashboard(category):
     if category not in CATEGORIES:
         return f"Category {category} not found.", None
-    df, csv_data = get_candidates_by_category(category, CATEGORIES[category])
+    df, csv_file = get_candidates_by_category(category, CATEGORIES[category])
     if df.empty:
         return f"No suitable candidates found for {category}.", None
-    return df, csv_data
+    return df, csv_file
 
 category_options = list(CATEGORIES.keys())