Spaces:
Runtime error
Runtime error
File size: 4,100 Bytes
d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 e363158 d895ad6 fe9822a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 | from src import gradio_fn
import gradio as gr
from src import script_gen
with gr.Blocks() as demo:
currComic = gr.State(None)
current_status = gr.State("")
episode_idx = gr.State(-1)
scene_idx = gr.State(0)
regenerate = gr.State(True)
instruction = gr.Markdown(value="Enter the comic meta data.")
with gr.Column(visible=True) as metadata:
with gr.Row():
comic_id = gr.Textbox(label="Enter Comic ID:", placeholder="Enter Comic ID")
comic_name = gr.Textbox(
label="Enter Comic Title:", placeholder="Enter Comic Title"
)
comic_location = gr.Textbox(
label="Enter Comic Location:", placeholder="Enter Comic Location"
)
comic_genre = gr.Dropdown(
choices=gradio_fn.GENRE_CHOICES, label="Current Genre", interactive=True
)
metadata_btn = gr.Button("Submit Metadata")
with gr.Column(visible=False) as synopsis:
with gr.Row():
user_input = gr.Textbox(label="Enter Your Idea for the comic:")
preamble = gr.Textbox(
label="Model Instruction:",
value=script_gen.generate_synopsis_instruction,
lines=10,
)
with gr.Row() as navigation:
previous_btn = gr.Button("Previous", visible=False)
generate_btn = gr.Button("Generate")
regenerate_btn = gr.Button("Re-Generate", visible=False)
next_btn = gr.Button("Next", visible=False)
model_response = gr.HTML(label="Model Output")
submit_btn = gr.Button("Submit", visible=False)
epi = gr.Textbox(label="Current Episode Synopsis", visible=False)
scene = gr.Textbox(label="Current Scene Synopsis", visible=False)
metadata_btn.click(
gradio_fn.load_metadata,
inputs=[comic_id, comic_name, comic_location, comic_genre],
outputs=[currComic, instruction, metadata, synopsis, current_status],
)
generate_btn.click(
gradio_fn.generate_llm_output,
inputs=[
user_input,
preamble,
current_status,
currComic,
episode_idx,
scene_idx,
],
outputs=[
previous_btn,
next_btn,
generate_btn,
regenerate_btn,
model_response,
submit_btn,
],
)
regenerate_btn.click(
gradio_fn.generate_llm_output,
inputs=[
user_input,
preamble,
current_status,
currComic,
episode_idx,
scene_idx,
regenerate, # regenrate option
],
outputs=[
previous_btn,
next_btn,
generate_btn,
regenerate_btn,
model_response,
submit_btn,
],
)
submit_btn.click(
gradio_fn.submit_data,
inputs=[current_status, currComic, episode_idx, model_response],
outputs=[
episode_idx,
current_status,
user_input,
preamble,
previous_btn,
next_btn,
generate_btn,
regenerate_btn,
model_response,
],
)
next_btn.click(
gradio_fn.next_button,
inputs=[current_status, currComic, episode_idx],
outputs=[
previous_btn,
next_btn,
generate_btn,
regenerate_btn,
model_response,
submit_btn,
episode_idx,
],
)
previous_btn.click(
gradio_fn.prev_button,
inputs=[current_status, currComic, episode_idx],
outputs=[
previous_btn,
next_btn,
generate_btn,
regenerate_btn,
model_response,
submit_btn,
episode_idx,
],
)
demo.launch(auth=("admin", "Qrt@12*34#immersfy"), share=True, ssr_mode=False, debug=True)
# demo.launch(share=True, ssr_mode=False, debug=True)
|