Update app.py

app.py

@@ -159,6 +159,32 @@ def get_supervisor_name_by_role(role):
         print(f"⚠️ Error fetching names: {e}")
         return []
 
+def get_projects_for_supervisor(supervisor_name):
+    try:
+        sf = Salesforce(
+            username=os.getenv('SF_USERNAME'),
+            password=os.getenv('SF_PASSWORD'),
+            security_token=os.getenv('SF_SECURITY_TOKEN'),
+            domain=os.getenv('SF_DOMAIN', 'login')
+        )
+        result = sf.query(f"SELECT Id FROM Supervisor__c WHERE Name = '{supervisor_name}' LIMIT 1")
+        if result['totalSize'] == 0:
+            return ""
+        supervisor_id = result['records'][0]['Id']
+        project_result = sf.query(f"SELECT Name FROM Project__c WHERE Supervisor_ID__c = '{supervisor_id}' LIMIT 1")
+        return project_result['records'][0]['Name'] if project_result['totalSize'] > 0 else ""
+    except Exception as e:
+        print(f"⚠️ Error fetching project: {e}")
+        return ""
+
+# Generate Salesforce dashboard URL
+def generate_salesforce_dashboard_url(supervisor_name, project_id):
+    return f"https://aicoachforsitesupervisors-dev-ed--c.develop.vf.force.com/apex/DashboardPage?supervisorName={supervisor_name}&projectId={project_id}"
+
+def open_dashboard(role, supervisor_name, project_id):
+    url = generate_salesforce_dashboard_url(supervisor_name, project_id)
+    return f'<a href="{url}" target="_blank">Open Salesforce Dashboard</a>'
+
 # Generate AI output
 def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
     if not all([role, supervisor_name, project_id, milestones, reflection]):
@@ -202,15 +228,6 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
 
     return checklist, suggestions, pdf_path, f'<a href="{pdf_url}" target="_blank">Download PDF</a>' if pdf_url else ""
 
-# Function to generate Salesforce dashboard URL
-def generate_salesforce_dashboard_url(supervisor_name, project_id):
-    return f"https://aicoachforsitesupervisors-dev-ed--c.develop.vf.force.com/apex/DashboardPage?supervisorName={supervisor_name}&projectId={project_id}"
-
-# Function to open dashboard
-def open_dashboard(role, supervisor_name, project_id):
-    url = generate_salesforce_dashboard_url(supervisor_name, project_id)
-    return f'<a href="{url}" target="_blank">Open Salesforce Dashboard</a>'
-
 # Gradio Interface
 def create_interface():
     roles = get_roles_from_salesforce()
@@ -238,10 +255,10 @@ def create_interface():
         dashboard_link = gr.HTML()
 
         role.change(fn=lambda r: gr.update(choices=get_supervisor_name_by_role(r)), inputs=role, outputs=supervisor_name)
+        supervisor_name.change(fn=get_projects_for_supervisor, inputs=supervisor_name, outputs=project_id)
 
         def handle_generate(role, supervisor_name, project_id, milestones, reflection):
             checklist, suggestions, pdf_path, html = generate_outputs(role, supervisor_name, project_id, milestones, reflection)
-            download_button.value = pdf_path
             return checklist, suggestions, pdf_path, html
 
         generate.click(fn=handle_generate,
@@ -254,7 +271,6 @@ def create_interface():
 
         refresh.click(fn=lambda: gr.update(choices=get_roles_from_salesforce()), outputs=role)
 
-        # Fix: Added the function for the button to work
         dashboard_btn.click(fn=open_dashboard,
                             inputs=[role, supervisor_name, project_id],
                             outputs=dashboard_link)
@@ -263,4 +279,4 @@ def create_interface():
 
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch()