Update app.py

app.py

@@ -95,6 +95,11 @@ 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
+
 css = '''
 h1#title {
   text-align: center;
@@ -139,12 +144,9 @@ with demo:
     combine_files_btn = gr.Button("Combine CSV Files")
     combined_file = gr.File(label="Download Combined 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])
+    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)
     combine_files_btn.click(fn=combine_csv_files, inputs=[output_zip_file, output_image_file], outputs=combined_file)