Upload folder using huggingface_hub

app.py

@@ -399,8 +399,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     performance_plot = gr.Plot(label="Performance Metrics")
 
-    with gr.Accordion("Recent Generations (last 32)", open=False):
-        recent_gallery = gr.Gallery(label="Prompt and Output", columns=3, rows=1, show_label=True)
+    with gr.Accordion("Recent Generations (last 8)", open=False):
+        recent_gallery = gr.Gallery(
+            label="Recent Generations",
+            show_label=False,
+            elem_id="recent-gallery",
+            columns=4,
+            height="auto",
+        ).style(grid=2)
 
     def update_recent_gallery(prompt, results):
         recent_generations.append({
@@ -409,13 +415,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             "hidream-dev": results["hidream-dev"],
             "hidream-full": results["hidream-full"],
         })
-        if len(recent_generations) > 32:
+        if len(recent_generations) > 8:
             recent_generations.pop(0)
-        gallery_items = [
-            (r["prompt"], r["flux-dev"], r["hidream-dev"], r["hidream-full"])
-            for r in reversed(recent_generations)
-        ]
-        return gr.update(value=gallery_items)
+        
+        # Flatten the images from each generation
+        all_images = []
+        for gen in reversed(recent_generations):  # Show newest first
+            all_images.extend([
+                gen["flux-dev"],
+                gen["hidream-dev"],
+                gen["hidream-full"]
+            ])
+
+        # Update the Gradio gallery with the new list
+        return recent_gallery.update(value=all_images)
 
     # Add custom CSS for the big status box
     css = """