Update main gradio block

app.py

@@ -1,34 +1,32 @@
+import animation
 import bgm
+import narrator
+import story
+import subtitles
 import gradio as gr
 
-def show_textbox(radio_state):
-    if(radio_state == "User Input"):
-        return [gr.Textbox.update(visible=False), gr.Textbox.update(visible=False), gr.Textbox.update(visible=True)]
-    elif(radio_state == "ChatGPT"):
-        return [gr.Textbox.update(visible=True), gr.Textbox.update(visible=True), gr.Textbox.update(visible=False)]
-    else:
-        return [gr.Textbox.update(visible=False) for _ in range(2)]
+def generate_video(text, auth_openai, auth_elevenlabs, auth_replicate, auth_rev):
+    generated_story = story.text2story(text, auth_openai)
+    narrator.text2voice(generated_story, "audio_out.mp3", auth_elevenlabs, 5)
+    deforum_str, max_frames = subtitles.audio2subtitle(auth_rev)
+    generated_animation = animation.story2video(deforum_str, max_frames, auth_replicate)
+    # generated_music = bgm.text2audio(text=text, duration=20, guidance_scale=5, random_seed=24, n_candidates=3)
 
-def generate_video(text_in):
-    ambient_music = bgm.text2audio(text=text_in, duration=20, guidance_scale=5, random_seed=24, n_candidates=3)
-    print(ambient_music)
-    return ambient_music
+    return generated_animation
 
 def download_video(v):
     pass
 
 with gr.Blocks() as demo:
-    gr.Markdown("Generate a narrated horror story video from a text prompt or your own story.")
+    gr.Markdown("Generate a narrated horror story video from a text prompt.")
 
     with gr.Row():
         with gr.Column():
-            radio_input = gr.Radio(["ChatGPT", "User Input"], type="value", label="Input method")
-            auth_input = gr.Textbox(label="Auth Key", max_lines=1, visible=False, interactive=True)
-            prompt_input = gr.Textbox(label="Prompt", lines=5, visible=False, interactive=True)
-            text_input = gr.Textbox(label="Your Story", lines=5, visible=False, interactive=True)
-
-            text_lst = [auth_input, prompt_input, text_input]
-            radio_input.change(show_textbox, radio_input, text_lst)
+            auth_openai_input = gr.Textbox(label="OpenAI Auth Key", max_lines=1, type='password')
+            auth_eleven_input = gr.Textbox(label="Eleven Labs Auth Key", max_lines=1, type='password')
+            auth_replicate_input = gr.Textbox(label="Replicate Auth Key", max_lines=1, type='password')
+            auth_rev_input = gr.Textbox(label="Rev AI Auth Key", max_lines=1, type='password')
+            prompt_input = gr.Textbox(label="Write me a story about...", lines=5)
 
             generate_button = gr.Button("Generate Video")
 
@@ -36,7 +34,7 @@ with gr.Blocks() as demo:
             video_out = gr.Video(label="Output", interactive=False)
             download_button = gr.Button("Download")
 
-    generate_button.click(generate_video, inputs=[text_input], outputs=[video_out])
+    generate_button.click(generate_video, inputs=[prompt_input, auth_openai_input, auth_eleven_input, auth_replicate_input, auth_rev_input], outputs=[video_out])
     download_button.click(download_video, inputs=video_out)
 
 demo.launch(debug=True, enable_queue=True)