Update app.py

app.py

@@ -1,67 +1,53 @@
-from fastapi import FastAPI, UploadFile, File, Form
-from fastapi.responses import FileResponse
-from moviepy.editor import *
+from fastapi import FastAPI, UploadFile, File
+from moviepy.editor import ImageClip, AudioFileClip, CompositeVideoClip, TextClip
 from gtts import gTTS
-import uuid, shutil, os
+import os
 
 app = FastAPI()
 
-@app.post("/generate")
-async def generate_video(
-    images: list[UploadFile] = File(...),
-    logo: UploadFile = File(...),
-    music: UploadFile = File(...),
-    narration_text: str = Form(...)
-):
-    uid = str(uuid.uuid4())
-    temp_dir = f"temp/{uid}"
-    os.makedirs(temp_dir, exist_ok=True)
+@app.post("/create-video/")
+async def create_video(image: UploadFile = File(...), caption: str = "مرحبا بك في الفيديو", logo: UploadFile = File(...)):
+    # Save uploaded files
+    image_path = "input.jpg"
+    logo_path = "logo.png"
+    await save_file(image, image_path)
+    await save_file(logo, logo_path)
 
-    # Save files
-    image_paths = []
-    for i, img in enumerate(images):
-        img_path = f"{temp_dir}/img_{i}.png"
-        with open(img_path, "wb") as f:
-            shutil.copyfileobj(img.file, f)
-        image_paths.append(img_path)
+    # Step 1: Create audio from caption
+    tts = gTTS(text=caption, lang="ar")
+    tts_path = "voice.mp3"
+    tts.save(tts_path)
 
-    logo_path = f"{temp_dir}/logo.png"
-    with open(logo_path, "wb") as f:
-        shutil.copyfileobj(logo.file, f)
+    # Step 2: Background music (replace with your own)
+    background_music = "calm.mp3"  # Place this in the root folder
 
-    music_path = f"{temp_dir}/music.mp3"
-    with open(music_path, "wb") as f:
-        shutil.copyfileobj(music.file, f)
+    # Step 3: Create base image clip
+    clip = ImageClip(image_path, duration=10).resize(height=720)
+    clip = clip.set_position("center").fx(vfx.zoom_in, 1.1)
 
-    # Generate Arabic voice
-    tts = gTTS(narration_text, lang='ar')
-    narration_path = f"{temp_dir}/narration.mp3"
-    tts.save(narration_path)
+    # Step 4: Add text caption
+    text = TextClip(caption, fontsize=40, color='white', font="Arial", method="caption", size=(clip.w * 0.9, None))
+    text = text.set_duration(10).set_position(("center", clip.h - 100))
 
-    # Create image clips with zoom effect
-    clips = []
-    for path in image_paths:
-        img_clip = ImageClip(path).set_duration(3).resize(width=720)
-        img_clip = img_clip.fx(vfx.zoom_in, 1.1)  # Simple Ken Burns
-        clips.append(img_clip)
+    # Step 5: Add logo
+    logo_clip = ImageClip(logo_path).set_duration(10).resize(height=80).set_position(("right", "top"))
 
-    video = concatenate_videoclips(clips, method="compose")
+    # Step 6: Add audio
+    voice_audio = AudioFileClip(tts_path)
+    bg_audio = AudioFileClip(background_music).volumex(0.1)
+    final_audio = CompositeAudioClip([bg_audio.set_duration(10), voice_audio.set_start(0)])
 
-    # Overlay logo
-    logo_clip = (ImageClip(logo_path)
-                 .set_duration(video.duration)
-                 .resize(height=80)
-                 .set_pos(("right", "top")))
-    final = CompositeVideoClip([video, logo_clip])
+    # Step 7: Combine all
+    final = CompositeVideoClip([clip, logo_clip, text])
+    final = final.set_audio(final_audio)
 
-    # Combine narration and music
-    narration_audio = AudioFileClip(narration_path)
-    bg_music = AudioFileClip(music_path).volumex(0.1)
-    audio = CompositeAudioClip([bg_music.set_duration(final.duration), narration_audio.set_start(0)])
-    final = final.set_audio(audio)
+    output_path = "final_video.mp4"
+    final.write_videofile(output_path, fps=24)
 
-    # Export
-    output_path = f"{temp_dir}/final_video.mp4"
-    final.write_videofile(output_path, fps=24, codec="libx264", audio_codec="aac")
+    return {"message": "🎬 Video created", "file": output_path}
 
-    return FileResponse(output_path, filename="final_video.mp4", media_type="video/mp4")
+
+async def save_file(uploaded: UploadFile, destination: str):
+    with open(destination, "wb") as f:
+        content = await uploaded.read()
+        f.write(content)