Update app.py

app.py

@@ -214,8 +214,9 @@ with gr.Blocks() as demo:
                 img_data = f"data:image/png;base64,{img_str}"
                 datasets[current_dataset_name].append({'image': img_data, 'prompt': prompt})
                 dataset = datasets[current_dataset_name]
-                # Reset page number to 0 when a new entry is added
+                # Reset page number to 0 and refresh HTML
                 page_number = 0
+                dataset = datasets[current_dataset_name]
                 html_content = display_dataset_html(dataset, page_number=page_number)
                 return datasets, page_number, gr.update(value=html_content), f"Entry added to dataset '{current_dataset_name}'."