Update app.py

app.py

@@ -10,7 +10,7 @@ from diffusers import WuerstchenDecoderPipeline, WuerstchenPriorPipeline
 from diffusers.pipelines.wuerstchen import DEFAULT_STAGE_C_TIMESTEPS
 from previewer.modules import Previewer
 
-from gallery_history import fetch_gallery_history, show_gallery_history
+import user_history
 
 os.environ['TOKENIZERS_PARALLELISM'] = 'false'
 
@@ -79,6 +79,7 @@ def generate(
     # decoder_timesteps: List[float] = None,
     decoder_guidance_scale: float = 0.0,
     num_images_per_prompt: int = 2,
+    profile: gr.OAuthProfile | None = None,
 ) -> PIL.Image.Image:
     generator = torch.Generator().manual_seed(seed)
 
@@ -111,6 +112,25 @@ def generate(
         generator=generator,
         output_type="pil",
     ).images
+
+    # Save images
+    for image in decoder_output:
+        user_history.save_image(
+            profile=profile,
+            image=image,
+            label=prompt,
+            metadata={
+                "negative_prompt": negative_prompt,
+                "seed": seed,
+                "width": width,
+                "height": height,
+                "prior_guidance_scale": prior_guidance_scale,
+                "decoder_num_inference_steps": decoder_num_inference_steps,
+                "decoder_guidance_scale": decoder_guidance_scale,
+                "num_images_per_prompt": num_images_per_prompt,
+            },
+        )
+
     yield decoder_output
 
 
@@ -119,7 +139,7 @@ examples = [
     "An astronaut riding a green horse",
 ]
 
-with gr.Blocks(css="style.css") as demo:
+with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
     gr.DuplicateButton(
         value="Duplicate Space for private use",
@@ -213,8 +233,6 @@ with gr.Blocks(css="style.css") as demo:
         cache_examples=CACHE_EXAMPLES,
     )
 
-    history = show_gallery_history()
-
     inputs = [
             prompt,
             negative_prompt,
@@ -240,8 +258,6 @@ with gr.Blocks(css="style.css") as demo:
         inputs=inputs,
         outputs=result,
         api_name="run",
-    ).then(
-        fn=fetch_gallery_history, inputs=[prompt, result], outputs=history, queue=False
     )
     negative_prompt.submit(
         fn=randomize_seed_fn,
@@ -254,8 +270,6 @@ with gr.Blocks(css="style.css") as demo:
         inputs=inputs,
         outputs=result,
         api_name=False,
-    ).then(
-        fn=fetch_gallery_history, inputs=[prompt, result], outputs=history, queue=False
     )
     run_button.click(
         fn=randomize_seed_fn,
@@ -268,9 +282,13 @@ with gr.Blocks(css="style.css") as demo:
         inputs=inputs,
         outputs=result,
         api_name=False,
-    ).then(
-        fn=fetch_gallery_history, inputs=[prompt, result], outputs=history, queue=False
     )
 
+with gr.Blocks(css="style.css") as demo_with_history:
+    with gr.Tab("App"):
+        demo.render()
+    with gr.Tab("Past generations"):
+        user_history.render()
+
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    demo_with_history.queue(max_size=20).launch()