Update main.py

main.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel, HttpUrl
-from typing import List
+from typing import List, Optional
 import os
 import asyncio
 import uuid
@@ -23,6 +23,7 @@ class SlideshowRequest(BaseModel):
     image_urls: List[HttpUrl]
     audio_url: HttpUrl
     duration: int
+    zoom: Optional[bool] = False  # Optional zoom parameter with default value of False
 
 def extract_google_drive_id(url):
     """Extract file ID from a Google Drive URL"""
@@ -54,7 +55,7 @@ async def download_file(url, local_path):
         print(f"Error downloading {url}: {str(e)}")
         return False
 
-async def create_slideshow(image_paths, audio_path, output_path, duration):
+async def create_slideshow(image_paths, audio_path, output_path, duration, zoom=False):
     """Generate slideshow from images and audio using ffmpeg asynchronously"""
     # Create temporary file list for ffmpeg concat
     concat_file = "temp_concat.txt"
@@ -70,12 +71,32 @@ async def create_slideshow(image_paths, audio_path, output_path, duration):
     
     # Run ffmpeg command to create slideshow with audio
     total_duration = len(image_paths) * duration
+
     cmd = [
         "ffmpeg",
         "-f", "concat",
         "-safe", "0",
         "-i", concat_file,
         "-i", audio_path,
+    ]
+    
+    # Add zoom effect if requested
+    if zoom:
+        for i, img_path in enumerate(image_paths):
+            cmd.extend([
+                "-filter_complex", 
+                f"[0:v]select=eq(n\\,{i}),setpts=PTS-STARTPTS,scale=1920:1080,zoompan=z='min(1.1,1+(in/out/8))':"
+                f"d={duration*25}:s=1920x1080:fps=25[v{i}]"
+            ])
+        
+        # Create video filter chain
+        filter_complex = ""
+        for i in range(len(image_paths)):
+            filter_complex += f"[v{i}]"
+        filter_complex += f"concat=n={len(image_paths)}:v=1:a=0[outv]"
+        cmd.extend(["-filter_complex", filter_complex, "-map", "[outv]", "-map", "1:a"])
+    
+    cmd.extend([
         "-c:v", "libx264",
         "-pix_fmt", "yuv420p",
         "-c:a", "aac",
@@ -83,7 +104,7 @@ async def create_slideshow(image_paths, audio_path, output_path, duration):
         "-y",
         "-t", str(total_duration),
         output_path
-    ]
+    ])
     
     try:
         process = await asyncio.create_subprocess_exec(
@@ -145,7 +166,7 @@ async def make_slideshow(request: SlideshowRequest):
         output_path = os.path.join(request_dir, "slideshow.mp4")
         
         # Generate slideshow
-        if not await create_slideshow(image_paths, audio_path, output_path, request.duration):
+        if not await create_slideshow(image_paths, audio_path, output_path, request.duration, request.zoom):
             raise HTTPException(status_code=500, detail="Failed to create slideshow")
         
         # Return URL to the video