Update app.py

app.py

@@ -26,7 +26,7 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 def process(image, progress=gr.Progress()):
     if image is None:
-        return None, None
+        return None, None, None
     try:
         progress(0, desc="Starting processing...")
         orig_image = Image.fromarray(image)
@@ -74,11 +74,21 @@ def process(image, progress=gr.Progress()):
         output_array = np.array(new_im.convert('RGBA'))
         
         progress(1.0, desc="Done!")
-        return output_array, gr.update(value=filepath, visible=True)
+        return (
+            output_array, 
+            gr.update(value=filepath, visible=True),
+            gr.update(value=f"""
+                <script>
+                    setTimeout(function() {{
+                        window.location.href = '/file={filepath}';
+                    }}, 1000);
+                </script>
+            """)
+        )
     
     except Exception as e:
         print(f"Error processing image: {str(e)}")
-        return None, None
+        return None, None, None
 
 css = """
 @import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;500;700&display=swap');
@@ -182,37 +192,15 @@ with gr.Blocks(css=css) as demo:
             download_button = gr.File(
                 label="Download Processed Image",
                 visible=True,
-                elem_classes="download-btn",
-                interactive=True
+                elem_classes="download-btn"
             )
-    
-    # Add JavaScript for auto-download
-    demo.load(None, None, None, _js="""
-        function autoDownload() {
-            const downloadButtons = document.querySelectorAll('.download-btn button');
-            if (downloadButtons.length > 0) {
-                downloadButtons[0].click();
-            }
-        }
-        
-        const observer = new MutationObserver((mutations) => {
-            mutations.forEach((mutation) => {
-                if (mutation.type === 'childList') {
-                    const downloadButtons = document.querySelectorAll('.download-btn button');
-                    if (downloadButtons.length > 0) {
-                        setTimeout(autoDownload, 1000);
-                    }
-                }
-            });
-        });
-        
-        observer.observe(document.body, { childList: true, subtree: true });
-    """)
+            
+            auto_download = gr.HTML(visible=False)
     
     input_image.change(
         fn=process,
         inputs=input_image,
-        outputs=[output_image, download_button]
+        outputs=[output_image, download_button, auto_download]
     )
 
 if __name__ == "__main__":