Update app.py

app.py

@@ -7,8 +7,8 @@ import base64
 import datetime
 from dotenv import load_dotenv
 from fpdf import FPDF
-import html
 import shutil
+import html
 
 # Load environment variables
 load_dotenv()
@@ -63,9 +63,10 @@ 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}"
+    file_path = f"./reports/{filename}"  # Save file in './reports' directory
     os.makedirs("reports", exist_ok=True)
 
+    # Generate PDF content
     pdf = FPDF()
     pdf.add_page()
     pdf.set_font("Arial", 'B', 14)
@@ -86,13 +87,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)
+    pdf.output(file_path)  # Save the PDF to the file path
 
-    # Copy the file to a temporary directory for Gradio to access
-    temp_pdf_path = "/tmp/" + os.path.basename(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
     shutil.copy(file_path, temp_pdf_path)
 
-    return temp_pdf_path, filename
+    return temp_pdf_path  # Return the path to Gradio
 
 # 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):
@@ -204,54 +205,25 @@ 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})"
+    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"
 
+    # 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():
-    roles = get_roles_from_salesforce()
-    with gr.Blocks(theme="soft") as demo:
+    with gr.Blocks() as demo:
         gr.Markdown("## 🧠 AI-Powered Supervisor Assistant")
 
         with gr.Row():
-            role = gr.Dropdown(choices=roles, label="Role")
-            supervisor_name = gr.Dropdown(choices=[], label="Supervisor Name")
-            project_id = gr.Textbox(label="Project ID", interactive=False)
+            role = gr.Dropdown(choices=["Supervisor", "Manager"], label="Role")
+            supervisor_name = gr.Textbox(label="Supervisor Name")
+            project_id = gr.Textbox(label="Project ID")
 
         milestones = gr.Textbox(label="Milestones (comma-separated KPIs)")
         reflection = gr.Textbox(label="Reflection Log", lines=4)
@@ -260,35 +232,19 @@ 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")
-        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)
+        download_button = gr.File(label="⬇ Download Report")  # For downloading the generated report
 
-        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=handle_generate,
+        generate.click(fn=generate_outputs,
                        inputs=[role, supervisor_name, project_id, milestones, reflection],
-                       outputs=[checklist_output, suggestions_output, download_button, pdf_link])
+                       outputs=[checklist_output, suggestions_output, download_button, download_button])
 
         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__":