Update app.py

app.py

@@ -1,16 +1,95 @@
-from fastapi import FastAPI
+# media_server.py
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 import subprocess
+import base64
+import os
+import uuid
+import shutil
 
 app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+class VideoRequest(BaseModel):
+    url: str
+    platform: str
+    is_pro: bool = False
+    fps: int = 2 # Default to 2 frames per second
 
-@app.get("/ffmpeg-version")
-def get_ffmpeg_version():
+def file_to_base64(filepath):
+    with open(filepath, "rb") as f:
+        return base64.b64encode(f.read()).decode('utf-8')
+
+@app.post("/process-video")
+def process_video(req: VideoRequest):
+    # 1. Create a dedicated folder for this specific task
+    job_id = str(uuid.uuid4())
+    work_dir = f"/tmp/viralcat_{job_id}"
+    os.makedirs(work_dir, exist_ok=True)
+    
+    video_path = f"{work_dir}/video.mp4"
+    audio_path = f"{work_dir}/audio.mp3"
+    
     try:
-        result = subprocess.run(["ffmpeg", "-version"], capture_output=True, text=True)
-        return {"ffmpeg_version": result.stdout.splitlines()[0]}  # Return only the first line
-    except FileNotFoundError:
-        return {"error": "FFmpeg is not installed or not found in PATH"}
+        # 2. Download the Video
+        subprocess.run([
+            "yt-dlp", 
+            "-f", "best[height<=720]", 
+            "-o", video_path, 
+            req.url
+        ], check=True, capture_output=True)
+        
+        # 3. Check Duration (Limit to 2 minutes / 120 seconds if not Pro)
+        probe = subprocess.run([
+            "ffprobe", "-v", "error", "-show_entries", 
+            "format=duration", "-of", "default=noprint_wrappers=1:nokey=1", 
+            video_path
+        ], capture_output=True, text=True, check=True)
+        
+        duration = float(probe.stdout.strip())
+        
+        if duration > 120 and not req.is_pro:
+            raise ValueError(f"Video is {duration:.1f}s long. Free users are limited to 120 seconds.")
+
+        # 4. Dynamic Frame Extraction (e.g., 2 frames per second)
+        # This creates files like frame_0001.jpg, frame_0002.jpg inside the work_dir
+        subprocess.run([
+            "ffmpeg", "-y", "-i", video_path, 
+            "-vf", f"fps={req.fps}", 
+            "-q:v", "2", f"{work_dir}/frame_%04d.jpg"
+        ], check=True, capture_output=True)
+
+        # 5. Extract Uncut Audio
+        subprocess.run([
+            "ffmpeg", "-y", "-i", video_path, 
+            "-q:a", "0", "-map", "a", "-ac", "1", "-b:a", "64k", audio_path
+        ], check=True, capture_output=True)
+
+        # 6. Gather all extracted frames and convert to Base64
+        # Sort them so they are in chronological order for the AI
+        frame_files = sorted([f for f in os.listdir(work_dir) if f.startswith("frame_") and f.endswith(".jpg")])
+        frames_b64 =[file_to_base64(os.path.join(work_dir, f)) for f in frame_files]
+        
+        # Get audio
+        audio_b64 = file_to_base64(audio_path) if os.path.exists(audio_path) else None
+
+        return {
+            "success": True,
+            "duration": duration,
+            "total_frames_extracted": len(frames_b64),
+            "frames": frames_b64,
+            "audio": audio_b64,
+            "thumbnail_url": "https://placehold.co/400x600/png"
+        }
+
+    except ValueError as ve:
+        raise HTTPException(status_code=400, detail=str(ve))
+    except subprocess.CalledProcessError as e:
+        raise HTTPException(status_code=500, detail=f"Media processing failed: {e.stderr.decode() if e.stderr else str(e)}")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+        
+    finally:
+        # 🚨 ABSOLUTE CLEANUP: Delete the entire UUID folder and all its contents
+        if os.path.exists(work_dir):
+            shutil.rmtree(work_dir, ignore_errors=True)
+            print(f"[CLEANUP] Deleted temporary folder: {work_dir}")