add docx file download

app.py

@@ -8,6 +8,7 @@ import zipfile
 import tempfile
 os.system("apt-get install poppler-utils")
 import datetime
+from docx import Document
 
 # Function to process a list of PDF files and convert them to images
 def process_pdfs(pdf_files):
@@ -130,7 +131,6 @@ def main_process(files, tanggal_berangkat, tanggal_pulang):
         file_path = file.name if hasattr(file, 'name') else file
 
         if file_path.lower().endswith('.zip'):
-            # Extract zip and process contained files
             extracted_files = extract_zip_and_collect_files(file_path)
             for extracted_file in extracted_files:
                 if extracted_file.lower().endswith('.pdf'):
@@ -149,7 +149,17 @@ def main_process(files, tanggal_berangkat, tanggal_pulang):
             raise ValueError(f"File {file_path} is not a valid image, PDF, or ZIP.")
 
     summary = gemini_analysis(all_images, tanggal_berangkat, tanggal_pulang)
-    return summary
+
+    # Save to DOCX
+    doc = Document()
+    doc.add_heading("Visa Document Check Summary", level=1)
+    for line in summary.split("\n"):
+        doc.add_paragraph(line)
+
+    temp_docx_path = os.path.join(tempfile.gettempdir(), f"summary_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.docx")
+    doc.save(temp_docx_path)
+
+    return summary, temp_docx_path
 
 
 # Gradio UI update: add ".zip" to accepted file types
@@ -173,9 +183,13 @@ with gr.Blocks() as demo:
             placeholder="Masukan Tanggal Kepulangan",
             type="text"
         )
-    run_btn = gr.Button("🏃 Run Analysis")
-    output = gr.Textbox(label="📝 Summary Result", lines=20)
+    with gr.Row():
+        run_btn = gr.Button("🏃 Run Analysis")
+        download_output = gr.File(label="📥 Download Summary as DOCX", visible=True)
 
-    run_btn.click(fn=main_process, inputs=[file_input, tanggal_berangkat, tanggal_pulang], outputs=output)
+    output = gr.Textbox(label="📝 Summary Result", lines=20)
 
+    run_btn.click(fn=main_process, 
+                  inputs=[file_input, tanggal_berangkat, tanggal_pulang], 
+                  outputs=[output, download_output])
 demo.launch()