[Feature] Add User history (#5)

- [Feature] Add User history (00d5d79e86d537b53c780c310b2fd7844023a138)
- Update README.md (a5d9316ae94fabc5582c2436c483f276d5f6e66f)
- Update requirements.txt (ad9844afd88e673012943650288abca23f71b01e)


Co-authored-by: Lucain Pouget <[email protected]>

README.md

@@ -7,6 +7,7 @@ sdk: gradio
 sdk_version: 3.44.0
 app_file: app.py
 pinned: false
+hf_oauth: true
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -4,7 +4,7 @@ import torch
 
 from diffusers import DiffusionPipeline
 from free_lunch_utils import register_free_upblock2d, register_free_crossattn_upblock2d
-
+import gradio_user_history as gr_user_history
 
 
 model_id = "stabilityai/stable-diffusion-2-1"
@@ -21,7 +21,7 @@ sd_options_prev = None
 seed_prev = None 
 sd_image_prev = None
 
-def infer(prompt, sd_options, seed, b1, b2, s1, s2):
+def infer(prompt, sd_options, seed, b1, b2, s1, s2, profile: gr.OAuthProfile | None):
     global prompt_prev
     global sd_options_prev
     global seed_prev
@@ -65,6 +65,9 @@ def infer(prompt, sd_options, seed, b1, b2, s1, s2):
     # First SD, then freeu
     images = [sd_image, freeu_image]
 
+    gr_user_history.save_image(label=prompt + ' (SD)', image=sd_image, profile=profile, metadata={"prompt": prompt, "pipe": sd_options, "b1": 1.0, "b2": 1.0, "s1": 1.0, "s2": 1.0})
+    gr_user_history.save_image(label=prompt + ' (FreeU)', image=freeu_image, profile=profile, metadata={"prompt": prompt, "pipe": "freeu", "b1": b1, "b2": b2, "s1": s1, "s2": s2})
+
     return images
 
 
@@ -209,7 +212,12 @@ with block:
                              maximum=1000,
                              step=1,
                              value=42)
-                    
+
+    with gr.Group():
+        with gr.Row():
+            with gr.Accordion("Past generations", open=False):
+                gr_user_history.render()
+    
     with gr.Row():
         with gr.Group():
             # btn = gr.Button("Generate image", scale=0)

requirements.txt

@@ -5,4 +5,6 @@ xformers==0.0.21
 git+https://github.com/huggingface/open-muse.git@patrickvonplaten-patch-1
 
 --extra-index-url https://download.pytorch.org/whl/cu118
-torch
+torch
+
+git+https://huggingface.co/spaces/Wauplin/gradio-user-history