247 lines
9.2 KiB
Python
247 lines
9.2 KiB
Python
import gradio as gr
|
|
import math
|
|
|
|
from . import util
|
|
from . import setting
|
|
from . import model
|
|
from . import classification
|
|
from . import ishortcut
|
|
|
|
|
|
def get_thumbnail_list(shortcut_types=None, only_downloaded=False, search=None, shortcut_basemodels=None, sc_classifications=None, page = 0):
|
|
|
|
total = 0
|
|
max_page = 1
|
|
shortlist = ishortcut.get_image_list(shortcut_types, search, shortcut_basemodels, sc_classifications)
|
|
result = None
|
|
|
|
if not shortlist:
|
|
return None, total, max_page
|
|
|
|
if only_downloaded:
|
|
if model.Downloaded_Models:
|
|
downloaded_list = list()
|
|
for short in shortlist:
|
|
sc_name = short[1]
|
|
mid = setting.get_modelid_from_shortcutname(sc_name)
|
|
if mid in model.Downloaded_Models.keys():
|
|
downloaded_list.append(short)
|
|
shortlist = downloaded_list
|
|
else:
|
|
shortlist = None
|
|
|
|
if shortlist:
|
|
total = len(shortlist)
|
|
result = shortlist
|
|
|
|
if total > 0:
|
|
# page 즉 페이징이 아닌 전체가 필요할때도 총페이지 수를 구할때도 있으므로..
|
|
# page == 0 은 전체 리스트를 반환한다
|
|
if setting.shortcut_count_per_page > 0:
|
|
max_page = math.ceil(total / setting.shortcut_count_per_page)
|
|
|
|
if page > 0 and setting.shortcut_count_per_page > 0:
|
|
item_start = setting.shortcut_count_per_page * (page - 1)
|
|
item_end = (setting.shortcut_count_per_page * page)
|
|
if total < item_end:
|
|
item_end = total
|
|
result = shortlist[item_start:item_end]
|
|
|
|
return result, total, max_page
|
|
|
|
def on_refresh_sc_list_change(sc_types, sc_search, sc_basemodels, sc_classifications, show_only_downloaded_sc,sc_page):
|
|
|
|
thumb_list , thumb_totals, thumb_max_page = get_thumbnail_list(sc_types,show_only_downloaded_sc,sc_search,sc_basemodels,sc_classifications,sc_page)
|
|
|
|
# 현재 페이지가 최대 페이지보다 크면 (최대 페이지를 현재 페이지로 넣고)다시한번 리스트를 구한다.
|
|
if thumb_max_page < sc_page:
|
|
sc_page = thumb_max_page
|
|
thumb_list , thumb_totals, thumb_max_page = get_thumbnail_list(sc_types,show_only_downloaded_sc,sc_search,sc_basemodels,sc_classifications,sc_page)
|
|
|
|
return gr.update(value=thumb_list),gr.update(choices=classification.get_list()),gr.update(minimum=1, maximum=thumb_max_page, value=sc_page, step=1, label=f"Total {thumb_max_page} Pages"),thumb_list
|
|
|
|
def on_shortcut_gallery_refresh(sc_types, sc_search, sc_basemodels, sc_classifications, show_only_downloaded_sc):
|
|
thumb_list , thumb_totals, thumb_max_page = get_thumbnail_list(sc_types,show_only_downloaded_sc,sc_search,sc_basemodels,sc_classifications,1)
|
|
return gr.update(value=thumb_list),gr.update(minimum=1, maximum=thumb_max_page, value=1, step=1, label=f"Total {thumb_max_page} Pages"),thumb_list
|
|
|
|
# def on_sc_classification_list_select(evt: gr.SelectData,sc_types, sc_search, sc_basemodels, show_only_downloaded_sc):
|
|
# keys, tags, clfs = util.get_search_keyword(sc_search)
|
|
# search = ""
|
|
# new_search = list()
|
|
|
|
# if keys:
|
|
# new_search.extend(keys)
|
|
|
|
# if tags:
|
|
# new_tags = [f"#{tag}" for tag in tags]
|
|
# new_search.extend(new_tags)
|
|
|
|
# if evt.value != setting.PLACEHOLDER:
|
|
# select_name = evt.value
|
|
|
|
# if select_name and len(select_name.strip()) > 0:
|
|
# new_search.append(f"@{select_name.strip()}")
|
|
|
|
# if new_search:
|
|
# search = ", ".join(new_search)
|
|
|
|
# thumb_list , thumb_totals, thumb_max_page = get_thumbnail_list(sc_types,show_only_downloaded_sc,search,sc_basemodels,1)
|
|
# return gr.update(value=search),gr.update(value=thumb_list),gr.update(minimum=1, maximum=thumb_max_page, value=1, step=1, label=f"Total {thumb_max_page} Pages"),thumb_list
|
|
|
|
def on_sc_gallery_page(sc_types, sc_search, sc_basemodels, sc_classifications, show_only_downloaded_sc,sc_page):
|
|
thumb_list , thumb_totals, thumb_max_page = get_thumbnail_list(sc_types,show_only_downloaded_sc,sc_search,sc_basemodels,sc_classifications,sc_page)
|
|
return gr.update(value=thumb_list),thumb_list
|
|
|
|
def on_ui():
|
|
|
|
thumb_list , thumb_totals, thumb_max_page = get_thumbnail_list(None,False,None,None,None,1)
|
|
|
|
with gr.Accordion("Search", open=True):
|
|
shortcut_type = gr.Dropdown(label='Filter Model Type', multiselect=True, choices=[k for k in setting.ui_typenames], interactive=True)
|
|
sc_search = gr.Textbox(label="Search", value="", placeholder="Search name, #tags, ....",interactive=True, lines=1)
|
|
sc_classification_list = gr.Dropdown(label='Classification',info="The selection options of classification are subject to the AND operation.", multiselect=True, choices=classification.get_list(), interactive=True)
|
|
shortcut_basemodel = gr.Dropdown(label='Filter Model BaseModel', multiselect=True, choices=[k for k in setting.model_basemodels], interactive=True)
|
|
show_only_downloaded_sc = gr.Checkbox(label="Show downloaded model's shortcut only", value=False)
|
|
sc_gallery_page = gr.Slider(minimum=1, maximum=thumb_max_page, value=1, step=1, label=f"Total {thumb_max_page} Pages", interactive=True, visible=True if setting.shortcut_count_per_page > 0 else False)
|
|
# elem_id 를 안써줘야 옆의 인포와 연동이 안된다. 인포쪽에는 써줘야 할것....
|
|
sc_gallery = gr.Gallery(show_label=False, value=thumb_list).style(grid=[setting.shortcut_column], height=["fit" if setting.shortcut_count_per_page != 0 else "auto"], object_fit=setting.gallery_thumbnail_image_style)
|
|
|
|
with gr.Row(visible=False):
|
|
refresh_sc_browser = gr.Textbox()
|
|
refresh_sc_gallery = gr.Textbox()
|
|
sc_gallery_result = gr.State(thumb_list)
|
|
|
|
refresh_sc_gallery.change(lambda x:x, sc_gallery_result, sc_gallery, show_progress=False)
|
|
|
|
sc_gallery_page.release(
|
|
fn = on_sc_gallery_page,
|
|
inputs = [
|
|
shortcut_type,
|
|
sc_search,
|
|
shortcut_basemodel,
|
|
sc_classification_list,
|
|
show_only_downloaded_sc,
|
|
sc_gallery_page
|
|
],
|
|
outputs=[
|
|
sc_gallery,
|
|
sc_gallery_result
|
|
]
|
|
)
|
|
|
|
refresh_sc_browser.change(
|
|
fn=on_refresh_sc_list_change,
|
|
inputs= [
|
|
shortcut_type,
|
|
sc_search,
|
|
shortcut_basemodel,
|
|
sc_classification_list,
|
|
show_only_downloaded_sc,
|
|
sc_gallery_page
|
|
],
|
|
outputs=[
|
|
sc_gallery,
|
|
sc_classification_list,
|
|
sc_gallery_page,
|
|
sc_gallery_result
|
|
],
|
|
show_progress=False
|
|
)
|
|
|
|
shortcut_type.change(
|
|
fn=on_shortcut_gallery_refresh,
|
|
inputs=[
|
|
shortcut_type,
|
|
sc_search,
|
|
shortcut_basemodel,
|
|
sc_classification_list,
|
|
show_only_downloaded_sc,
|
|
],
|
|
outputs=[
|
|
sc_gallery,
|
|
sc_gallery_page,
|
|
sc_gallery_result
|
|
]
|
|
)
|
|
|
|
sc_search.submit(
|
|
fn=on_shortcut_gallery_refresh,
|
|
inputs=[
|
|
shortcut_type,
|
|
sc_search,
|
|
shortcut_basemodel,
|
|
sc_classification_list,
|
|
show_only_downloaded_sc,
|
|
],
|
|
outputs=[
|
|
sc_gallery,
|
|
sc_gallery_page,
|
|
sc_gallery_result
|
|
]
|
|
)
|
|
|
|
shortcut_basemodel.change(
|
|
fn=on_shortcut_gallery_refresh,
|
|
inputs=[
|
|
shortcut_type,
|
|
sc_search,
|
|
shortcut_basemodel,
|
|
sc_classification_list,
|
|
show_only_downloaded_sc,
|
|
],
|
|
outputs=[
|
|
sc_gallery,
|
|
sc_gallery_page,
|
|
sc_gallery_result
|
|
]
|
|
)
|
|
|
|
show_only_downloaded_sc.change(
|
|
fn=on_shortcut_gallery_refresh,
|
|
inputs=[
|
|
shortcut_type,
|
|
sc_search,
|
|
shortcut_basemodel,
|
|
sc_classification_list,
|
|
show_only_downloaded_sc,
|
|
],
|
|
outputs=[
|
|
sc_gallery,
|
|
sc_gallery_page,
|
|
sc_gallery_result
|
|
]
|
|
)
|
|
|
|
sc_classification_list.change(
|
|
fn=on_shortcut_gallery_refresh,
|
|
inputs=[
|
|
shortcut_type,
|
|
sc_search,
|
|
shortcut_basemodel,
|
|
sc_classification_list,
|
|
show_only_downloaded_sc,
|
|
],
|
|
outputs=[
|
|
sc_gallery,
|
|
sc_gallery_page,
|
|
sc_gallery_result
|
|
]
|
|
)
|
|
|
|
# sc_classification_list.select(
|
|
# fn=on_sc_classification_list_select,
|
|
# inputs=[
|
|
# shortcut_type,
|
|
# sc_search,
|
|
# shortcut_basemodel,
|
|
# show_only_downloaded_sc,
|
|
# ],
|
|
# outputs=[
|
|
# sc_search,
|
|
# sc_gallery,
|
|
# sc_gallery_page,
|
|
# sc_gallery_result
|
|
# ]
|
|
# )
|
|
|
|
return sc_gallery, refresh_sc_browser, refresh_sc_gallery |