Spaces:
Build error
Build error
| import copy | |
| import random | |
| import gradio as gr | |
| from constants.css import STYLE | |
| from constants.init_values import ( | |
| genres, places, moods, jobs, ages, random_names, personalities, default_character_images, styles | |
| ) | |
| from constants import desc | |
| from interfaces import ( | |
| ui, chat_ui, story_gen_ui, view_change_ui, export_ui | |
| ) | |
| from modules.llms import get_llm_factory | |
| with gr.Blocks(css=STYLE) as demo: | |
| factory = get_llm_factory("PaLM") #TODO: Replace with selected LLM factory | |
| ui_pp_manager = factory.create_ui_pp_manager() | |
| chat_mode = gr.State("setting_chat") | |
| chat_state = gr.State({ | |
| "setting_chat": ui_pp_manager, | |
| "story_chat": ui_pp_manager, | |
| "export_chat": ui_pp_manager, | |
| }) | |
| cur_cursor = gr.State(0) | |
| cursors = gr.State([]) | |
| gallery_images1 = gr.State(default_character_images) | |
| gallery_images2 = gr.State(default_character_images) | |
| gallery_images3 = gr.State(default_character_images) | |
| gallery_images4 = gr.State(default_character_images) | |
| selected_main_char_image1 = gr.State(default_character_images[0]) | |
| selected_side_char_image1 = gr.State(default_character_images[0]) | |
| selected_side_char_image2 = gr.State(default_character_images[0]) | |
| selected_side_char_image3 = gr.State(default_character_images[0]) | |
| with gr.Column(visible=True, elem_id="pre_phase") as pre_phase: | |
| gr.Markdown("# 📖 Zero2Story", elem_classes=["markdown-center"]) | |
| gr.Markdown(desc.pre_phase_description, elem_classes=["markdown-justify"]) | |
| pre_to_setup_btn = gr.Button("create a custom story", elem_classes=["wrap", "control-button"]) | |
| with gr.Column(visible=False, elem_id="background_setup_phase") as background_setup_phase: | |
| gr.Markdown("# 🌐 World setup", elem_classes=["markdown-center"]) | |
| gr.Markdown(desc.background_setup_phase_description, elem_classes=["markdown-justify"]) | |
| with gr.Row(): | |
| with gr.Column(): | |
| genre_dd = gr.Dropdown(label="genre", choices=genres, value=genres[0], interactive=True, elem_classes=["center-label"]) | |
| with gr.Column(): | |
| place_dd = gr.Dropdown(label="place", choices=places["Middle Ages"], value=places["Middle Ages"][0], allow_custom_value=True, interactive=True, elem_classes=["center-label"]) | |
| with gr.Column(): | |
| mood_dd = gr.Dropdown(label="mood", choices=moods["Middle Ages"], value=moods["Middle Ages"][0], allow_custom_value=True, interactive=True, elem_classes=["center-label"]) | |
| with gr.Row(): | |
| back_to_pre_btn = gr.Button("← back", elem_classes=["wrap", "control-button"], scale=1) | |
| world_setup_confirm_btn = gr.Button("character setup →", elem_classes=["wrap", "control-button"], scale=2) | |
| with gr.Column(visible=False, elem_id="character_setup_phase") as character_setup_phase: | |
| gr.Markdown("# 👥 Character setup") | |
| gr.Markdown(desc.character_setup_phase_description, elem_classes=["markdown-justify"]) | |
| with gr.Tab("Main character"): | |
| with gr.Column(): | |
| gr.Checkbox(label="character include/enable", value=True, interactive=False) | |
| char_gallery1 = gr.Gallery(value=default_character_images, height=256, preview=True) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("name", elem_classes=["markdown-left"], scale=3) | |
| name_txt1 = gr.Textbox(random_names[0], elem_classes=["no-label"], scale=3) | |
| random_name_btn1 = gr.Button("random", elem_classes=["wrap", "control-button-green", "left-margin"], scale=1) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("age", elem_classes=["markdown-left"], scale=3) | |
| age_dd1 = gr.Dropdown(label=None, choices=ages, value=ages[0], elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("nature", elem_classes=["markdown-left"], scale=3) | |
| personality_dd1 = gr.Dropdown(label=None, choices=personalities, value=personalities[0], interactive=True, elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("job", elem_classes=["markdown-left"], scale=3) | |
| job_dd1 = gr.Dropdown(label=None, choices=jobs["Middle Ages"], value=jobs["Middle Ages"][0], allow_custom_value=True, interactive=True, elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"], visible=False): | |
| gr.Markdown("style", elem_classes=["markdown-left"], scale=3) | |
| creative_dd1 = gr.Dropdown(choices=styles, value=styles[0], allow_custom_value=True, interactive=True, elem_classes=["no-label"], scale=4) | |
| gen_char_btn1 = gr.Button("gen character", elem_classes=["wrap", "control-button-green"]) | |
| with gr.Tab("Side character 1"): | |
| with gr.Column(): | |
| side_char_enable_ckb1 = gr.Checkbox(label="character include/enable", value=False) | |
| char_gallery2 = gr.Gallery(value=default_character_images, height=256, preview=True) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("name", elem_classes=["markdown-left"], scale=3) | |
| name_txt2 = gr.Textbox(random_names[1], elem_classes=["no-label"], scale=3) | |
| random_name_btn2 = gr.Button("random", elem_classes=["wrap", "control-button-green", "left-margin"], scale=1) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("age", elem_classes=["markdown-left"], scale=3) | |
| age_dd2 = gr.Dropdown(label=None, choices=ages, value=ages[1], elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("nature", elem_classes=["markdown-left"], scale=3) | |
| personality_dd2 = gr.Dropdown(label=None, choices=personalities, value=personalities[1], interactive=True, elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("job", elem_classes=["markdown-left"], scale=3) | |
| job_dd2 = gr.Dropdown(label=None, choices=jobs["Middle Ages"], value=jobs["Middle Ages"][1], allow_custom_value=True, interactive=True, elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"], visible=False): | |
| gr.Markdown("style", elem_classes=["markdown-left"], scale=3) | |
| creative_dd2 = gr.Dropdown(choices=styles, value=styles[0], allow_custom_value=True, interactive=True, elem_classes=["no-label"], scale=4) | |
| gen_char_btn2 = gr.Button("gen character", elem_classes=["wrap", "control-button-green"]) | |
| with gr.Tab("Side character 2"): | |
| with gr.Column(): | |
| side_char_enable_ckb2 = gr.Checkbox(label="character include/enable", value=False) | |
| char_gallery3 = gr.Gallery(value=default_character_images, height=256, preview=True) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("name", elem_classes=["markdown-left"], scale=3) | |
| name_txt3 = gr.Textbox(random_names[2], elem_classes=["no-label"], scale=3) | |
| random_name_btn3 = gr.Button("random", elem_classes=["wrap", "control-button-green", "left-margin"], scale=1) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("age", elem_classes=["markdown-left"], scale=3) | |
| age_dd3 = gr.Dropdown(label=None, choices=ages, value=ages[2], elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("nature", elem_classes=["markdown-left"], scale=3) | |
| personality_dd3 = gr.Dropdown(label=None, choices=personalities, value=personalities[2], interactive=True, elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("job", elem_classes=["markdown-left"], scale=3) | |
| job_dd3 = gr.Dropdown(label=None, choices=jobs["Middle Ages"], value=jobs["Middle Ages"][2], allow_custom_value=True, interactive=True, elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"], visible=False): | |
| gr.Markdown("style", elem_classes=["markdown-left"], scale=3) | |
| creative_dd3 = gr.Dropdown(choices=styles, value=styles[0], allow_custom_value=True, interactive=True, elem_classes=["no-label"], scale=4) | |
| gen_char_btn3 = gr.Button("gen character", elem_classes=["wrap", "control-button-green"]) | |
| with gr.Tab("Side character 3"): | |
| with gr.Column(): | |
| side_char_enable_ckb3 = gr.Checkbox(label="character include/enable", value=False) | |
| char_gallery4 = gr.Gallery(value=default_character_images, height=256, preview=True) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("name", elem_classes=["markdown-left"], scale=3) | |
| name_txt4 = gr.Textbox(random_names[3], elem_classes=["no-label"], scale=3) | |
| random_name_btn4 = gr.Button("random", elem_classes=["wrap", "control-button-green", "left-margin"], scale=1) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("age", elem_classes=["markdown-left"], scale=3) | |
| age_dd4 = gr.Dropdown(label=None, choices=ages, value=ages[3], elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("nature", elem_classes=["markdown-left"], scale=3) | |
| personality_dd4 = gr.Dropdown(label=None, choices=personalities, value=personalities[3], interactive=True, elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"]): | |
| gr.Markdown("job", elem_classes=["markdown-left"], scale=3) | |
| job_dd4 = gr.Dropdown(label=None, choices=jobs["Middle Ages"], value=jobs["Middle Ages"][3], allow_custom_value=True, interactive=True, elem_classes=["no-label"], scale=4) | |
| with gr.Row(elem_classes=["no-gap"], visible=False): | |
| gr.Markdown("style", elem_classes=["markdown-left"], scale=3) | |
| creative_dd4 = gr.Dropdown(choices=styles, value=styles[0], allow_custom_value=True, interactive=True, elem_classes=["no-label"], scale=4) | |
| gen_char_btn4 = gr.Button("gen character", elem_classes=["wrap", "control-button-green"]) | |
| with gr.Row(): | |
| back_to_background_setup_btn = gr.Button("← back", elem_classes=["wrap", "control-button"], scale=1) | |
| character_setup_confirm_btn = gr.Button("generate first stories →", elem_classes=["wrap", "control-button"], scale=2) | |
| gr.Markdown("### 💡 Plot setup", visible=False) | |
| with gr.Accordion("generate chapter titles and each plot", open=False, visible=False) as plot_setup_section: | |
| title = gr.Textbox("Title Undetermined Yet", elem_classes=["no-label", "font-big"]) | |
| # plot = gr.Textbox(lines=10, elem_classes=["no-label", "small-big-textarea"]) | |
| gr.Textbox("Rising action", elem_classes=["no-label"]) | |
| with gr.Row(elem_classes=["left-margin"]): | |
| chapter1_plot = gr.Textbox(placeholder="The plot of the first chapter will be generated here", lines=3, elem_classes=["no-label"]) | |
| gr.Textbox("Crisis", elem_classes=["no-label"]) | |
| with gr.Row(elem_classes=["left-margin"]): | |
| chapter2_plot = gr.Textbox(placeholder="The plot of the second chapter will be generated here", lines=3, elem_classes=["no-label"]) | |
| gr.Textbox("Climax", elem_classes=["no-label"]) | |
| with gr.Row(elem_classes=["left-margin"]): | |
| chapter3_plot = gr.Textbox(placeholder="The plot of the third chapter will be generated here", lines=3, elem_classes=["no-label"]) | |
| gr.Textbox("Falling action", elem_classes=["no-label"]) | |
| with gr.Row(elem_classes=["left-margin"]): | |
| chapter4_plot = gr.Textbox(placeholder="The plot of the fourth chapter will be generated here", lines=3, elem_classes=["no-label"]) | |
| gr.Textbox("Denouement", elem_classes=["no-label"]) | |
| with gr.Row(elem_classes=["left-margin"]): | |
| chapter5_plot = gr.Textbox(placeholder="The plot of the fifth chapter will be generated here", lines=3, elem_classes=["no-label"]) | |
| with gr.Row(): | |
| plot_gen_temp = gr.Slider(0.0, 2.0, 1.0, step=0.1, label="temperature") | |
| plot_gen_btn = gr.Button("gen plot", elem_classes=["control-button"]) | |
| plot_setup_confirm_btn = gr.Button("confirm", elem_classes=["control-button"]) | |
| with gr.Column(visible=False, elem_id="writing_phase") as writing_phase: | |
| gr.Markdown("# ✍🏼 Story writing") | |
| gr.Markdown(desc.story_generation_phase_description, elem_classes=["markdown-justify"]) | |
| progress_comp = gr.Textbox(label=None, elem_classes=["no-label"], interactive=False) | |
| title_display = gr.Markdown("# Title Undetermined Yet", elem_classes=["markdown-center"], visible=False) | |
| subtitle_display = gr.Markdown("### Title Undetermined Yet", elem_classes=["markdown-center"], visible=False) | |
| with gr.Row(): | |
| image_gen_btn = gr.Button("🏞️ Image", interactive=False, elem_classes=["control-button-green"]) | |
| audio_gen_btn = gr.Button("🔊 Audio", interactive=False, elem_classes=["control-button-green"]) | |
| img_audio_combine_btn = gr.Button("📀 Image + Audio", interactive=False, elem_classes=["control-button-green"]) | |
| story_image = gr.Image(None, visible=False, type="filepath", interactive=False, elem_classes=["no-label-image-audio"]) | |
| story_audio = gr.Audio(None, visible=False, type="filepath", interactive=False, elem_classes=["no-label-image-audio"]) | |
| story_video = gr.Video(visible=False, interactive=False, elem_classes=["no-label-gallery"]) | |
| story_progress = gr.Slider( | |
| 1, 2, 1, step=1, interactive=True, | |
| label="1/2", visible=False | |
| ) | |
| story_content = gr.Textbox( | |
| "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Integer interdum eleifend tincidunt. Vivamus dapibus, massa ut imperdiet condimentum, quam ipsum vehicula eros, a accumsan nisl metus at nisl. Nullam tortor nibh, vehicula sed tellus at, accumsan efficitur enim. Sed mollis purus vitae nisl ornare volutpat. In vitae tortor nec neque sagittis vehicula. In vestibulum velit eu lorem pulvinar dignissim. Donec eu sapien et sapien cursus pretium elementum eu urna. Proin lacinia ipsum maximus, commodo dui tempus, convallis tortor. Nulla sodales mi libero, nec eleifend eros interdum quis. Pellentesque nulla lectus, scelerisque et consequat vitae, blandit at ante. Sed nec …….", | |
| lines=12, | |
| elem_classes=["no-label", "small-big-textarea"] | |
| ) | |
| with gr.Accordion("regeneration controls", open=False): | |
| with gr.Row(): | |
| regen_actions_btn = gr.Button("Re-suggest actions", interactive=True, elem_classes=["control-button-green"]) | |
| regen_story_btn = gr.Button("Re-suggest story and actions", interactive=True, elem_classes=["control-button-green"]) | |
| custom_prompt_txt = gr.Textbox(placeholder="Re-suggest story and actions based on your own custom request", interactive=True, elem_classes=["no-label", "small-big-textarea"]) | |
| with gr.Row(): | |
| action_btn1 = gr.Button("Action Choice 1", interactive=False, elem_classes=["control-button-green"]) | |
| action_btn2 = gr.Button("Action Choice 2", interactive=False, elem_classes=["control-button-green"]) | |
| action_btn3 = gr.Button("Action Choice 3", interactive=False, elem_classes=["control-button-green"]) | |
| custom_action_txt = gr.Textbox(placeholder="write your own custom action", elem_classes=["no-label", "small-big-textarea"], scale=3) | |
| with gr.Row(): | |
| restart_from_story_generation_btn = gr.Button("← back", elem_classes=["wrap", "control-button"], scale=1) | |
| story_writing_done_btn = gr.Button("export your story →", elem_classes=["wrap", "control-button"], scale=2) | |
| with gr.Column(visible=False, elem_id="export_phase") as export_phase: | |
| gr.Markdown("# 📤 Export your story") | |
| gr.Markdown(desc.export_phase_description, elem_classes=["markdown-justify"]) | |
| title_txt = gr.Textbox("Your Own Story", elem_classes=["no-label"]) | |
| title_gen_btn = gr.Button("gnerate a title", elem_classes=["control-button-green"]) | |
| with gr.Row(): | |
| back_to_story_writing_btn = gr.Button("← back", elem_classes=["wrap", "control-button"], scale=1) | |
| restart_from_export_btn = gr.Button("start over", elem_classes=["wrap", "control-button"], scale=1) | |
| export_done_btn = gr.Button("exported story →", elem_classes=["wrap", "control-button"], scale=1) | |
| with gr.Column(visible=False, elem_id="export_view_phase") as export_view_phase: | |
| export_html = gr.HTML() | |
| with gr.Row(): | |
| restart_from_export_view_btn = gr.Button("start over", elem_classes=["wrap", "control-button"]) | |
| export_to_file_btn = gr.Button("(TBD) Download as ZIP file", elem_classes=["wrap", "control-button"], interactive=False, scale=2) | |
| with gr.Accordion("💬", open=False, elem_id="chat-section", visible=False) as chat_section: | |
| with gr.Column(scale=1): | |
| chatbot = gr.Chatbot( | |
| [], | |
| avatar_images=("assets/user.png", "assets/ai.png"), | |
| elem_id="chatbot", | |
| elem_classes=["no-label-chatbot"]) | |
| chat_input_txt = gr.Textbox(placeholder="enter...", interactive=True, elem_id="chat-input", elem_classes=["no-label"]) | |
| with gr.Row(elem_id="chat-buttons"): | |
| regen_btn = gr.Button("regen", interactive=False, elem_classes=["control-button"]) | |
| clear_btn = gr.Button("clear", elem_classes=["control-button"]) | |
| pre_to_setup_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[pre_phase, background_setup_phase], | |
| _js=view_change_ui.pre_to_setup_js | |
| ) | |
| back_to_pre_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[pre_phase, background_setup_phase], | |
| _js=view_change_ui.back_to_pre_js | |
| ) | |
| world_setup_confirm_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[background_setup_phase, character_setup_phase], | |
| _js=view_change_ui.world_setup_confirm_js | |
| ) | |
| back_to_background_setup_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[background_setup_phase, character_setup_phase], | |
| _js=view_change_ui.back_to_background_setup_js | |
| ) | |
| restart_from_story_generation_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[pre_phase, writing_phase], | |
| _js=view_change_ui.restart_from_story_generation_js | |
| ) | |
| story_writing_done_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[writing_phase, export_phase], | |
| _js=view_change_ui.story_writing_done_js | |
| ) | |
| title_gen_btn.click( | |
| export_ui.title_gen, | |
| inputs=[cursors], | |
| outputs=[title_txt] | |
| ) | |
| export_done_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[export_phase, export_view_phase], | |
| _js=view_change_ui.export_done_js | |
| ) | |
| export_done_btn.click( | |
| export_ui.export, | |
| inputs=[ | |
| title_txt, | |
| cursors, | |
| selected_main_char_image1, name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, selected_side_char_image1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, selected_side_char_image2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, selected_side_char_image3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| export_html | |
| ] | |
| ) | |
| back_to_story_writing_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[writing_phase, export_phase], | |
| _js=view_change_ui.back_to_story_writing_js | |
| ) | |
| restart_from_export_view_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[pre_phase, export_view_phase], | |
| _js=view_change_ui.restart_from_export_view_js | |
| ) | |
| restart_from_export_view_btn.click( | |
| ui.reset, | |
| inputs=None, | |
| outputs=[ | |
| cursors, cur_cursor, | |
| chat_state, chat_mode, | |
| gallery_images1, gallery_images2, gallery_images3, gallery_images4, | |
| selected_main_char_image1, selected_side_char_image1, selected_side_char_image2, selected_side_char_image3, | |
| genre_dd, place_dd, mood_dd, | |
| char_gallery1, job_dd1, | |
| side_char_enable_ckb1, char_gallery2, job_dd2, | |
| side_char_enable_ckb2, char_gallery3, job_dd3, | |
| side_char_enable_ckb3, char_gallery4, job_dd4, | |
| story_image, story_audio, story_video, | |
| story_content, story_progress, | |
| custom_prompt_txt, action_btn1, action_btn2, action_btn3, custom_action_txt, | |
| title_txt, export_html | |
| ] | |
| ) | |
| restart_from_export_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[pre_phase, export_phase], | |
| _js=view_change_ui.restart_from_export_js | |
| ) | |
| restart_from_export_btn.click( | |
| ui.reset, | |
| inputs=None, | |
| outputs=[ | |
| cursors, cur_cursor, | |
| chat_state, chat_mode, | |
| gallery_images1, gallery_images2, gallery_images3, gallery_images4, | |
| selected_main_char_image1, selected_side_char_image1, selected_side_char_image2, selected_side_char_image3, | |
| genre_dd, place_dd, mood_dd, | |
| char_gallery1, job_dd1, | |
| side_char_enable_ckb1, char_gallery2, job_dd2, | |
| side_char_enable_ckb2, char_gallery3, job_dd3, | |
| side_char_enable_ckb3, char_gallery4, job_dd4, | |
| story_image, story_audio, story_video, | |
| story_content, story_progress, | |
| custom_prompt_txt, action_btn1, action_btn2, action_btn3, custom_action_txt, | |
| title_txt, export_html | |
| ] | |
| ) | |
| character_setup_confirm_btn.click( | |
| fn=None, | |
| inputs=None, | |
| outputs=[character_setup_phase, writing_phase], | |
| _js=view_change_ui.character_setup_confirm_js | |
| ) | |
| character_setup_confirm_btn.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.first_story_gen, | |
| inputs=[ | |
| cursors, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| cursors, cur_cursor, story_content, story_progress, image_gen_btn, audio_gen_btn, | |
| story_image, story_audio, story_video | |
| ] | |
| ).then( | |
| story_gen_ui.actions_gen, | |
| inputs=[ | |
| cursors, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| action_btn1, action_btn2, action_btn3, progress_comp | |
| ] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| regen_actions_btn.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.actions_gen, | |
| inputs=[ | |
| cursors, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| action_btn1, action_btn2, action_btn3, progress_comp | |
| ] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| regen_story_btn.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.update_story_gen, | |
| inputs=[ | |
| cursors, cur_cursor, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| cursors, cur_cursor, story_content, story_progress, image_gen_btn, audio_gen_btn | |
| ] | |
| ).then( | |
| story_gen_ui.actions_gen, | |
| inputs=[ | |
| cursors, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| action_btn1, action_btn2, action_btn3, progress_comp | |
| ] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| #### Setups | |
| genre_dd.select( | |
| ui.update_on_age, | |
| outputs=[place_dd, mood_dd, job_dd1, job_dd2, job_dd3, job_dd4] | |
| ) | |
| gen_char_btn1.click( | |
| ui.gen_character_image, | |
| inputs=[ | |
| gallery_images1, name_txt1, age_dd1, personality_dd1, job_dd1, genre_dd, place_dd, mood_dd, creative_dd1], | |
| outputs=[char_gallery1, gallery_images1, selected_main_char_image1] | |
| ) | |
| gen_char_btn2.click( | |
| ui.gen_character_image, | |
| inputs=[gallery_images2, name_txt2, age_dd2, personality_dd2, job_dd2, genre_dd, place_dd, mood_dd, creative_dd2], | |
| outputs=[char_gallery2, gallery_images2, selected_side_char_image1] | |
| ) | |
| gen_char_btn3.click( | |
| ui.gen_character_image, | |
| inputs=[gallery_images3, name_txt3, age_dd3, personality_dd3, job_dd3, genre_dd, place_dd, mood_dd, creative_dd3], | |
| outputs=[char_gallery3, gallery_images3, selected_side_char_image2] | |
| ) | |
| gen_char_btn4.click( | |
| ui.gen_character_image, | |
| inputs=[gallery_images4, name_txt4, age_dd4, personality_dd4, job_dd4, genre_dd, place_dd, mood_dd, creative_dd4], | |
| outputs=[char_gallery4, gallery_images4, selected_side_char_image3] | |
| ) | |
| random_name_btn1.click( | |
| fn=None, | |
| inputs=[name_txt1, name_txt2, name_txt3, name_txt4], | |
| outputs=[name_txt1], | |
| _js=ui.get_random_name_js | |
| ) | |
| random_name_btn2.click( | |
| fn=None, | |
| inputs=[name_txt2, name_txt1, name_txt3, name_txt4], | |
| outputs=[name_txt2], | |
| _js=ui.get_random_name_js | |
| ) | |
| random_name_btn3.click( | |
| fn=None, | |
| inputs=[name_txt3, name_txt1, name_txt2, name_txt4], | |
| outputs=[name_txt3], | |
| _js=ui.get_random_name_js | |
| ) | |
| random_name_btn4.click( | |
| fn=None, | |
| inputs=[name_txt4, name_txt1, name_txt2, name_txt3], | |
| outputs=[name_txt4], | |
| _js=ui.get_random_name_js | |
| ) | |
| ### Story generation | |
| story_content.input( | |
| story_gen_ui.update_story_content, | |
| inputs=[story_content, cursors, cur_cursor], | |
| outputs=[cursors], | |
| ) | |
| image_gen_btn.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.image_gen, | |
| inputs=[ | |
| genre_dd, place_dd, mood_dd, title, story_content, cursors, cur_cursor | |
| ], | |
| outputs=[ | |
| story_image, cursors, progress_comp, | |
| ] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| audio_gen_btn.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.audio_gen, | |
| inputs=[ | |
| genre_dd, place_dd, mood_dd, title, story_content, cursors, cur_cursor | |
| ], | |
| outputs=[story_audio, cursors, progress_comp] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| img_audio_combine_btn.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.video_gen, | |
| inputs=[ | |
| story_image, story_audio, story_content, cursors, cur_cursor | |
| ], | |
| outputs=[ | |
| story_image, story_audio, story_video, cursors, progress_comp | |
| ], | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| story_progress.input( | |
| story_gen_ui.move_story_cursor, | |
| inputs=[ | |
| story_progress, cursors | |
| ], | |
| outputs=[ | |
| cur_cursor, | |
| story_progress, | |
| story_content, | |
| story_image, story_audio, story_video, | |
| action_btn1, action_btn2, action_btn3, | |
| ] | |
| ) | |
| action_btn1.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.next_story_gen, | |
| inputs=[ | |
| cursors, | |
| action_btn1, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| cursors, cur_cursor, | |
| story_content, story_progress, | |
| image_gen_btn, audio_gen_btn, | |
| story_image, story_audio, story_video | |
| ] | |
| ).then( | |
| story_gen_ui.actions_gen, | |
| inputs=[ | |
| cursors, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| action_btn1, action_btn2, action_btn3, progress_comp | |
| ] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| action_btn2.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.next_story_gen, | |
| inputs=[ | |
| cursors, | |
| action_btn2, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| cursors, cur_cursor, | |
| story_content, story_progress, | |
| image_gen_btn, audio_gen_btn, | |
| story_image, story_audio, story_video | |
| ] | |
| ).then( | |
| story_gen_ui.actions_gen, | |
| inputs=[ | |
| cursors, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| action_btn1, action_btn2, action_btn3, progress_comp | |
| ] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| action_btn3.click( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.next_story_gen, | |
| inputs=[ | |
| cursors, | |
| action_btn3, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| cursors, cur_cursor, | |
| story_content, story_progress, | |
| image_gen_btn, audio_gen_btn, | |
| story_image, story_audio, story_video | |
| ] | |
| ).then( | |
| story_gen_ui.actions_gen, | |
| inputs=[ | |
| cursors, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| action_btn1, action_btn2, action_btn3, progress_comp | |
| ] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| custom_action_txt.submit( | |
| story_gen_ui.disable_btns, | |
| inputs=None, | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ).then( | |
| story_gen_ui.next_story_gen, | |
| inputs=[ | |
| cursors, | |
| custom_action_txt, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| cursors, cur_cursor, | |
| story_content, story_progress, | |
| image_gen_btn, audio_gen_btn, | |
| story_image, story_audio, story_video | |
| ] | |
| ).then( | |
| story_gen_ui.actions_gen, | |
| inputs=[ | |
| cursors, | |
| genre_dd, place_dd, mood_dd, | |
| name_txt1, age_dd1, personality_dd1, job_dd1, | |
| side_char_enable_ckb1, name_txt2, age_dd2, personality_dd2, job_dd2, | |
| side_char_enable_ckb2, name_txt3, age_dd3, personality_dd3, job_dd3, | |
| side_char_enable_ckb3, name_txt4, age_dd4, personality_dd4, job_dd4, | |
| ], | |
| outputs=[ | |
| action_btn1, action_btn2, action_btn3, progress_comp | |
| ] | |
| ).then( | |
| story_gen_ui.enable_btns, | |
| inputs=[story_image, story_audio], | |
| outputs=[ | |
| image_gen_btn, audio_gen_btn, img_audio_combine_btn, | |
| regen_actions_btn, regen_story_btn, custom_prompt_txt, | |
| action_btn1, action_btn2, action_btn3, | |
| custom_action_txt, | |
| restart_from_story_generation_btn, story_writing_done_btn | |
| ] | |
| ) | |
| ### Chatbot | |
| # chat_input_txt.submit( | |
| # chat_ui.chat, | |
| # inputs=[ | |
| # chat_input_txt, chat_mode, chat_state, | |
| # genre_dd, place_dd, mood_dd, | |
| # name_txt1, age_dd1, personality_dd1, job_dd1, | |
| # name_txt2, age_dd2, personality_dd2, job_dd2, | |
| # name_txt3, age_dd3, personality_dd3, job_dd3, | |
| # name_txt4, age_dd4, personality_dd4, job_dd4, | |
| # chapter1_title, chapter2_title, chapter3_title, chapter4_title, | |
| # chapter1_plot, chapter2_plot, chapter3_plot, chapter4_plot, | |
| # side_char_enable_ckb1, side_char_enable_ckb2, side_char_enable_ckb3, | |
| # ], | |
| # outputs=[chat_input_txt, chat_state, chatbot, regen_btn] | |
| # ) | |
| regen_btn.click( | |
| chat_ui.rollback_last_ui, | |
| inputs=[chatbot], outputs=[chatbot] | |
| ).then( | |
| chat_ui.chat_regen, | |
| inputs=[chat_mode, chat_state], | |
| outputs=[chat_state, chatbot] | |
| ) | |
| clear_btn.click( | |
| chat_ui.chat_reset, | |
| inputs=[chat_mode, chat_state], | |
| outputs=[chat_input_txt, chat_state, chatbot, regen_btn] | |
| ) | |
| char_gallery1.select( | |
| ui.update_selected_char_image, | |
| inputs=None, | |
| outputs=[selected_main_char_image1] | |
| ) | |
| char_gallery2.select( | |
| ui.update_selected_char_image, | |
| inputs=None, | |
| outputs=[selected_side_char_image1] | |
| ) | |
| char_gallery3.select( | |
| ui.update_selected_char_image, | |
| inputs=None, | |
| outputs=[selected_side_char_image2] | |
| ) | |
| char_gallery4.select( | |
| ui.update_selected_char_image, | |
| inputs=None, | |
| outputs=[selected_side_char_image3] | |
| ) | |
| demo.queue().launch() | |