Update app.py

app.py

@@ -116,7 +116,7 @@ with gr.Blocks() as demo:
         if name in datasets:
             return gr.update(), f"Dataset '{name}' already exists."
         datasets[name] = []
-        return gr.Dropdown.update(choices=list(datasets.keys()), value=name), f"Dataset '{name}' created."
+        return gr.update(choices=list(datasets.keys()), value=name), f"Dataset '{name}' created."
 
     create_button.click(
         create_dataset,
@@ -131,7 +131,7 @@ with gr.Blocks() as demo:
         if dataset_name in datasets:
             return gr.update(), f"Dataset '{dataset_name}' already exists."
         datasets[dataset_name] = dataset
-        return gr.Dropdown.update(choices=list(datasets.keys()), value=dataset_name), f"Dataset '{dataset_name}' uploaded."
+        return gr.update(choices=list(datasets.keys()), value=dataset_name), f"Dataset '{dataset_name}' uploaded."
 
     upload_button.click(
         upload_dataset,
@@ -143,9 +143,9 @@ with gr.Blocks() as demo:
         if dataset_name in datasets:
             dataset = datasets[dataset_name]
             html_content = display_dataset_html(dataset)
-            return current_dataset_name.update(value=dataset_name), dataset_html.update(value=html_content), ""
+            return current_dataset_name.update(value=dataset_name), gr.update(value=html_content), ""
         else:
-            return current_dataset_name.update(value=""), dataset_html.update(value="<div>Select a dataset.</div>"), ""
+            return current_dataset_name.update(value=""), gr.update(value="<div>Select a dataset.</div>"), ""
 
     dataset_selector.change(
         select_dataset,
@@ -161,12 +161,12 @@ with gr.Blocks() as demo:
 
     def add_entry(image_data, prompt, current_dataset_name, datasets):
         if not current_dataset_name:
-            return datasets, dataset_html.update(), "No dataset selected."
+            return datasets, gr.update(), "No dataset selected."
         if image_data is None or not prompt:
-            return datasets, dataset_html.update(), "Please provide both an image and a prompt."
+            return datasets, gr.update(), "Please provide both an image and a prompt."
         datasets[current_dataset_name].append({'image': image_data, 'prompt': prompt})
         html_content = display_dataset_html(datasets[current_dataset_name])
-        return datasets, dataset_html.update(value=html_content), f"Entry added to dataset '{current_dataset_name}'."
+        return datasets, gr.update(value=html_content), f"Entry added to dataset '{current_dataset_name}'."
 
     add_button.click(
         add_entry,
@@ -183,16 +183,16 @@ with gr.Blocks() as demo:
 
     def edit_entry(index, new_prompt, current_dataset_name, datasets):
         if not current_dataset_name:
-            return datasets, dataset_html.update(), "No dataset selected."
-        if index is None or not new_prompt:
-            return datasets, dataset_html.update(), "Please provide both index and new prompt."
+            return datasets, gr.update(), "No dataset selected."
+        if index is None or new_prompt is None or new_prompt.strip() == '':
+            return datasets, gr.update(), "Please provide both index and new prompt."
         index = int(index)
         if 0 <= index < len(datasets[current_dataset_name]):
             datasets[current_dataset_name][index]['prompt'] = new_prompt
             html_content = display_dataset_html(datasets[current_dataset_name])
-            return datasets, dataset_html.update(value=html_content), f"Entry {index} updated."
+            return datasets, gr.update(value=html_content), f"Entry {index} updated."
         else:
-            return datasets, dataset_html.update(), "Invalid index."
+            return datasets, gr.update(), "Invalid index."
 
     edit_button.click(
         edit_entry,
@@ -202,16 +202,16 @@ with gr.Blocks() as demo:
 
     def delete_entry(index, current_dataset_name, datasets):
         if not current_dataset_name:
-            return datasets, dataset_html.update(), "No dataset selected."
+            return datasets, gr.update(), "No dataset selected."
         if index is None:
-            return datasets, dataset_html.update(), "Please provide an index."
+            return datasets, gr.update(), "Please provide an index."
         index = int(index)
         if 0 <= index < len(datasets[current_dataset_name]):
             del datasets[current_dataset_name][index]
             html_content = display_dataset_html(datasets[current_dataset_name])
-            return datasets, dataset_html.update(value=html_content), f"Entry {index} deleted."
+            return datasets, gr.update(value=html_content), f"Entry {index} deleted."
         else:
-            return datasets, dataset_html.update(), "Invalid index."
+            return datasets, gr.update(), "Invalid index."
 
     delete_button.click(
         delete_entry,
@@ -233,11 +233,14 @@ with gr.Blocks() as demo:
             outputs=[download_output, message_box]
         )
 
-    # Initially update dataset_html
+    # Initially update dataset_selector
+    def initialize_datasets(datasets):
+        return gr.update(choices=list(datasets.keys()))
+    
     demo.load(
-        lambda datasets: gr.Dropdown.update(choices=list(datasets.keys())),
-        inputs=None,
-        outputs=dataset_selector
+        initialize_datasets,
+        inputs=[datasets],
+        outputs=[dataset_selector]
     )
 
 demo.launch()