Spaces:
Running
Running
File size: 3,530 Bytes
66af092 |
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 |
import gradio as gr
from openai import OpenAI
import os
import edge_tts
import json
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
async def generate_video(topic, progress=gr.Progress()):
try:
# Ensure audio directory exists
os.makedirs("audio", exist_ok=True)
SAMPLE_FILE_NAME = os.path.join("audio", "audio_tts.wav")
VIDEO_SERVER = "pexel"
progress(0.1, desc="Generating script...")
response = generate_script(topic)
progress(0.3, desc="Generating audio...")
await generate_audio(response, SAMPLE_FILE_NAME)
progress(0.5, desc="Generating captions...")
timed_captions = generate_timed_captions(SAMPLE_FILE_NAME)
progress(0.6, desc="Generating search terms...")
search_terms = getVideoSearchQueriesTimed(response, timed_captions)
if search_terms is not None:
progress(0.7, desc="Fetching background videos...")
background_video_urls = generate_video_url(search_terms, VIDEO_SERVER)
background_video_urls = merge_empty_intervals(background_video_urls)
if background_video_urls is not None:
progress(0.9, desc="Rendering final video...")
video = get_output_media(SAMPLE_FILE_NAME, timed_captions, background_video_urls, VIDEO_SERVER)
return video, "Video generated successfully!"
return None, "Failed to generate video. No suitable background videos found."
except Exception as e:
return None, f"Error: {str(e)}"
# Create Gradio interface
def create_interface():
with gr.Blocks(theme=gr.themes.Soft()) as demo:
gr.Markdown(
"""
# AI Video Generator
Generate engaging videos from any topic using AI
"""
)
with gr.Row():
with gr.Column():
topic_input = gr.Textbox(
label="Enter Your Topic",
placeholder="E.g., The history of artificial intelligence",
lines=2
)
generate_btn = gr.Button("Generate Video", variant="primary")
with gr.Column():
output_video = gr.Video(label="Generated Video")
status_text = gr.Textbox(label="Status", interactive=False)
generate_btn.click(
fn=generate_video,
inputs=[topic_input],
outputs=[output_video, status_text]
)
gr.Markdown(
"""
### How it works:
1. Enter a topic you want to create a video about
2. Click 'Generate Video'
3. Wait while the AI generates the script, audio, and matches appropriate visuals
4. Download your finished video!
"""
)
return demo
if __name__ == "__main__":
demo = create_interface()
demo.launch(share=True)
|