Update app.py

app.py

@@ -67,6 +67,10 @@ def bulk_output(excel, pdf, progress: gr.Progress = gr.Progress(track_tqdm=True)
 
     return "Emails have been drafted and saved as Word documents."
 
+def generate_and_download(name, company, ai_email):
+    file_path = config.write_email(name, company, ai_email)
+    return file_path
+
 with gr.Blocks() as demo:
     with gr.Tabs():
         with gr.TabItem("Single Email"):
@@ -87,7 +91,7 @@ with gr.Blocks() as demo:
 
             state = gr.State()
             submit_btn.click(output, inputs=[pdf_input, text1, text2, text3, text4, text5, text6], outputs=[output_box, state])
-            write_btn.click(config.write_email, inputs=[text1, text2, state])
+            write_btn.click(generate_and_download, inputs=[text1, text2, state])
 
         with gr.TabItem("Batch Email"):
             with gr.Row():