Added History (#1)

- Added History (9097f863eef879fdaa9f285a5573b4e16ae78a10)


Co-authored-by: Nishith Jain <[email protected]>

app.py

@@ -270,7 +270,7 @@ def remove_custom_lora(selected_indices, current_loras):
         lora_image_2
     )
 
-@spaces.GPU(duration=70)
+@spaces.GPU(duration=90)
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
     print("Generating image...")
     pipe.to("cuda")
@@ -291,7 +291,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
             yield img
     pipe.to("cpu")
 
-@spaces.GPU(duration=70)
+@spaces.GPU(duration=90)
 def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, seed):
     pipe_i2i.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
@@ -311,7 +311,7 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
     pipe_i2i.to("cpu")
     return final_image
 
-@spaces.GPU(duration=70)
+@spaces.GPU(duration=90)
 def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
     if not selected_indices:
         raise gr.Error("You must select at least one LoRA before proceeding.")
@@ -435,6 +435,13 @@ def check_custom_model(link):
         # Assume it's a Hugging Face model path
         return get_huggingface_safetensors(link)
 
+def update_history(new_image, history):
+    """Updates the history gallery with the new image."""
+    if history is None:
+        history = []
+    history.insert(0, new_image)
+    return history
+
 css = '''
 #gen_btn{height: 100%}
 #title{text-align: center}
@@ -514,6 +521,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 3600)) as app:
         with gr.Column():
             progress_bar = gr.Markdown(elem_id="progress", visible=False)
             result = gr.Image(label="Generated Image")
+            history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain")
+
     with gr.Row():
         with gr.Accordion("Advanced Settings", open=False):
             with gr.Row():
@@ -566,6 +575,10 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 3600)) as app:
         fn=run_lora,
         inputs=[prompt, input_image, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state],
         outputs=[result, seed, progress_bar]
+    ).then(  # Update the history gallery
+        fn=lambda x, history: update_history(x[1], history),
+        inputs=[result, history_gallery],
+        outputs=history_gallery,
     )
 
 app.queue()