Update app.py

app.py

@@ -186,6 +186,58 @@ def get_projects_for_supervisor(supervisor_name):
         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]):
+        return "❗ Please fill all fields.", "", None, ""
+
+    prompt = PROMPT_TEMPLATE.format(role=role, project_id=project_id, milestones=milestones, reflection=reflection)
+    inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True, max_length=512)
+    try:
+        with torch.no_grad():
+            outputs = model.generate(
+                inputs['input_ids'],
+                max_new_tokens=150,
+                no_repeat_ngram_size=2,
+                do_sample=True,
+                top_p=0.9,
+                temperature=0.7,
+                pad_token_id=tokenizer.pad_token_id
+            )
+        result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    except Exception as e:
+        print(f"⚠️ Generation error: {e}")
+        return "", "", None, ""
+
+    def extract_between(text, start, end):
+        s = text.find(start)
+        e = text.find(end, s) if end else len(text)
+        return text[s + len(start):e].strip() if s != -1 else ""
+
+    checklist = extract_between(result, "Checklist:\n", "Suggestions:")
+    suggestions = extract_between(result, "Suggestions:\n", None)
+
+    if not checklist.strip():
+        checklist = "- Perform daily safety inspection"
+    if not suggestions.strip():
+        suggestions = "- Monitor team coordination\n- Review safety protocols with the team"
+
+    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})"
+
+    # Ensure that pdf_path is being passed correctly to gr.File component
+    return checklist, suggestions, pdf_path, f'<a href="{pdf_url}" target="_blank">Download PDF</a>' if pdf_url else ""
+
 # Gradio Interface
 def create_interface():
     roles = get_roles_from_salesforce()