Update app.py

app.py

@@ -12,6 +12,7 @@ def process_zips(camera_zips, regex_pattern, start_frame, end_frame):
     
     Args:
         camera_zips (list): A list of uploaded ZIP files (each representing one camera folder).
+                              Each file is a dictionary with keys "name" and "data".
         regex_pattern (str): A regex pattern with one capturing group that extracts the numeric frame value.
         start_frame (str): The start frame number (as string, will be converted to int).
         end_frame (str): The end frame number (as string, will be converted to int).
@@ -47,13 +48,15 @@ def process_zips(camera_zips, regex_pattern, start_frame, end_frame):
     camera_folders = []
     
     for uploaded_zip in camera_zips:
+        # uploaded_zip is now a dict with keys "name" and "data"
+        zip_name = uploaded_zip["name"]
         # Save the uploaded ZIP to a temporary file
-        temp_zip_path = os.path.join(tempfile.gettempdir(), uploaded_zip.name)
+        temp_zip_path = os.path.join(tempfile.gettempdir(), zip_name)
         with open(temp_zip_path, "wb") as f:
-            f.write(uploaded_zip.read())
+            f.write(uploaded_zip["data"])
         
         # Create a temporary directory to extract the ZIP contents
-        extract_dir = os.path.join(tempfile.gettempdir(), uploaded_zip.name + "_extracted")
+        extract_dir = os.path.join(tempfile.gettempdir(), zip_name + "_extracted")
         os.makedirs(extract_dir, exist_ok=True)
         with zipfile.ZipFile(temp_zip_path, "r") as zip_ref:
             zip_ref.extractall(extract_dir)
@@ -64,7 +67,7 @@ def process_zips(camera_zips, regex_pattern, start_frame, end_frame):
         camera_folder = subdirs[0] if len(subdirs) == 1 else extract_dir
         
         # Use the ZIP filename (without extension) as the camera name
-        camera_name = os.path.splitext(uploaded_zip.name)[0]
+        camera_name = os.path.splitext(zip_name)[0]
         camera_folders.append((camera_name, camera_folder))
     
     # Build a mapping for each camera: { frame_index: file_full_path }
@@ -111,11 +114,11 @@ def process_zips(camera_zips, regex_pattern, start_frame, end_frame):
     
     return output_zip_path
 
-# Create the Gradio interface
+# Create the Gradio interface with type "binary" for file uploads
 iface = gr.Interface(
     fn=process_zips,
     inputs=[
-        gr.File(label="Camera Folder ZIPs", file_count="multiple", type="bytes"),
+        gr.File(label="Camera Folder ZIPs", file_count="multiple", type="binary"),
         gr.Textbox(label="Regex Pattern (with one capturing group)"),
         gr.Textbox(label="Start Frame Number"),
         gr.Textbox(label="End Frame Number")