Update app.py

app.py

@@ -159,7 +159,7 @@ with gr.Blocks() as demo:
 
     with gr.Tab("Add Entry"):
         with gr.Row():
-            image_input = gr.Image(label="Upload Image")
+            image_input = gr.Image(label="Upload Image", type="numpy")
             prompt_input = gr.Textbox(label="Prompt")
         add_button = gr.Button("Add Entry")
 
@@ -189,7 +189,6 @@ with gr.Blocks() as demo:
 
     with gr.Tab("Edit / Delete Entry"):
         with gr.Column():
-            # Entry selector is referenced here, defined at the top level
             selected_image = gr.Image(label="Selected Image", interactive=False)
             selected_prompt = gr.Textbox(label="Current Prompt", interactive=False)
             new_prompt_input = gr.Textbox(label="New Prompt (for Edit)")
@@ -256,15 +255,20 @@ with gr.Blocks() as demo:
 
     with gr.Tab("Download Dataset"):
         download_button = gr.Button("Download Dataset")
-        download_output = gr.File(label="Download Zip")
+        download_output = gr.File(label="Download Zip", interactive=False)
         def download_dataset(current_dataset_name, datasets):
             if not current_dataset_name:
                 return None, "No dataset selected."
             if not datasets[current_dataset_name]:
                 return None, "Dataset is empty."
             zip_buffer = save_dataset_to_zip(current_dataset_name, datasets[current_dataset_name])
-            # Return a dictionary with 'name' and 'data'
-            return {'name': f"{current_dataset_name}.zip", 'data': zip_buffer.getvalue()}, f"Dataset '{current_dataset_name}' is ready for download."
+            # Write zip_buffer to a temporary file
+            temp_dir = tempfile.mkdtemp()
+            zip_path = os.path.join(temp_dir, f"{current_dataset_name}.zip")
+            with open(zip_path, 'wb') as f:
+                f.write(zip_buffer.getvalue())
+            return zip_path, f"Dataset '{current_dataset_name}' is ready for download."
+
         download_button.click(
             download_dataset,
             inputs=[current_dataset_name, datasets],