javascript_autotrigger

app.py

@@ -67,9 +67,10 @@ def merge_overlapping_rectangles(rectangles):
         merged_rects.append(merged_rect)
     return merged_rects
 
-def markup_color_regions(doc, color_comment_pairs, tolerance=30):
+def markup_color_regions(doc, color_comment_pairs, tolerance=30, progress=None):
     for page_num in range(len(doc)):
         page = doc[page_num]
+        progress((page_num + 1) / len(doc))  # Update the progress as each page is processed
         for color_comment_pair in color_comment_pairs:
             target_color = color_comment_pair['color']
             comment = color_comment_pair['comment']
@@ -85,7 +86,7 @@ def markup_color_regions(doc, color_comment_pairs, tolerance=30):
                     annot.set_info({"title": "Markup", "content": comment})
                     annot.update()
 
-def process_pdf_files(input_pdfs, selected_color_comment_indices, tolerance, custom_color, custom_comment, custom_stroke_color):
+def process_pdf_files(input_pdfs, selected_color_comment_indices, tolerance, custom_color, custom_comment, custom_stroke_color, progress=gr.Progress()):
     color_comment_pairs = [
         {
             "color": (235, 128, 138),
@@ -116,10 +117,10 @@ def process_pdf_files(input_pdfs, selected_color_comment_indices, tolerance, cus
     modified_pdfs = []
 
     # Process each input PDF file
-    for pdf_file in input_pdfs:
+    for i, pdf_file in enumerate(input_pdfs):
         with open(pdf_file.name, "rb") as file_stream:
             doc = fitz.open(stream=file_stream.read(), filetype="pdf")
-            markup_color_regions(doc, selected_color_comment_pairs, tolerance)
+            markup_color_regions(doc, selected_color_comment_pairs, tolerance, progress=lambda p: progress((i + p) / len(input_pdfs)))
 
             # Save the modified PDF in memory
             pdf_in_memory = io.BytesIO()
@@ -152,8 +153,18 @@ def process_pdf_files(input_pdfs, selected_color_comment_indices, tolerance, cus
     with open(temp_filename, 'wb') as tmp_file:
         tmp_file.write(zip_in_memory.read())
 
-    # Return the downloadable file path as a link
-    return f'<a href="{temp_filename}" download>Click here to download the ZIP file: {zip_filename}</a>'
+    # Return the downloadable file path as a JS script for auto-download
+    return f'''
+        <script>
+            var link = document.createElement('a');
+            link.href = '{temp_filename}';
+            link.download = '{zip_filename}';
+            document.body.appendChild(link);
+            link.click();
+            document.body.removeChild(link);
+        </script>
+        File is being downloaded automatically.
+    '''
 
 # Define the Gradio interface
 interface = gr.Interface(
@@ -170,7 +181,7 @@ interface = gr.Interface(
         gr.Textbox(label="Custom Comment", placeholder="Enter custom comment for this color"),
         gr.Textbox(label="Custom Stroke Color (R,G,B)", placeholder="Enter stroke color in RGB format, e.g., 0,0,255")
     ],
-    outputs="html",  # Return HTML content with a download link
+    outputs="html",  # Return HTML content with JavaScript for download
     title="PDF Color Region Markup"
 )