Update app.py

app.py

@@ -95,7 +95,7 @@ def save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions
     return temp_pdf_path, filename
 
 # Upload to Salesforce and create a new record with the generated URL
-def upload_pdf_to_salesforce_and_create_new_record(role, supervisor_name, project_id, pdf_path, pdf_name):
+def upload_pdf_to_salesforce_and_create_new_record(role, supervisor_name, project_id, pdf_path, pdf_name, checklist, suggestions):
     try:
         # Connecting to Salesforce
         sf = Salesforce(
@@ -195,6 +195,51 @@ 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 and return the correct file for download
+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_create_new_record(supervisor_name, project_id, pdf_path, pdf_name, checklist, suggestions)
+
+    if pdf_url:
+        suggestions += f"\n\n🔗 [Download PDF Report]({pdf_url})"
+
+    # Return the generated checklist, suggestions, and the path to the PDF
+    return checklist, suggestions, pdf_path, pdf_name
+
 # Gradio Interface
 def create_interface():
     roles = get_roles_from_salesforce()