Update app.py

app.py

@@ -4,7 +4,6 @@ import spaces
 import torch
 import random
 import os
-import time
 
 # from diffusers import QwenImageEditInpaintPipeline
 from optimization import optimize_pipeline_
@@ -143,26 +142,6 @@ def clear_result():
     """Clears the result image."""
     return gr.update(value=None)
 
-def update_history(new_image, history):
-    """Updates the history gallery with the new image."""
-    time.sleep(0.5)  # Small delay to ensure image is ready
-    if history is None:
-        history = []
-    if new_image is not None:
-        # Convert to list if needed (Gradio sometimes returns tuples)
-        if not isinstance(history, list):
-            history = list(history) if history else []
-        history.insert(0, new_image)
-    # Keep only the last 20 images in history
-    history = history[:20]
-    return history
-
-def use_history_as_input(evt: gr.SelectData, history):
-    """Sets the selected history image as the new input image."""
-    if history and evt.index < len(history):
-        return gr.update(value=history[evt.index][0])
-    return gr.update()
-
 def use_output_as_input(output_image):
     """Sets the generated output as the new input image."""
     if output_image is not None:
@@ -262,9 +241,6 @@ css = """
     width: 400px;
 }
 #edit_text{margin-top: -62px !important}
-.gallery-container {
-    margin-top: 20px;
-}
 """
 
 
@@ -317,23 +293,6 @@ with gr.Blocks(css=css) as demo:
                 result = gr.ImageSlider(label="Result", show_label=False)
                 
                 use_as_input_button = gr.Button("🔄 Use as Input Image", visible=False, variant="secondary")
-                
-                gr.Markdown("---")
-                
-                with gr.Row():
-                    gr.Markdown("### 📜 History")
-                    clear_history_button = gr.Button("🗑️ Clear History", size="sm", variant="stop")
-                
-                history_gallery = gr.Gallery(
-                    label="Click any image to use as input", 
-                    columns=4, 
-                    rows=2,
-                    object_fit="contain", 
-                    height="auto",
-                    interactive=False,
-                    show_label=True,
-                    elem_classes=["gallery-container"]
-                )
         
         with gr.Accordion("Advanced Settings", open=False):
             
@@ -386,22 +345,8 @@ with gr.Blocks(css=css) as demo:
         outputs=[edit_image],
         show_api=False
     )
-    
-    history_gallery.select(
-        fn=use_history_as_input,
-        inputs=[history_gallery],
-        outputs=[edit_image],
-        show_api=False
-    )
-    
-    clear_history_button.click(
-        fn=lambda: [],
-        inputs=None,
-        outputs=history_gallery,
-        show_api=False
-    )
 
-    # Main generation pipeline with result clearing, history update, and button visibility
+    # Main generation pipeline with result clearing and button visibility
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=clear_result,
@@ -417,11 +362,6 @@ with gr.Blocks(css=css) as demo:
         inputs=None,
         outputs=use_as_input_button,
         show_api=False
-    ).then(
-        fn=update_history,
-        inputs=[result, history_gallery],
-        outputs=history_gallery,
-        show_api=False
     )
 
 demo.launch()