Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import pandas as pd
 import json
+import io
 
 # ----------------------------
 # CONFIG
@@ -98,26 +99,34 @@ Respond only 'Yes' or 'No'.
                 "Salary": person.get("annual_salary_expectation", {}).get("full-time", "N/A")
             })
     if len(candidates) == 0:
-        return f"No suitable candidates found for {category_name}."
-    return pd.DataFrame(candidates)
+        return pd.DataFrame(), None
+    df = pd.DataFrame(candidates)
+    # Convert to CSV for download
+    csv_buffer = io.StringIO()
+    df.to_csv(csv_buffer, index=False)
+    csv_data = csv_buffer.getvalue()
+    return df, csv_data
 
 # ----------------------------
 # GRADIO INTERFACE
 # ----------------------------
 def run_dashboard(category):
     if category not in CATEGORIES:
-        return f"Category {category} not found."
-    df = get_candidates_by_category(category, CATEGORIES[category])
-    return df
+        return f"Category {category} not found.", None
+    df, csv_data = get_candidates_by_category(category, CATEGORIES[category])
+    if df.empty:
+        return f"No suitable candidates found for {category}.", None
+    return df, csv_data
 
 category_options = list(CATEGORIES.keys())
 
 demo = gr.Interface(
     fn=run_dashboard,
     inputs=gr.Dropdown(category_options, label="Select Category"),
-    outputs=gr.Dataframe(label="Suitable Candidates"),
+    outputs=[gr.Dataframe(label="Suitable Candidates"), gr.File(label="Download CSV")],
     live=False,
-    title="Startup Candidate Dashboard - Zephyr-7B-Beta"
+    title="Startup Candidate Dashboard - Zephyr-7B-Beta",
+    description="Select a category to view suitable candidates. You can also download the results as CSV."
 )
 
 if __name__ == "__main__":