Update app.py

app.py

@@ -10,19 +10,33 @@ def generate_script(prompt):
     result = generator(f"Scene: {prompt}\nDescription:", max_length=80, num_return_sequences=1)
     return result[0]["generated_text"]
 
-# === VOICE GENERATOR ===
+# === VOICE GENERATOR (CPU version using Bark) ===
 def generate_voice(script):
     with open("input.txt", "w") as f:
         f.write(script)
     os.system("python3 bark_infer_cpu.py --text_file input.txt --output bark_output.wav")
-    return "xtts_output.wav"
+    return "bark_output.wav"
+
+# === BACKGROUND MUSIC MIXER ===
+def merge_voice_and_music(voice_path, music_path, output_path="final_audio.wav"):
+    command = [
+        "ffmpeg", "-y",
+        "-i", voice_path,
+        "-i", music_path,
+        "-filter_complex", "[1:0]volume=0.3[a1];[0:0][a1]amix=inputs=2:duration=shortest",
+        "-c:a", "aac",
+        "-shortest",
+        output_path
+    ]
+    subprocess.call(command)
+    return output_path
 
 # === VIDEO GENERATOR ===
 def generate_video(prompt):
     os.system(f"python3 generate.py --prompt \"{prompt}\" --output modelscope_output.mp4")
     return "modelscope_output.mp4"
 
-# === MERGE AUDIO + VIDEO ===
+# === FINAL MERGE: AUDIO + VIDEO ===
 def merge_audio_video(audio_path, video_path):
     output = "final_output.mp4"
     subprocess.call([
@@ -31,25 +45,26 @@ def merge_audio_video(audio_path, video_path):
     ])
     return output
 
-# === FULL PIPELINE ===
+# === PIPELINE ===
 def full_pipeline(prompt):
     script = generate_script(prompt)
-    audio = generate_voice(script)
+    voice = generate_voice(script)
+    final_audio = merge_voice_and_music(voice, "bg_music.mp3")  # Replace with your background music file
     video = generate_video(prompt)
-    final = merge_audio_video(audio, video)
-    return final, script
+    final_video = merge_audio_video(final_audio, video)
+    return final_video, script
 
-# === GRADIO UI ===
+# === UI ===
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎬 Free Veo 3 Alternative – AI Video Generator")
-    gr.Markdown("Enter a scene below to generate video + voice using AI.")
+    gr.Markdown("# 🎬 Free Veo 3 Style AI Video Generator (with Background Music)")
+    gr.Markdown("Enter your scene. Get AI script + voice + music + video.")
 
     with gr.Row():
-        input_prompt = gr.Textbox(label="🎤 Your Scene Prompt", placeholder="e.g. A glowing temple in the rain")
-        generate_btn = gr.Button("🎞️ Generate")
+        input_prompt = gr.Textbox(label="🎤 Scene Prompt", placeholder="e.g. A monk walking through glowing forest in rain")
+        generate_btn = gr.Button("🎞️ Generate Video")
 
-    output_video = gr.Video(label="AI Generated Video")
-    output_script = gr.Textbox(label="Generated Script")
+    output_video = gr.Video(label="📽️ Final Video")
+    output_script = gr.Textbox(label="📝 AI Generated Script")
 
     generate_btn.click(fn=full_pipeline, inputs=input_prompt, outputs=[output_video, output_script])