Update app.py

app.py

@@ -172,6 +172,12 @@ def create_interface():
         voices = await get_voices()
         return [v[1] for v in voices]
 
+    async def on_generate(voice, video_url, text):
+        if not voice:
+            return None, "Invalid voice selected."
+        voice_id = next(v[0] for v in await get_voices() if v[1] == voice)
+        return await process_video(voice_id, video_url, text)
+
     with gr.Blocks() as app:
         gr.Markdown("# JSON Train")
         with gr.Row():
@@ -180,13 +186,17 @@ def create_interface():
                 video_url_input = gr.Textbox(label="Enter Video URL")
                 text_input = gr.Textbox(label="Enter text", lines=3)
                 generate_btn = gr.Button("Generate Video")
-                return None, "Invalid voice selected."
-            return await process_video(voice_id, video_url, text)
+                output = gr.Textbox(label="Output")
 
         generate_btn.click(
             fn=on_generate,
             inputs=[voice_dropdown, video_url_input, text_input],
+            outputs=[output]
+        )
+
+    return app
 
 if __name__ == "__main__":
     app = create_interface()
+    app.queue(concurrency_count=5)  # Allow 5 concurrent tasks
     app.launch()