Update app.py

app.py

@@ -124,10 +124,19 @@ def display_dataset_html(dataset, page_number=0, items_per_page=2):
             idx = start_idx + idx_offset
             image_data = entry['image']
             prompt = entry['prompt']
+            # Decode base64 image data to numpy array
+            image_bytes = base64.b64decode(image_data.split(",")[1])
+            image = Image.open(BytesIO(image_bytes))
+            # Compress image
+            image.thumbnail((100, 100))  # Resize image to 100x100 pixels
+            buffered = BytesIO()
+            image.save(buffered, format="PNG")
+            img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
+            img_data = f"data:image/png;base64,{img_str}"
             html_content += f"""
             <div style="display: flex; flex-direction: column; align-items: center; margin-right: 20px;">'''
             <div style="margin-bottom: 5px;">{idx}</div>
-            <img src="{image_data}" alt="Image {idx}" style="max-height: 150px;"/>
+            <img src="{img_data}" alt="Image {idx}" style="max-height: 150px;"/>
             <div style="max-width: 150px; word-wrap: break-word; text-align: center;">{prompt}</div>
             </div>
             """
@@ -444,7 +453,7 @@ with gr.Blocks() as demo:
 
     # Update all components when an entry is deleted
     def update_all_components_after_delete(current_dataset_name, datasets):
-        time.sleep(2)  # Add a small delay to allow other componenets to finish processing
+        time.sleep(0.1)  # Add a small delay
         dataset = datasets[current_dataset_name]
         html_content = display_dataset_html(dataset, page_number=0)
         entry_options = [f"{idx}: {entry['prompt'][:30]}" for idx, entry in enumerate(dataset)]