John6666 commited on
Commit
58fed57
·
verified ·
1 Parent(s): d389b1c

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +55 -46
app.py CHANGED
@@ -1536,10 +1536,11 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
1536
  [menu_model, menu_from_image, menu_negative, menu_gen, menu_hires, menu_lora, menu_advanced,
1537
  menu_example, task_gui, quick_speed_gui],
1538
  queue=False,
 
1539
  )
1540
- model_name_gui.change(get_t2i_model_info, [model_name_gui], [model_info_gui], queue=False)
1541
- translate_prompt_gui.click(translate_to_en, [prompt_gui], [prompt_gui], queue=False)\
1542
- .then(translate_to_en, [neg_prompt_gui], [neg_prompt_gui], queue=False)
1543
 
1544
  gr.on(
1545
  triggers=[quick_model_type_gui.change, quick_genre_gui.change, quick_speed_gui.change, quick_aspect_gui.change],
@@ -1548,6 +1549,7 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
1548
  outputs=[quality_selector_gui, style_selector_gui, sampler_selector_gui, optimization_gui, insert_prompt_gui],
1549
  queue=False,
1550
  trigger_mode="once",
 
1551
  )
1552
  gr.on(
1553
  triggers=[quality_selector_gui.change, style_selector_gui.change, insert_prompt_gui.change],
@@ -1556,6 +1558,7 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
1556
  outputs=[prompt_gui, neg_prompt_gui, quick_model_type_gui],
1557
  queue=False,
1558
  trigger_mode="once",
 
1559
  )
1560
  sampler_selector_gui.change(set_sampler_settings, [sampler_selector_gui], [sampler_gui, steps_gui, cfg_gui, clip_skip_gui, img_width_gui, img_height_gui, optimization_gui], queue=False)
1561
  optimization_gui.change(set_optimization, [optimization_gui, steps_gui, cfg_gui, sampler_gui, clip_skip_gui, lora5_gui, lora_scale_5_gui], [steps_gui, cfg_gui, sampler_gui, clip_skip_gui, lora5_gui, lora_scale_5_gui], queue=False)
@@ -1578,14 +1581,15 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
1578
  lora7_gui, lora_scale_7_gui, lora7_info_gui, lora7_copy_gui, lora7_desc_gui],
1579
  queue=False,
1580
  trigger_mode="once",
 
1581
  )
1582
- lora1_copy_gui.click(apply_lora_prompt, [prompt_gui, lora1_info_gui], [prompt_gui], queue=False)
1583
- lora2_copy_gui.click(apply_lora_prompt, [prompt_gui, lora2_info_gui], [prompt_gui], queue=False)
1584
- lora3_copy_gui.click(apply_lora_prompt, [prompt_gui, lora3_info_gui], [prompt_gui], queue=False)
1585
- lora4_copy_gui.click(apply_lora_prompt, [prompt_gui, lora4_info_gui], [prompt_gui], queue=False)
1586
- lora5_copy_gui.click(apply_lora_prompt, [prompt_gui, lora5_info_gui], [prompt_gui], queue=False)
1587
- lora6_copy_gui.click(apply_lora_prompt, [prompt_gui, lora6_info_gui], [prompt_gui], queue=False)
1588
- lora7_copy_gui.click(apply_lora_prompt, [prompt_gui, lora7_info_gui], [prompt_gui], queue=False)
1589
  gr.on(
1590
  triggers=[search_civitai_button_lora.click, search_civitai_query_lora.submit],
1591
  fn=search_civitai_lora,
@@ -1594,45 +1598,48 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
1594
  outputs=[search_civitai_result_lora, search_civitai_desc_lora, search_civitai_button_lora, search_civitai_query_lora, search_civitai_gallery_lora],
1595
  queue=True,
1596
  scroll_to_output=True,
 
1597
  )
1598
- search_civitai_result_lora.change(select_civitai_lora, [search_civitai_result_lora], [text_lora, search_civitai_desc_lora], queue=False, scroll_to_output=True)
1599
  search_civitai_gallery_lora.select(update_civitai_selection, None, [search_civitai_result_lora], queue=False, show_api=False)
1600
- button_lora.click(get_my_lora, [text_lora, romanize_text], [lora1_gui, lora2_gui, lora3_gui, lora4_gui, lora5_gui, lora6_gui, lora7_gui, new_lora_status], scroll_to_output=True)
1601
- upload_button_lora.upload(upload_file_lora, [upload_button_lora], [file_output_lora, upload_button_lora]).success(
1602
- move_file_lora, [file_output_lora], [lora1_gui, lora2_gui, lora3_gui, lora4_gui, lora5_gui, lora6_gui, lora7_gui], scroll_to_output=True)
1603
 
1604
- use_textual_inversion_gui.change(set_textual_inversion_prompt, [use_textual_inversion_gui, prompt_gui, neg_prompt_gui, prompt_syntax_gui], [prompt_gui, neg_prompt_gui])
1605
 
1606
  generate_from_image_btn_gui.click(
1607
- lambda: ("", "", ""), None, [series_dbt, character_dbt, prompt_gui], queue=False,
1608
  ).success(
1609
  predict_tags_wd,
1610
  [input_image_gui, prompt_gui, image_algorithms, general_threshold_gui, character_threshold_gui],
1611
  [series_dbt, character_dbt, prompt_gui, copy_button_dbt],
 
1612
  ).success(
1613
- compose_prompt_to_copy, [character_dbt, series_dbt, prompt_gui], [prompt_gui], queue=False,
1614
  ).success(
1615
- remove_specific_prompt, [prompt_gui, keep_tags_gui], [prompt_gui], queue=False,
1616
  ).success(
1617
- convert_danbooru_to_e621_prompt, [prompt_gui, tag_type_gui], [prompt_gui], queue=False,
1618
  ).success(
1619
- insert_recom_prompt, [prompt_gui, neg_prompt_gui, recom_prompt_gui], [prompt_gui, neg_prompt_gui], queue=False,
1620
  )
1621
 
1622
- prompt_type_button.click(convert_danbooru_to_e621_prompt, [prompt_gui, prompt_type_gui], [prompt_gui], queue=False)
1623
- random_character_gui.click(select_random_character, [series_dbt, character_dbt], [series_dbt, character_dbt], queue=False)
1624
  generate_db_random_button.click(
1625
  v2_random_prompt,
1626
  [prompt_gui, series_dbt, character_dbt,
1627
  rating_dbt, aspect_ratio_dbt, length_dbt, identity_dbt, ban_tags_dbt, model_name_dbt],
1628
  [prompt_gui, series_dbt, character_dbt],
 
1629
  ).success(
1630
- convert_danbooru_to_e621_prompt, [prompt_gui, tag_type_gui], [prompt_gui], queue=False,
1631
  )
1632
 
1633
- translate_prompt_button.click(translate_prompt, [prompt_gui], [prompt_gui], queue=False)
1634
- translate_prompt_button.click(translate_prompt, [character_dbt], [character_dbt], queue=False)
1635
- translate_prompt_button.click(translate_prompt, [series_dbt], [series_dbt], queue=False)
1636
 
1637
  generate_button.click(
1638
  fn=insert_model_recom_prompt,
@@ -1832,47 +1839,49 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
1832
  copy_prompt_btn_pony = gr.Button(value="Copy to primary prompt", size="sm", interactive=False)
1833
  description_ui()
1834
 
1835
- translate_input_prompt_button.click(translate_prompt, inputs=[input_general], outputs=[input_general], queue=False)
1836
- translate_input_prompt_button.click(translate_prompt, inputs=[input_character], outputs=[input_character], queue=False)
1837
- translate_input_prompt_button.click(translate_prompt, inputs=[input_copyright], outputs=[input_copyright], queue=False)
1838
 
1839
  generate_from_image_btn.click(
1840
- lambda: ("", "", ""), None, [input_copyright, input_character, input_general], queue=False,
1841
  ).success(
1842
  predict_tags_wd,
1843
  [input_image, input_general, image_algorithms, general_threshold, character_threshold],
1844
  [input_copyright, input_character, input_general, copy_input_btn],
 
1845
  ).success(
1846
- remove_specific_prompt, inputs=[input_general, keep_tags], outputs=[input_general], queue=False,
1847
  ).success(
1848
- convert_danbooru_to_e621_prompt, inputs=[input_general, input_tag_type], outputs=[input_general], queue=False,
1849
  ).success(
1850
- insert_recom_prompt, inputs=[input_general, dummy_np, recom_prompt], outputs=[input_general, dummy_np], queue=False,
1851
  ).success(lambda: gr.update(interactive=True), None, [copy_prompt_btn_input], queue=False)
1852
- copy_input_btn.click(compose_prompt_to_copy, inputs=[input_character, input_copyright, input_general], outputs=[input_tags_to_copy])\
1853
- .success(gradio_copy_text, inputs=[input_tags_to_copy], js=COPY_ACTION_JS)
1854
- copy_prompt_btn_input.click(compose_prompt_to_copy, inputs=[input_character, input_copyright, input_general], outputs=[input_tags_to_copy])\
1855
- .success(gradio_copy_prompt, inputs=[input_tags_to_copy], outputs=[prompt_gui])
1856
 
1857
- pick_random_character.click(select_random_character, [input_copyright, input_character], [input_copyright, input_character])
1858
 
1859
  generate_btn.click(
1860
  v2_upsampling_prompt,
1861
  [model_name, input_copyright, input_character, input_general,
1862
  input_rating, input_aspect_ratio, input_length, input_identity, input_ban_tags],
1863
  [output_text],
 
1864
  ).success(
1865
- convert_danbooru_to_e621_prompt, inputs=[output_text, tag_type], outputs=[output_text_pony], queue=False,
1866
  ).success(
1867
- insert_recom_prompt, inputs=[output_text, dummy_np, recom_animagine], outputs=[output_text, dummy_np], queue=False,
1868
  ).success(
1869
- insert_recom_prompt, inputs=[output_text_pony, dummy_np, recom_pony], outputs=[output_text_pony, dummy_np], queue=False,
1870
  ).success(lambda: (gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)),
1871
- None, [copy_btn, copy_btn_pony, copy_prompt_btn, copy_prompt_btn_pony], queue=False)
1872
- copy_btn.click(gradio_copy_text, inputs=[output_text], js=COPY_ACTION_JS)
1873
- copy_btn_pony.click(gradio_copy_text, inputs=[output_text_pony], js=COPY_ACTION_JS)
1874
- copy_prompt_btn.click(gradio_copy_prompt, inputs=[output_text], outputs=[prompt_gui])
1875
- copy_prompt_btn_pony.click(gradio_copy_prompt, inputs=[output_text_pony], outputs=[prompt_gui])
1876
 
1877
  # Stable programmatic endpoint
1878
  from typing import Any, Dict, List, Union, Generator
 
1536
  [menu_model, menu_from_image, menu_negative, menu_gen, menu_hires, menu_lora, menu_advanced,
1537
  menu_example, task_gui, quick_speed_gui],
1538
  queue=False,
1539
+ show_api=False,
1540
  )
1541
+ model_name_gui.change(get_t2i_model_info, [model_name_gui], [model_info_gui], queue=False, show_api=False)
1542
+ translate_prompt_gui.click(translate_to_en, [prompt_gui], [prompt_gui], queue=False, show_api=False)\
1543
+ .then(translate_to_en, [neg_prompt_gui], [neg_prompt_gui], queue=False, show_api=False)
1544
 
1545
  gr.on(
1546
  triggers=[quick_model_type_gui.change, quick_genre_gui.change, quick_speed_gui.change, quick_aspect_gui.change],
 
1549
  outputs=[quality_selector_gui, style_selector_gui, sampler_selector_gui, optimization_gui, insert_prompt_gui],
1550
  queue=False,
1551
  trigger_mode="once",
1552
+ show_api=False,
1553
  )
1554
  gr.on(
1555
  triggers=[quality_selector_gui.change, style_selector_gui.change, insert_prompt_gui.change],
 
1558
  outputs=[prompt_gui, neg_prompt_gui, quick_model_type_gui],
1559
  queue=False,
1560
  trigger_mode="once",
1561
+ show_api=False,
1562
  )
1563
  sampler_selector_gui.change(set_sampler_settings, [sampler_selector_gui], [sampler_gui, steps_gui, cfg_gui, clip_skip_gui, img_width_gui, img_height_gui, optimization_gui], queue=False)
1564
  optimization_gui.change(set_optimization, [optimization_gui, steps_gui, cfg_gui, sampler_gui, clip_skip_gui, lora5_gui, lora_scale_5_gui], [steps_gui, cfg_gui, sampler_gui, clip_skip_gui, lora5_gui, lora_scale_5_gui], queue=False)
 
1581
  lora7_gui, lora_scale_7_gui, lora7_info_gui, lora7_copy_gui, lora7_desc_gui],
1582
  queue=False,
1583
  trigger_mode="once",
1584
+ show_api=False,
1585
  )
1586
+ lora1_copy_gui.click(apply_lora_prompt, [prompt_gui, lora1_info_gui], [prompt_gui], queue=False, show_api=False)
1587
+ lora2_copy_gui.click(apply_lora_prompt, [prompt_gui, lora2_info_gui], [prompt_gui], queue=False, show_api=False)
1588
+ lora3_copy_gui.click(apply_lora_prompt, [prompt_gui, lora3_info_gui], [prompt_gui], queue=False, show_api=False)
1589
+ lora4_copy_gui.click(apply_lora_prompt, [prompt_gui, lora4_info_gui], [prompt_gui], queue=False, show_api=False)
1590
+ lora5_copy_gui.click(apply_lora_prompt, [prompt_gui, lora5_info_gui], [prompt_gui], queue=False, show_api=False)
1591
+ lora6_copy_gui.click(apply_lora_prompt, [prompt_gui, lora6_info_gui], [prompt_gui], queue=False, show_api=False)
1592
+ lora7_copy_gui.click(apply_lora_prompt, [prompt_gui, lora7_info_gui], [prompt_gui], queue=False, show_api=False)
1593
  gr.on(
1594
  triggers=[search_civitai_button_lora.click, search_civitai_query_lora.submit],
1595
  fn=search_civitai_lora,
 
1598
  outputs=[search_civitai_result_lora, search_civitai_desc_lora, search_civitai_button_lora, search_civitai_query_lora, search_civitai_gallery_lora],
1599
  queue=True,
1600
  scroll_to_output=True,
1601
+ show_api=False,
1602
  )
1603
+ search_civitai_result_lora.change(select_civitai_lora, [search_civitai_result_lora], [text_lora, search_civitai_desc_lora], queue=False, scroll_to_output=True, show_api=False)
1604
  search_civitai_gallery_lora.select(update_civitai_selection, None, [search_civitai_result_lora], queue=False, show_api=False)
1605
+ button_lora.click(get_my_lora, [text_lora, romanize_text], [lora1_gui, lora2_gui, lora3_gui, lora4_gui, lora5_gui, lora6_gui, lora7_gui, new_lora_status], scroll_to_output=True, show_api=False)
1606
+ upload_button_lora.upload(upload_file_lora, [upload_button_lora], [file_output_lora, upload_button_lora], show_api=False).success(
1607
+ move_file_lora, [file_output_lora], [lora1_gui, lora2_gui, lora3_gui, lora4_gui, lora5_gui, lora6_gui, lora7_gui], scroll_to_output=True, show_api=False)
1608
 
1609
+ use_textual_inversion_gui.change(set_textual_inversion_prompt, [use_textual_inversion_gui, prompt_gui, neg_prompt_gui, prompt_syntax_gui], [prompt_gui, neg_prompt_gui], show_api=False)
1610
 
1611
  generate_from_image_btn_gui.click(
1612
+ lambda: ("", "", ""), None, [series_dbt, character_dbt, prompt_gui], queue=False, show_api=False,
1613
  ).success(
1614
  predict_tags_wd,
1615
  [input_image_gui, prompt_gui, image_algorithms, general_threshold_gui, character_threshold_gui],
1616
  [series_dbt, character_dbt, prompt_gui, copy_button_dbt],
1617
+ show_api=False,
1618
  ).success(
1619
+ compose_prompt_to_copy, [character_dbt, series_dbt, prompt_gui], [prompt_gui], queue=False, show_api=False,
1620
  ).success(
1621
+ remove_specific_prompt, [prompt_gui, keep_tags_gui], [prompt_gui], queue=False, show_api=False,
1622
  ).success(
1623
+ convert_danbooru_to_e621_prompt, [prompt_gui, tag_type_gui], [prompt_gui], queue=False, show_api=False,
1624
  ).success(
1625
+ insert_recom_prompt, [prompt_gui, neg_prompt_gui, recom_prompt_gui], [prompt_gui, neg_prompt_gui], queue=False, show_api=False,
1626
  )
1627
 
1628
+ prompt_type_button.click(convert_danbooru_to_e621_prompt, [prompt_gui, prompt_type_gui], [prompt_gui], queue=False, show_api=False)
1629
+ random_character_gui.click(select_random_character, [series_dbt, character_dbt], [series_dbt, character_dbt], queue=False, show_api=False)
1630
  generate_db_random_button.click(
1631
  v2_random_prompt,
1632
  [prompt_gui, series_dbt, character_dbt,
1633
  rating_dbt, aspect_ratio_dbt, length_dbt, identity_dbt, ban_tags_dbt, model_name_dbt],
1634
  [prompt_gui, series_dbt, character_dbt],
1635
+ show_api=False,
1636
  ).success(
1637
+ convert_danbooru_to_e621_prompt, [prompt_gui, tag_type_gui], [prompt_gui], queue=False, show_api=False,
1638
  )
1639
 
1640
+ translate_prompt_button.click(translate_prompt, [prompt_gui], [prompt_gui], queue=False, show_api=False)
1641
+ translate_prompt_button.click(translate_prompt, [character_dbt], [character_dbt], queue=False, show_api=False)
1642
+ translate_prompt_button.click(translate_prompt, [series_dbt], [series_dbt], queue=False, show_api=False)
1643
 
1644
  generate_button.click(
1645
  fn=insert_model_recom_prompt,
 
1839
  copy_prompt_btn_pony = gr.Button(value="Copy to primary prompt", size="sm", interactive=False)
1840
  description_ui()
1841
 
1842
+ translate_input_prompt_button.click(translate_prompt, inputs=[input_general], outputs=[input_general], queue=False, show_api=False)
1843
+ translate_input_prompt_button.click(translate_prompt, inputs=[input_character], outputs=[input_character], queue=False, show_api=False)
1844
+ translate_input_prompt_button.click(translate_prompt, inputs=[input_copyright], outputs=[input_copyright], queue=False, show_api=False)
1845
 
1846
  generate_from_image_btn.click(
1847
+ lambda: ("", "", ""), None, [input_copyright, input_character, input_general], queue=False, show_api=False,
1848
  ).success(
1849
  predict_tags_wd,
1850
  [input_image, input_general, image_algorithms, general_threshold, character_threshold],
1851
  [input_copyright, input_character, input_general, copy_input_btn],
1852
+ show_api=False,
1853
  ).success(
1854
+ remove_specific_prompt, inputs=[input_general, keep_tags], outputs=[input_general], queue=False, show_api=False,
1855
  ).success(
1856
+ convert_danbooru_to_e621_prompt, inputs=[input_general, input_tag_type], outputs=[input_general], queue=False, show_api=False,
1857
  ).success(
1858
+ insert_recom_prompt, inputs=[input_general, dummy_np, recom_prompt], outputs=[input_general, dummy_np], queue=False, show_api=False,
1859
  ).success(lambda: gr.update(interactive=True), None, [copy_prompt_btn_input], queue=False)
1860
+ copy_input_btn.click(compose_prompt_to_copy, inputs=[input_character, input_copyright, input_general], outputs=[input_tags_to_copy], show_api=False)\
1861
+ .success(gradio_copy_text, inputs=[input_tags_to_copy], js=COPY_ACTION_JS, show_api=False)
1862
+ copy_prompt_btn_input.click(compose_prompt_to_copy, inputs=[input_character, input_copyright, input_general], outputs=[input_tags_to_copy], show_api=False)\
1863
+ .success(gradio_copy_prompt, inputs=[input_tags_to_copy], outputs=[prompt_gui], show_api=False)
1864
 
1865
+ pick_random_character.click(select_random_character, [input_copyright, input_character], [input_copyright, input_character], show_api=False)
1866
 
1867
  generate_btn.click(
1868
  v2_upsampling_prompt,
1869
  [model_name, input_copyright, input_character, input_general,
1870
  input_rating, input_aspect_ratio, input_length, input_identity, input_ban_tags],
1871
  [output_text],
1872
+ show_api=False,
1873
  ).success(
1874
+ convert_danbooru_to_e621_prompt, inputs=[output_text, tag_type], outputs=[output_text_pony], queue=False, show_api=False,
1875
  ).success(
1876
+ insert_recom_prompt, inputs=[output_text, dummy_np, recom_animagine], outputs=[output_text, dummy_np], queue=False, show_api=False,
1877
  ).success(
1878
+ insert_recom_prompt, inputs=[output_text_pony, dummy_np, recom_pony], outputs=[output_text_pony, dummy_np], queue=False, show_api=False,
1879
  ).success(lambda: (gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True)),
1880
+ None, [copy_btn, copy_btn_pony, copy_prompt_btn, copy_prompt_btn_pony], queue=False, show_api=False)
1881
+ copy_btn.click(gradio_copy_text, inputs=[output_text], js=COPY_ACTION_JS, show_api=False)
1882
+ copy_btn_pony.click(gradio_copy_text, inputs=[output_text_pony], js=COPY_ACTION_JS, show_api=False)
1883
+ copy_prompt_btn.click(gradio_copy_prompt, inputs=[output_text], outputs=[prompt_gui], show_api=False)
1884
+ copy_prompt_btn_pony.click(gradio_copy_prompt, inputs=[output_text_pony], outputs=[prompt_gui], show_api=False)
1885
 
1886
  # Stable programmatic endpoint
1887
  from typing import Any, Dict, List, Union, Generator