Update app.py

app.py

@@ -4,6 +4,7 @@ from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 import torch
 from PIL import Image, ImageOps
 import gradio as gr
+import user_history
 
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipe = StableDiffusionXLPipeline.from_pretrained(
@@ -297,7 +298,7 @@ def rotate_output(has_flipped):
     else:
         return gr.Image(elem_classes="rotated"), gr.Button("Rotate to see prompt 1!"), not has_flipped
 
-def simple_call(prompt1, prompt2):
+def simple_call(prompt1, prompt2, profile: gr.OAuthProfile | None=None):
     generator = [torch.Generator(device="cuda").manual_seed(5)]
     res = call(
         pipe,
@@ -314,12 +315,20 @@ def simple_call(prompt1, prompt2):
         generator=generator
     )
     image1 = res.images[0]
+
+    # save generated images (if logged in)
+    user_history.save_image(label=f"{prompt1} / {prompt2}", image=image1, profile=profile, metadata={
+        "prompt2": prompt1,
+        "prompt1": prompt2,
+        "seed": seed,
+    })
+
     return image1
 css = '''
 #result_image{ transition: transform 2s ease-in-out }
 #result_image.rotated{transform: rotate(180deg)}
 '''
-with gr.Blocks(css=css) as app:
+with gr.Blocks() as app:
     gr.Markdown(
         '''
         <center>
@@ -362,7 +371,13 @@ with gr.Blocks(css=css) as app:
         show_progress=False
     )
 
-app.queue(max_size=20)
+with gr.Blocks(css=css) as app_with_history:
+    with gr.Tab("Upside Down Diffusion"):
+        app.render()
+    with gr.Tab("Past generations"):
+        user_history.render()
+
+app_with_history.queue(max_size=20)
 
 if __name__ == "__main__":
-    app.launch(debug=True)
+    app_with_history.launch(debug=True)