Update app.py

app.py

@@ -8,6 +8,7 @@ import datetime
 from dotenv import load_dotenv
 from fpdf import FPDF
 import html
+import shutil
 
 # Load environment variables
 load_dotenv()
@@ -185,57 +186,6 @@ 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})"
-
-    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()