Update app.py

app.py

@@ -63,10 +63,9 @@ def clean_text_for_pdf(text):
 def save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions):
     now = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
     filename = f"report_{supervisor_name}_{project_id}_{now}.pdf"
-    file_path = f"./reports/{filename}"  # Save file in './reports' directory
+    file_path = f"./reports/{filename}"
     os.makedirs("reports", exist_ok=True)
 
-    # Generate PDF content
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", 'B', 14)
@@ -87,13 +86,13 @@ def save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions
     pdf.set_font("Arial", size=12)
     for line in suggestions.split("\n"):
         pdf.multi_cell(0, 10, clean_text_for_pdf(line))
-    pdf.output(file_path)  # Save the PDF to the file path
+    pdf.output(file_path)
 
-    # Copy the file to /tmp/ (Gradio accessible location)
-    temp_pdf_path = "/tmp/" + os.path.basename(file_path)  # Use /tmp/ directory for Gradio access
+    # Copy the file to a temporary directory for Gradio to access
+    temp_pdf_path = "/tmp/" + os.path.basename(file_path)  # Use /tmp/ or current directory for Gradio
     shutil.copy(file_path, temp_pdf_path)
 
-    return temp_pdf_path  # Return the path to Gradio
+    return temp_pdf_path, filename
 
 # Upload to Salesforce and update record with the generated URL
 def upload_pdf_to_salesforce_and_update_link(supervisor_name, project_id, pdf_path, pdf_name):
@@ -205,25 +204,54 @@ 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 = f"Generate a Daily Checklist and Suggestions based on the inputs."
-    checklist = "- Perform daily safety inspection"
-    suggestions = "- Monitor team coordination\n- Review safety protocols with the team"
+    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})"
 
-    # Save the checklist and suggestions to a PDF and copy to /tmp/ directory for download
-    pdf_path = save_report_as_pdf(role, supervisor_name, project_id, checklist, suggestions)
-    
-    # Return the file path for Gradio's download button
     return checklist, suggestions, pdf_path, pdf_path
 
 # Gradio Interface
 def create_interface():
-    with gr.Blocks() as demo:
+    roles = get_roles_from_salesforce()
+    with gr.Blocks(theme="soft") as demo:
         gr.Markdown("## 🧠 AI-Powered Supervisor Assistant")
 
         with gr.Row():
-            role = gr.Dropdown(choices=["Supervisor", "Manager"], label="Role")
-            supervisor_name = gr.Textbox(label="Supervisor Name")
-            project_id = gr.Textbox(label="Project ID")
+            role = gr.Dropdown(choices=roles, label="Role")
+            supervisor_name = gr.Dropdown(choices=[], label="Supervisor Name")
+            project_id = gr.Textbox(label="Project ID", interactive=False)
 
         milestones = gr.Textbox(label="Milestones (comma-separated KPIs)")
         reflection = gr.Textbox(label="Reflection Log", lines=4)
@@ -232,19 +260,35 @@ def create_interface():
             generate = gr.Button("Generate")
             clear = gr.Button("Clear")
             refresh = gr.Button("🔄 Refresh Roles")
+            dashboard_btn = gr.Button("Dashboard")
 
         checklist_output = gr.Textbox(label="✅ Daily Checklist")
         suggestions_output = gr.Textbox(label="💡 Focus Suggestions")
-        download_button = gr.File(label="⬇ Download Report")  # For downloading the generated report
+        download_button = gr.File(label="⬇ Download Report")
+        pdf_link = gr.HTML()
+        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, _ = generate_outputs(role, supervisor_name, project_id, milestones, reflection)
+            return checklist, suggestions, pdf_path, pdf_path
 
-        generate.click(fn=generate_outputs,
+        generate.click(fn=handle_generate,
                        inputs=[role, supervisor_name, project_id, milestones, reflection],
-                       outputs=[checklist_output, suggestions_output, download_button, download_button])
+                       outputs=[checklist_output, suggestions_output, download_button, pdf_link])
 
         clear.click(fn=lambda: ("", "", "", "", ""),
                     inputs=None,
                     outputs=[role, supervisor_name, project_id, milestones, reflection])
 
+        refresh.click(fn=lambda: gr.update(choices=get_roles_from_salesforce()), outputs=role)
+
+        dashboard_btn.click(fn=open_dashboard,
+                            inputs=[role, supervisor_name, project_id],
+                            outputs=dashboard_link)
+
     return demo
 
 if __name__ == "__main__":