test gallery clearing

app.py

@@ -7,19 +7,16 @@ import os
 import warnings
 from dotenv import load_dotenv
 
-
+# load secrets
 load_dotenv()
 SD_KEY = os.getenv("SD_KEY")
 HF_KEY = os.getenv("HF_KEY")
 USERNAME = os.getenv("USERNAME")
 PASS = os.getenv("PASS")
-CD_KEY = os.getenv("CD_KEY")
-
 
+# set up dataset writer
 hf_writer_gen = gr.HuggingFaceDatasetSaver(HF_KEY, "helms/master-thesis-generated-images-1", private=True, separate_dirs=False)
 
-
-
 def infer(prompt):
     stability_api = client.StabilityInference(
         key=SD_KEY,  # AaPI Key reference.
@@ -44,19 +41,21 @@ def infer(prompt):
         # Defaults to k_dpmpp_2m if not specified. Clip Guidance only supports ancestral samplers.
         # (Available Samplers: ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_dpmpp_2s_ancestral, k_lms, k_dpmpp_2m)
     )
-    infer.results = []
+    #infer.results = []
     for resp in answers:
         for artifact in resp.artifacts:
             if artifact.finish_reason == generation.FILTER:
                 warnings.warn(
-                    "Your request activated the API's safety filters and could not be processed."
+                    "Your request activated the APIs safety filters and could not be processed."
                     "Please modify the prompt and try again.")
             if artifact.type == generation.ARTIFACT_IMAGE:
                 img = Image.open(io.BytesIO(artifact.binary))
+                #infer.results.insert(0, img)
                 return img
 
 
 with gr.Blocks(title="Master Thesis Image Generator") as demo:
+    storage = gr.State(value=[])
     with gr.Row():
         gr.HTML(""" 
                     <div style="text-align: justify; margin: 0 auto;">
@@ -81,14 +80,15 @@ with gr.Blocks(title="Master Thesis Image Generator") as demo:
     with gr.Row(equal_height=True):
         with gr.Column(scale=1):
             user = gr.Textbox(placeholder="Please enter your name.", label="Name")
-            inp = gr.Textbox(placeholder="Enter your prompt here.", lines=3, label="Prompt")
-            btn_gen = gr.Button("Generate Images", variant="primary", size="lg", visible=True)
+            inp = gr.Textbox(placeholder="Enter your prompt here.", lines=5, label="Prompt")
+            btn_gen = gr.Button("Generate Image", variant="primary", size="lg", visible=True)
             with gr.Box():
-                gr.HTML("""<div style="text-align: center; margin: 0 auto; padding-bottom: 10px"><p style="font-size:10pt">Generated Images</p></div>""")
-                store = gr.Gallery(visible=True, show_share_button=False, allow_preview=False, columns=1)
+                gr.HTML(
+                    """<div style="text-align: center; margin: 0 auto; padding-bottom: 10px"><p style="font-size:10pt">Generated Images</p></div>""")
+                store = gr.Gallery(value=storage.value, visible=True, show_share_button=False, allow_preview=False, columns=1)
+                gr.ClearButton(store)
         with gr.Column(scale=3):
-            out = gr.Image(interactive=False, visible=True, container=True, width=1024,
-                            show_share_button=False)
+            out = gr.Image(interactive=False, visible=True, container=True, width=1024, show_share_button=False)
     with gr.Row():
         with gr.Accordion("See Examples:", open=False):
             examples = [
@@ -98,19 +98,28 @@ with gr.Blocks(title="Master Thesis Image Generator") as demo:
                 ["Beautiful waterfall in a lush jungle, with sunlight shining through the trees"]
             ]
             ex = gr.Examples(examples=examples, inputs=inp)
+
     def make_visible():
         return gr.update(visible=True)
+
     def make_invisible():
         return gr.update(visible=False)
+
     def make_inactive():
         return gr.update(interactive=False)
+
     def make_active():
         return gr.update(interactive=True)
 
     def add_to_gallery(img):
-        gall = store.value
+        gall = storage.value
         gall.insert(0, img)
         return gall
+
+    def clear_gallery():
+        gall = []
+        return gall, gall
+
     def select_img(evt: gr.SelectData):
         sel_img = store.value[evt.index]
         return sel_img
@@ -133,12 +142,6 @@ with gr.Blocks(title="Master Thesis Image Generator") as demo:
 
     store.select(fn=select_img, outputs=out)
 
+    demo.load(fn=clear_gallery, outputs=[store, storage], api_name=False)
 
-
-
-
-
-
-
-
-demo.launch(show_api=False, auth=(USERNAME, PASS))
+demo.launch(share=True, show_api=False, auth=(USERNAME, PASS))