state bug fix

app.py

@@ -218,13 +218,13 @@ class WebApp():
             print(f"Error catched: {e}")
             gr.Markdown(f"**Error catched: {e}**")
 
-    def run_ditail_alt(self, gr_args, *values):
+    def run_ditail_alt(self, *values):
         gr_args = self.args_base.copy()
         print(self.args_input.keys())
         for k, v in zip(list(self.args_input.keys()), values):
             gr_args[k] = v
         # quick fix for example
-        gr_args['lora'] = 'none' if not isinstance(gr_args['lora'], str) else gr_state['lora']
+        gr_args['lora'] = 'none' if not isinstance(gr_args['lora'], str) else gr_args['lora']
         print('selected lora: ', gr_args['lora'])
         # map inversion model to url
         gr_args['pos_prompt'] = ', '.join(LORA_TRIGGER_WORD.get(gr_args['lora'], [])+[gr_args['pos_prompt']])
@@ -236,19 +236,19 @@ class WebApp():
         ditail = DitailDemo(gr_args)
         
         metadata_to_show = ['inv_model', 'spl_model', 'lora', 'lora_scale', 'inv_steps', 'spl_steps', 'pos_prompt', 'alpha', 'neg_prompt', 'beta', 'omega']
-        self.args_to_show = {}
+        args_to_show = {}
         for key in metadata_to_show:
-            self.args_to_show[key] = gr_args[key]
+            args_to_show[key] = gr_args[key]
 
         img = ditail.run_ditail()
 
         # reset ditail
         ditail = None
 
-        return gr_args, img, self.args_to_show
+        return img, args_to_show
 
     def run_example(self, img, prompt, inv_model, spl_model, lora):
-        return self.run_ditail_alt(self.gr_state, img, prompt, spl_model, gr.State(lora), inv_model)
+        return self.run_ditail_alt(img, prompt, spl_model, gr.State(lora), inv_model)
 
     def show_credits(self):
         # gr.Markdown(
@@ -289,12 +289,9 @@ class WebApp():
                 # expected_output_image = gr.Image(label="expected output image", visible=False)
                 metadata = gr.JSON(label='metadata')
 
-                # init a gradio state
-                self.gr_state = gr.State()
-
                 submit_btn.click(self.run_ditail_alt,
-                                inputs=[self.gr_state] + list(self.args_input.values()),
-                                outputs=[self.gr_state, output_image, metadata],
+                                inputs=list(self.args_input.values()),
+                                outputs=[output_image, metadata],
                                 scroll_to_output=True,
                                 )
 
@@ -304,7 +301,7 @@ class WebApp():
                     examples=[[os.path.join(os.path.dirname(__file__), "example", "Lenna.png"), 'a woman called Lenna wearing a feathered hat', list(BASE_MODEL.keys())[1], list(BASE_MODEL.keys())[2], 'none']],
                     inputs=[self.args_input['img'], self.args_input['pos_prompt'], self.args_input['inv_model'], self.args_input['spl_model'], gr.Textbox(label='LoRA', visible=False), ],
                     fn = self.run_example,
-                    outputs=[self.gr_state, output_image, metadata],
+                    outputs=[output_image, metadata],
                     run_on_click=True,
                     cache_examples=cache_examples,
                 )