Update app.py

app.py

@@ -177,11 +177,10 @@ def get_projects_for_supervisor(supervisor_name):
         print(f"⚠️ Error fetching project: {e}")
         return ""
 
-# Function to generate Salesforce dashboard URL
+# 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}"
 
-# Define the open_dashboard function
 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>'
@@ -224,8 +223,10 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
 
     pdf_path, pdf_name = save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions)
     pdf_url = upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_path, pdf_name)
+    if pdf_url:
+        suggestions += f"\n\n🔗 [Download PDF Report]({pdf_url})"
 
-    return checklist, suggestions, pdf_url, f'<a href="{pdf_url}" target="_blank">Download PDF</a>' if pdf_url else ""
+    return checklist, suggestions, pdf_path, f'<a href="{pdf_url}" target="_blank">Download PDF</a>' if pdf_url else ""
 
 # Gradio Interface
 def create_interface():
@@ -257,8 +258,8 @@ def create_interface():
         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_url, html = generate_outputs(role, supervisor_name, project_id, milestones, reflection)
-            return checklist, suggestions, pdf_url, html
+            checklist, suggestions, pdf_path, html = generate_outputs(role, supervisor_name, project_id, milestones, reflection)
+            return checklist, suggestions, pdf_path, html
 
         generate.click(fn=handle_generate,
                        inputs=[role, supervisor_name, project_id, milestones, reflection],
@@ -278,4 +279,4 @@ def create_interface():
 
 if __name__ == "__main__":
     app = create_interface()
-    app.launch()
+    app.launch()