Update app.py

app.py

@@ -8,6 +8,8 @@ import os
 import spaces
 import random
 
+import user_history
+
 is_shared_ui = True if "fffiloni/sd-xl-lora-fusion" in os.environ['SPACE_ID'] else False
 hf_token = os.environ.get("HF_TOKEN")
 login(token = hf_token)
@@ -75,7 +77,7 @@ def load_sfts(repo_1_id, repo_2_id):
     return gr.update(choices=sfts_available_files_1, value=sfts_available_files_1[0], visible=True), gr.update(choices=sfts_available_files_2, value=sfts_available_files_2[0], visible=True), gr.update(value=instance_prompt_1, visible=True), gr.update(value=instance_prompt_2, visible=True)
     
 @spaces.GPU
-def infer(lora_1_id, lora_1_sfts, lora_2_id, lora_2_sfts, prompt, negative_prompt, lora_1_scale, lora_2_scale, seed):
+def infer(lora_1_id, lora_1_sfts, lora_2_id, lora_2_sfts, prompt, negative_prompt, lora_1_scale, lora_2_scale, seed, profile: gr.OAuthProfile | None):
 
     unet = copy.deepcopy(original_pipe.unet)
     text_encoder = copy.deepcopy(original_pipe.text_encoder)
@@ -148,7 +150,18 @@ def infer(lora_1_id, lora_1_sfts, lora_2_id, lora_2_sfts, prompt, negative_promp
     ).images[0]
 
     pipe.unfuse_lora()
-    
+
+    # 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": lora_1_id,
+        "lora_2_repo_id": lora_2_id,
+        "lora_1_scale": lora_1_scale,
+        "lora_2_scale": lora_2_scale,
+        "seed": seed,
+    })
+
     return image, seed
 
 css="""
@@ -290,6 +303,9 @@ with gr.Blocks(css=css) as demo:
                 label = "Output"
             )
 
+        with gr.Accordion("Past generations", open=False):
+            user_history.render()
+
         # Advanced Settings
         with gr.Accordion("Advanced Settings", open=False):