history v1

app.py

@@ -215,10 +215,7 @@ generation_history = []
 
 # Function to update the history list
 def update_history_list():
-    return [
-        gr.Image.update(value=item["image"], visible=True)
-        for item in generation_history
-    ]
+    return [item["image"] for item in generation_history]
 
 # Function to handle image click in history
 def handle_image_click(evt: gr.SelectData):
@@ -357,11 +354,18 @@ with gr.Blocks(css="style.css") as demo:
         generate_from_json = gr.Button("Generate from JSON")
 
     # Add history accordion
-    with gr.Accordion("Generation History", open=False) as history_accordion:
-        history_gallery = gr.Gallery(label="History", show_label=False, elem_id="history_gallery").style(grid=5)
-        with gr.Row():
-            selected_image = gr.Image(label="Selected Image", interactive=False)
-            selected_metadata = gr.JSON(label="Selected Metadata", show_label=False)
+with gr.Accordion("Generation History", open=False) as history_accordion:
+    history_gallery = gr.Gallery(
+        label="History",
+        show_label=False,
+        elem_id="history_gallery",
+        columns=5,
+        rows=2,
+        height="auto"
+    )
+    with gr.Row():
+        selected_image = gr.Image(label="Selected Image", interactive=False)
+        selected_metadata = gr.JSON(label="Selected Metadata", show_label=False)
 
     gr.Examples(
         examples=config.examples,