Revert

app.py

@@ -219,34 +219,26 @@ def generate(
         pipe.scheduler = backup_scheduler
         utils.free_memory()
 
-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
+generation_history = []
 
 def update_history_list():
-    return fetch_recent_images()
+    return [item["image"] for item in generation_history]
 
 def handle_image_click(evt: gr.SelectData):
-    selected = fetch_recent_images()[evt.index]
+    selected = generation_history[evt.index]
     return selected["image"], json.dumps(selected["metadata"], indent=2)
 
 def generate_and_update_history(*args, **kwargs):
     images, metadata = generate(*args, **kwargs)
-    utils.save_image(images[0], metadata, OUTPUT_DIR)
+    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()
     return images[0], json.dumps(metadata, indent=2), update_history_list()
 
 with open('characterfull.txt', 'r') as f:
@@ -380,13 +372,12 @@ 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=4,
+            rows=2,
             height="auto"
         )
         with gr.Row():
@@ -492,11 +483,5 @@ 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)