Update app.py

app.py

@@ -200,10 +200,6 @@ if default_image:
 else:
     default_image = None
 
-def on_demo_load(request: gr.Request):
-    url_params = gr.State()
-    print("url params", url_params, request)
-
 with gr.Blocks(css=css,head=js,fill_height=True) as demo:
     with gr.Row(equal_height=False):
         with gr.Group():                
@@ -223,10 +219,19 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
                 )
                 random_button = gr.Button("Surprise Me", scale=1, min_width=10)
             run_button = gr.Button( "GO!", scale=1, min_width=20, variant="primary",icon="https://huggingface.co/spaces/nsfwalex/sd_card/resolve/main/hot.svg")
-
-    random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
+            
+    def on_demo_load(request: gr.Request):
+        params = dict(request.query_params)
+        print("url params", params)
+        if params.get("e", "0") == "1":
+            #update the image
+            result.update(value=[default_image])
+            #bind events
+            random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
+
+    result.change(fn=lambda x:x, inputs=[prompt,result], outputs=[], js=f'''(p,img)=>window.uploadImage(p, img,"process_finished","demo_hf_{cfg.get("name")}_card", "finish")''')    
     run_button.click(generate, inputs=[prompt], outputs=[result], js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
-    result.change(fn=lambda x:x, inputs=[prompt,result], outputs=[], js=f'''(p,img)=>window.uploadImage(p, img,"process_finished","demo_hf_{cfg.get("name")}_card", "finish")''')
+            
     demo.load(fn=on_demo_load, inputs=[], outputs=[])#, js='''(img)=>checkDomain(img, "huggingface")''')
 if __name__ == "__main__":
     demo.queue().launch(show_api=False)