Update app.py

app.py

@@ -53,14 +53,14 @@ def save_dataset_to_zip(dataset_name, dataset):
     zip_buffer.seek(0)
     return zip_buffer
 
-def load_dataset_from_zip(zip_file):
+def load_dataset_from_zip(zip_file_path):
     temp_dir = tempfile.mkdtemp()
     try:
-        with zipfile.ZipFile(zip_file.name, 'r') as zip_ref:
+        with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
             zip_ref.extractall(temp_dir)
 
         # Get dataset name from zip file name
-        dataset_name_guess = os.path.splitext(os.path.basename(zip_file.name))[0]
+        dataset_name_guess = os.path.splitext(os.path.basename(zip_file_path))[0]
         dataset_path = os.path.join(temp_dir, dataset_name_guess)
 
         if os.path.exists(dataset_path):
@@ -142,7 +142,7 @@ with gr.Blocks() as demo:
                 create_button = gr.Button("Create Dataset")
             with gr.Column():
                 gr.Markdown("### Upload Existing Dataset")
-                upload_input = gr.File(label="Upload Dataset Zip", type="file")
+                upload_input = gr.File(label="Upload Dataset Zip", type="filepath")
                 upload_button = gr.Button("Upload Dataset")
 
     def create_dataset(name, datasets):
@@ -159,10 +159,10 @@ with gr.Blocks() as demo:
         outputs=[dataset_selector, message_box]
     )
 
-    def upload_dataset(zip_file, datasets):
-        if zip_file is None:
+    def upload_dataset(zip_file_path, datasets):
+        if not zip_file_path:
             return gr.update(), "Please upload a zip file."
-        dataset_name, dataset = load_dataset_from_zip(zip_file)
+        dataset_name, dataset = load_dataset_from_zip(zip_file_path)
         if dataset_name is None:
             return gr.update(), "Failed to load dataset from zip file."
         if dataset_name in datasets:
@@ -224,7 +224,7 @@ with gr.Blocks() as demo:
 
     with gr.Tab("Edit / Delete Entry"):
         with gr.Column():
-            selected_image = gr.Image(label="Selected Image", interactive=False)
+            selected_image = gr.Image(label="Selected Image", interactive=False, type="numpy")
             selected_prompt = gr.Textbox(label="Current Prompt", interactive=False)
             new_prompt_input = gr.Textbox(label="New Prompt (for Edit)")
             with gr.Row():