Update app.py

app.py

@@ -113,19 +113,17 @@ with demo:
     with gr.Row():
         with gr.Column(scale=1):
             new_zip_files = gr.File(label="Upload Zip Files", type="filepath", file_count="multiple")
+            zip_captions_btn = gr.Button("Generate Zip Captions")
+            output_zip_file = gr.File(label="Download Caption File")
+        with gr.Column(scale=1):
             new_image_files = gr.File(label="Upload Images", type="filepath", file_count="multiple")
-        with gr.Column(scale=3):
-            output_file = gr.File(label="Download Caption File")
+            image_captions_btn = gr.Button("Generate Image Captions")
+            output_image_file = gr.File(label="Download Caption File")
     
-    add_files_btn = gr.Button("Add More Files")
-    generate_captions_btn = gr.Button("Generate Captions")
-
-    def add_files(zip_files, image_files, new_zip_files, new_image_files):
-        zip_files.extend(new_zip_files)
-        image_files.extend(new_image_files)
-        return zip_files, image_files
+   
 
-    add_files_btn.click(add_files, inputs=[zip_files, image_files, new_zip_files, new_image_files], outputs=[zip_files, image_files])
-    generate_captions_btn.click(fn=gr_process, inputs=[zip_files, image_files], outputs=output_file)
+    
+    zip_captions_btn.click(fn=gr_process, inputs=zip_files, outputs=output_zip_file)
+    image_captions_btn.click(fn=gr_process, inputs=image_files, outputs=output_image_file)
 
 demo.launch(share=True)