from Videobook import Videobook
import gradio as gr

gen = Videobook()
with gr.Blocks() as demo:
  with gr.Row():
    with gr.Column():
      story = gr.inputs.Textbox(lines = 5, label = "Story")
      api_key = gr.inputs.Textbox(label = "Segmind API Key")
      tags = gr.inputs.Textbox(default = "high quality, 3d render", label = "Tags")
      style = gr.Dropdown(["Cartoon", "Anime Style", "Realistic Image"], value = "Cartoon", label = "Style")
      model = gr.Dropdown(["Stable Diffusion v2.1", "Kadinsky"], value = "Stable Diffusion v2.1", label = "Model")
      with gr.Row():
        steps = gr.Radio([25, 50], value = 50, label = "Steps")
        fps = gr.Radio([10, 24, 60], value = 10, label = "FPS")
    output = gr.outputs.Video()
  run = gr.Button(label = "Generate Video")
  run.click(gen.generate, inputs = [story, api_key, fps, style, tags, model, steps], outputs = output)

demo.launch()