Update app.py

app.py

@@ -1,48 +1,61 @@
-from openai import OpenAI
-import os
-import edge_tts
-import json
+import gradio as gr
 import asyncio
-import whisper_timestamped as whisper
 from utility.script_generator import generate_script
 from utility.audio_generator import generate_audio
 from utility.timed_captions_generator import generate_timed_captions
 from utility.background_video_generator import generate_video_url
 from utility.render_engine import get_output_media
 from utility.video_search_query_generator import getVideoSearchQueriesTimed, merge_empty_intervals
-import argparse
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Generate a video from a topic.")
-    parser.add_argument("topic", type=str, help="The topic for the video")
-
-    args = parser.parse_args()
-    SAMPLE_TOPIC = args.topic
-    SAMPLE_FILE_NAME = "audio_tts.wav"
-    VIDEO_SERVER = "pexel"
-
-    response = generate_script(SAMPLE_TOPIC)
-    print("script: {}".format(response))
-
-    asyncio.run(generate_audio(response, SAMPLE_FILE_NAME))
 
+SAMPLE_FILE_NAME = "audio_tts.wav"
+VIDEO_SERVER = "pexel"
+
+async def process_topic(topic):
+    # Generate script from topic
+    response = generate_script(topic)
+    print("Script:", response)
+    
+    # Generate audio from script
+    await generate_audio(response, SAMPLE_FILE_NAME)
+    
+    # Generate timed captions from audio
     timed_captions = generate_timed_captions(SAMPLE_FILE_NAME)
-    print(timed_captions)
-
+    print("Timed Captions:", timed_captions)
+    
+    # Generate video search queries based on the script and captions
     search_terms = getVideoSearchQueriesTimed(response, timed_captions)
-    print(search_terms)
-
+    print("Search Terms:", search_terms)
+    
+    # Get background video URLs
     background_video_urls = None
     if search_terms is not None:
         background_video_urls = generate_video_url(search_terms, VIDEO_SERVER)
-        print(background_video_urls)
+        print("Background Video URLs:", background_video_urls)
     else:
         print("No background video")
-
+    
+    # Merge empty intervals
     background_video_urls = merge_empty_intervals(background_video_urls)
-
+    
+    # Generate the final video
     if background_video_urls is not None:
         video = get_output_media(SAMPLE_FILE_NAME, timed_captions, background_video_urls, VIDEO_SERVER)
-        print(video)
+        print("Generated Video:", video)
+        return video
     else:
-        print("No video")
+        print("No video generated")
+        return "No video generated"
+
+def generate_video(topic):
+    video_url = asyncio.run(process_topic(topic))
+    return video_url
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Video Generator from Topic")
+    topic_input = gr.Textbox(label="Enter the Topic", placeholder="Type a topic for the video...")
+    output_video = gr.Video(label="Generated Video")
+    generate_btn = gr.Button("Generate Video")
+    
+    generate_btn.click(fn=generate_video, inputs=topic_input, outputs=output_video)
+
+demo.launch()