getattr just to be sure

pull/183/head
ljleb 2023-02-22 10:57:28 -05:00
parent dc85f2c1f0
commit 36c092fc01
1 changed files with 3 additions and 6 deletions

View File

@ -704,12 +704,9 @@ class Img2ImgTabTracker:
component.click(fn=self.save_submit_img2img_tab, inputs=[], outputs=[]) component.click(fn=self.save_submit_img2img_tab, inputs=[], outputs=[])
return return
if not hasattr(component, "parent"): tab = getattr(component, 'parent', None)
return is_tab = type(tab) is gr.Tab and getattr(tab, 'elem_id', None) is not None
is_img2img_tab = is_tab and getattr(tab, 'parent', None) is not None and getattr(tab.parent, 'elem_id', None) == 'mode_img2img'
tab = component.parent
is_tab = type(tab) is gr.Tab and tab.elem_id is not None
is_img2img_tab = is_tab and tab.parent is not None and tab.parent.elem_id == 'mode_img2img'
if is_img2img_tab and tab.elem_id not in self.img2img_tabs: if is_img2img_tab and tab.elem_id not in self.img2img_tabs:
tab.select(fn=self.set_active_img2img_tab, inputs=gr.State(tab), outputs=[]) tab.select(fn=self.set_active_img2img_tab, inputs=gr.State(tab), outputs=[])
self.img2img_tabs.add(tab.elem_id) self.img2img_tabs.add(tab.elem_id)