Update app.py

app.py

@@ -14,6 +14,7 @@ import copy
 import gc
 import pickle
 import spaces
+import user_history
 
 lora_list = hf_hub_download(repo_id="multimodalart/LoraTheExplorer", filename="sdxl_loras.json", repo_type="space")
 
@@ -93,7 +94,7 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
 original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
 
 @spaces.GPU
-def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, seed=-1):
+def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, seed=-1, profile: gr.OAuthProfile | None=None):
   
   repo_id_1 = shuffled_items[0]['repo']
   repo_id_2 = shuffled_items[1]['repo']
@@ -134,6 +135,18 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
       seed = random.randint(0, 2147483647)
   generator = torch.Generator(device="cuda").manual_seed(seed)
   image = pipe(prompt=prompt, negative_prompt=negative_prompt, num_inference_steps=20, width=768, height=768, generator=generator).images[0]
+
+  # save generated images (if logged in)
+  user_history.save_image(label=prompt, image=image, profile=profile, metadata={
+    "prompt": prompt,
+    "negative_prompt": negative_prompt,
+    "lora_1_repo_id": shuffled_items[0]['repo'],
+    "lora_2_repo_id": shuffled_items[1]['repo'],
+    "lora_1_scale": lora_1_scale,
+    "lora_2_scale": lora_2_scale,
+    "seed": seed,
+  })
+
   return image, gr.update(visible=True), seed, gr.update(visible=True, interactive=True), gr.update(visible=False), gr.update(visible=True, interactive=True), gr.update(visible=False)
 
 def get_description(item):
@@ -176,7 +189,7 @@ def save_preferences(lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, g
 def hide_post_gen_info():
     return gr.update(visible=False)
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks() as demo:
   shuffled_items = gr.State()
   title = gr.HTML(
         '''<h1>LoRA Roulette 🎰</h1>
@@ -245,5 +258,11 @@ with gr.Blocks(css=css) as demo:
   thumbs_down.click(save_preferences, inputs=[lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, output_image, gr.State("down"), last_used_seed], outputs=[thumbs_down, thumbs_down_clicked, thumbs_up])
   share_button.click(None, [], [], _js=share_js)
 
-demo.queue(concurrency_count=2)
-demo.launch()
+with gr.Blocks(css=css) as demo_with_history:
+    with gr.Tab("Lora Roulette"):
+        demo.render()
+    with gr.Tab("Past generations"):
+        user_history.render()
+
+demo_with_history.queue(concurrency_count=2)
+demo_with_history.launch()