Update app.py

app.py

@@ -35,9 +35,10 @@ def process_images(image_files):
                 captions.append((os.path.basename(image_file), caption))
         except Exception as e:
             print(f"Skipping file {image_file}: {e}")
-    
-    return csv_file_path = save_captions_to_csv(captions, csv_file_path='image_captions.csv')
+    csv_file_path = save_captions_to_csv(captions, csv_file_path='zip_captions.csv')
+    return csv_file_path
 
+   
 def process_zip_files(zip_file_paths):
     # Create a directory to extract images
     extract_dir = 'extracted_images'
@@ -59,8 +60,8 @@ def process_zip_files(zip_file_paths):
                         captions.append((file, caption))
                 except Exception as e:
                     print(f"Skipping file {file}: {e}")
-    
-    return csv_file_path = save_captions_to_csv(captions, csv_file_path='zip_captions.csv')
+        csv_file_path = save_captions_to_csv(captions, csv_file_path='zip_captions.csv')
+    return csv_file_path
 
 def save_captions_to_csv(captions, csv_file_path='image_captions.csv'):
     with open(csv_file_path, mode='w', newline='') as file: