final fix for the gallery, change endpoint to SDXL 1.0

app.py

@@ -15,13 +15,13 @@ USERNAME = os.getenv("USERNAME")
 PASS = os.getenv("PASS")
 
 # set up dataset writer
-hf_writer_gen = gr.HuggingFaceDatasetSaver(HF_KEY, "helms/master-thesis-generated-images-1", private=True, separate_dirs=False)
+hf_writer_gen = gr.HuggingFaceDatasetSaver(HF_KEY, "helms/master-thesis-generated-images", private=True, separate_dirs=False)
 
 def infer(prompt):
     stability_api = client.StabilityInference(
         key=SD_KEY,  # AaPI Key reference.
         verbose=True,  # Print debug messages.
-        engine="stable-diffusion-v1",  # Set the engine to use for generation.
+        engine="stable-diffusion-xl-1024-v1-0",  # Set the engine to use for generation.
         # Available engines: stable-diffusion-xl-1024-v1-0 stable-diffusion-xl-1024-v0-9 stable-diffusion-v1 stable-diffusion-v1-5 stable-diffusion-512-v2-0 stable-diffusion-768-v2-0
         # stable-diffusion-512-v2-1 stable-diffusion-768-v2-1 stable-diffusion-xl-beta-v2-2-2 stable-inpainting-v1-0 stable-inpainting-512-v2-0
     )
@@ -34,8 +34,8 @@ def infer(prompt):
         cfg_scale=7.0,  # Influences how strongly your generation is guided to match your prompt.
         # Setting this value higher increases the strength in which it tries to match your prompt.
         # Defaults to 7.0 if not specified.
-        width=512,  # Generation width, defaults to 512 if not included.
-        height=512,  # Generation height, defaults to 512 if not included.
+        width=1024,  # Generation width, defaults to 512 if not included.
+        height=1024,  # Generation height, defaults to 512 if not included.
         samples=1,  # Number of images to generate, defaults to 1 if not included.
         sampler=generation.SAMPLER_K_DPMPP_2M  # Choose which sampler we want to denoise our generation with.
         # Defaults to k_dpmpp_2m if not specified. Clip Guidance only supports ancestral samplers.
@@ -55,7 +55,6 @@ def infer(prompt):
 
 
 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;">
@@ -85,8 +84,7 @@ with gr.Blocks(title="Master Thesis Image Generator") as demo:
             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(value=storage.value, visible=True, show_share_button=False, allow_preview=False, columns=1)
-                gr.ClearButton(store)
+                store = gr.Gallery(value=[], visible=True, show_share_button=False, allow_preview=False, columns=1)
         with gr.Column(scale=3):
             out = gr.Image(interactive=False, visible=True, container=True, width=1024, show_share_button=False)
     with gr.Row():
@@ -111,14 +109,8 @@ with gr.Blocks(title="Master Thesis Image Generator") as demo:
     def make_active():
         return gr.update(interactive=True)
 
-    def add_to_gallery(img):
-        gall = storage.value
-        gall.insert(0, img)
-        return gall
-
-    def clear_gallery():
-        gall = []
-        return gall, gall
+    def add_to_gallery(img, gall):
+        return [img] + [d['name'] for d in gall]
 
     def select_img(evt: gr.SelectData):
         sel_img = store.value[evt.index]
@@ -129,19 +121,17 @@ with gr.Blocks(title="Master Thesis Image Generator") as demo:
     btn_gen.click(
         fn=make_inactive, outputs=btn_gen).then(
         fn=infer, inputs=inp, outputs=out, scroll_to_output=True).then(
-        fn=add_to_gallery, inputs=out, outputs=store).then(
+        fn=add_to_gallery, inputs=[out, store], outputs=store).then(
         fn=make_active, outputs=btn_gen).then(
         lambda *args: hf_writer_gen.flag(args), inputs=[user, inp, out], outputs=None, preprocess=False)
 
     inp.submit(
         fn=make_inactive, outputs=btn_gen).then(
         fn=infer, inputs=inp, outputs=out, scroll_to_output=True).then(
-        fn=add_to_gallery, inputs=out, outputs=store).then(
+        fn=add_to_gallery, inputs=[out, store], outputs=store).then(
         fn=make_active, outputs=btn_gen).then(
         lambda *args: hf_writer_gen.flag(args), inputs=[user, inp, out], outputs=None, preprocess=False)
 
     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))