Update app.py

app.py

@@ -69,16 +69,21 @@ def save_captions_to_csv(captions, csv_file_path='image_captions.csv'):
         writer.writerows(captions)
     return csv_file_path
 
-def gr_process(zip_files, image_files):
+def gr_process_zip(zip_files):
     captions = []
     if zip_files:
         zip_file_paths = [zip_file.name for zip_file in zip_files]
         captions.extend(process_zip_files(zip_file_paths))
+    
+    return save_captions_to_csv(captions, 'zip_image_captions.csv')
+
+def gr_process_images(image_files):
+    captions = []
     if image_files:
         image_file_paths = [image_file.name for image_file in image_files]
         captions.extend(process_images(image_file_paths))
     
-    return save_captions_to_csv(captions)
+    return save_captions_to_csv(captions, 'image_captions.csv')
 
 def combine_csv_files(file1, file2, output_file='combined_captions.csv'):
     with open(output_file, mode='w', newline='') as outfile:
@@ -96,9 +101,9 @@ def combine_csv_files(file1, file2, output_file='combined_captions.csv'):
     return output_file
 
 def add_files(zip_files, image_files, new_zip_files, new_image_files):
-           zip_files += new_zip_files
-           image_files += new_image_files
-           return zip_files, image_files
+    zip_files += new_zip_files
+    image_files += new_image_files
+    return zip_files, image_files
 
 css = '''
 h1#title {
@@ -134,21 +139,20 @@ with demo:
         with gr.Column(scale=1):
             new_zip_files = gr.File(label="Upload Zip Files", type="filepath", file_count="multiple")
             generate_zip_captions_btn = gr.Button("Generate Zip Captions")
-            output_zip_file = gr.File(label="Download Zip Captions")
+            output_zip_file = gr.File(label="Download Zip Captions", interactive=False)
         with gr.Column(scale=1):
             new_image_files = gr.File(label="Upload Images", type="filepath", file_count="multiple")
             generate_image_captions_btn = gr.Button("Generate Image Captions")
-            output_image_file = gr.File(label="Download Image Captions")
-
-    add_files_btn = gr.Button("Add More Files")
-    combine_files_btn = gr.Button("Combine CSV Files")
-    combined_file = gr.File(label="Download Combined Captions")
+            output_image_file = gr.File(label="Download Image Captions", interactive=False)
 
+    with gr.Row():
+        add_files_btn = gr.Button("Add More Files")
+        combine_files_btn = gr.Button("Combine CSV Files")
+        combined_file = gr.File(label="Download Combined Captions", interactive=False)
       
-
     add_files_btn.click(add_files, inputs=[zip_files, image_files, new_zip_files, new_image_files], outputs=[zip_files, image_files])
-    generate_zip_captions_btn.click(fn=gr_process, inputs=zip_files, outputs=output_zip_file)
-    generate_image_captions_btn.click(fn=gr_process, inputs=image_files, outputs=output_image_file)
+    generate_zip_captions_btn.click(fn=gr_process_zip, inputs=zip_files, outputs=output_zip_file)
+    generate_image_captions_btn.click(fn=gr_process_images, inputs=image_files, outputs=output_image_file)
     combine_files_btn.click(fn=combine_csv_files, inputs=[output_zip_file, output_image_file], outputs=combined_file)
 
 demo.launch(share=True)