Update app.py

app.py

@@ -173,7 +173,7 @@ def open_dashboard(role, supervisor_name, project_id):
 
 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.", "", ""
+        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)
@@ -191,7 +191,7 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
         result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     except Exception as e:
         print(f"⚠️ Generation error: {e}")
-        return "", "", ""
+        return "", "", None, ""
 
     def extract_between(text, start, end):
         s = text.find(start)
@@ -211,7 +211,7 @@ def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
     if pdf_url:
         suggestions += f"\n\n🔗 [Download PDF Report]({pdf_url})"
 
-    return checklist, suggestions, f'<a href="{pdf_url}" target="_blank">Download PDF</a>' if pdf_url else ""
+    return checklist, suggestions, pdf_path, f'<a href="{pdf_url}" target="_blank">Download PDF</a>' if pdf_url else ""
 
 def create_interface():
     roles = get_roles_from_salesforce()
@@ -234,15 +234,20 @@ def create_interface():
 
         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)
 
-        generate.click(fn=generate_outputs,
+        def handle_generate(role, supervisor_name, project_id, milestones, reflection):
+            checklist, suggestions, pdf_path, html = generate_outputs(role, supervisor_name, project_id, milestones, reflection)
+            return checklist, suggestions, pdf_path, html
+
+        generate.click(fn=handle_generate,
                        inputs=[role, supervisor_name, project_id, milestones, reflection],
-                       outputs=[checklist_output, suggestions_output, pdf_link])
+                       outputs=[checklist_output, suggestions_output, download_button, pdf_link])
 
         clear.click(fn=lambda: ("", "", "", "", ""),
                     inputs=None,