v3p1

app.py

@@ -219,26 +219,34 @@ def generate(
         pipe.scheduler = backup_scheduler
         utils.free_memory()
 
-generation_history = []
+def fetch_recent_images(num_images=20):
+    image_files = [f for f in os.listdir(OUTPUT_DIR) if f.endswith('.png')]
+    image_files.sort(key=lambda x: os.path.getmtime(os.path.join(OUTPUT_DIR, x)), reverse=True)
+    recent_images = []
+    
+    for img_file in image_files[:num_images]:
+        img_path = os.path.join(OUTPUT_DIR, img_file)
+        img = Image.open(img_path)
+        metadata = utils.get_image_metadata(img_path)
+        recent_images.append({
+            "image": img_path,
+            "prompt": metadata.get("prompt", ""),
+            "timestamp": datetime.fromtimestamp(os.path.getmtime(img_path)).strftime("%Y-%m-%d %H:%M:%S"),
+            "metadata": metadata
+        })
+    
+    return recent_images
 
 def update_history_list():
-    return [item["image"] for item in generation_history]
+    return fetch_recent_images()
 
 def handle_image_click(evt: gr.SelectData):
-    selected = generation_history[evt.index]
+    selected = fetch_recent_images()[evt.index]
     return selected["image"], json.dumps(selected["metadata"], indent=2)
 
 def generate_and_update_history(*args, **kwargs):
     images, metadata = generate(*args, **kwargs)
-    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    generation_history.insert(0, {
-        "prompt": metadata["prompt"],
-        "timestamp": timestamp,
-        "image": images[0],
-        "metadata": metadata
-    })
-    if len(generation_history) > 20:
-        generation_history.pop()
+    utils.save_image(images[0], metadata, OUTPUT_DIR)
     return images[0], json.dumps(metadata, indent=2), update_history_list()
 
 with open('characterfull.txt', 'r') as f:
@@ -372,12 +380,13 @@ with gr.Blocks(css="style.css") as demo:
         generate_from_json = gr.Button("Generate from JSON")
 
     with gr.Accordion("Generation History", open=False) as history_accordion:
+        update_gallery_button = gr.Button("Update Gallery")
         history_gallery = gr.Gallery(
             label="History",
             show_label=False,
             elem_id="history_gallery",
             columns=5,
-            rows=2,
+            rows=4,
             height="auto"
         )
         with gr.Row():
@@ -483,5 +492,11 @@ with gr.Blocks(css="style.css") as demo:
         inputs=[],
         outputs=[selected_image, selected_metadata]
     )
-    
+
+    update_gallery_button.click(
+        fn=update_history_list,
+        inputs=[],
+        outputs=[history_gallery]
+    )
+
 demo.queue(max_size=20).launch(debug=IS_COLAB, share=IS_COLAB)