ffmpeg_cats

Running

App Files Files Community

Pepguy commited on 21 days ago

Commit

2f6bc27

verified ·

1 Parent(s): a662741

Update app.py

Browse files

Files changed (1) hide show

app.py +21 -124

app.py CHANGED Viewed

@@ -1,115 +1,27 @@
-from fastapi import FastAPI, HTTPException, Request
-from fastapi.responses import HTMLResponse
-from pydantic import BaseModel
 import subprocess
 import base64
 import os
 import uuid
 import shutil
-import yt_dlp
-import socket
-import re
-# Force IPv4
-orig_getaddrinfo = socket.getaddrinfo
-def hooked_getaddrinfo(*args, **kwargs):
-    res = orig_getaddrinfo(*args, **kwargs)
-    return [x for x in res if x[0] == socket.AF_INET]
-socket.getaddrinfo = hooked_getaddrinfo
 app = FastAPI()
-class VideoRequest(BaseModel):
-    url: str
-    platform: str
-    is_pro: bool = False
-    fps: int = 2
 def file_to_base64(filepath):
     if not os.path.exists(filepath): return None
     with open(filepath, "rb") as f:
         return base64.b64encode(f.read()).decode('utf-8')
-def clean_url(url: str):
-    # Removes trailing text like "This post is shared via..."
-    match = re.search(r'(https?://[^\s]+)', url)
-    return match.group(0) if match else url
-@app.get("/", response_class=HTMLResponse)
-def index():
-    return """
-    <!DOCTYPE html>
-    <html>
-    <head>
-        <title>Viral Cat | Media Worker</title>
-        <script src="https://cdn.tailwindcss.com"></script>
-        <style>
-            body { background: #090A0F; color: white; font-family: sans-serif; }
-            .neon { color: #12D8C3; text-shadow: 0 0 10px rgba(18,216,195,0.5); }
-            input { background: #16181F; border: 1px solid #2D3748; color: white; }
-        </style>
-    </head>
-    <body class="p-10">
-        <div class="max-w-2xl mx-auto bg-[#16181F] p-8 rounded-3xl border border-[#2D3748]">
-            <h1 class="text-3xl font-bold mb-2">Viral Cat <span class="neon">Media Worker</span></h1>
-            <p class="text-gray-400 mb-6">Test TikTok or Instagram links below.</p>
-            <input type="text" id="url" placeholder="Paste link here..." class="w-full p-4 rounded-xl mb-4">
-            <select id="platform" class="w-full p-4 rounded-xl mb-4 bg-[#090A0F] border border-[#2D3748]">
-                <option value="tiktok">TikTok</option>
-                <option value="instagram">Instagram Reel</option>
-            </select>
-            <button onclick="process()" id="btn" class="w-full bg-[#12D8C3] text-black font-bold p-4 rounded-xl">Process Video</button>
-            <div id="status" class="mt-6 text-sm text-gray-400"></div>
-            <div id="results" class="mt-6 grid grid-cols-4 gap-2"></div>
-        </div>
-        <script>
-            async function process() {
-                const btn = document.getElementById('btn');
-                const status = document.getElementById('status');
-                const results = document.getElementById('results');
-                btn.disabled = true;
-                btn.innerText = "Processing...";
-                status.innerText = "Downloading and extracting frames...";
-                results.innerHTML = "";
-                try {
-                    const res = await fetch('/process-video', {
-                        method: 'POST',
-                        headers: {'Content-Type': 'application/json'},
-                        body: JSON.stringify({
-                            url: document.getElementById('url').value,
-                            platform: document.getElementById('platform').value,
-                            fps: 2
-                        })
-                    });
-                    const data = await res.json();
-                    if(data.success) {
-                        status.innerText = "Success! Extracted " + data.total_frames + " frames.";
-                        data.frames.slice(0, 12).forEach(img => {
-                            results.innerHTML += `<img src="data:image/jpeg;base64,${img}" class="rounded-lg">`;
-                        });
-                    } else {
-                        status.innerText = "Error: " + data.error;
-                    }
-                } catch(e) {
-                    status.innerText = "Network Error: " + e.message;
-                }
-                btn.disabled = false;
-                btn.innerText = "Process Video";
-            }
-        </script>
-    </body>
-    </html>
-    """
 @app.post("/process-video")
-def process_video(req: VideoRequest):
     job_id = str(uuid.uuid4())
     work_dir = f"/tmp/viralcat_{job_id}"
     os.makedirs(work_dir, exist_ok=True)
@@ -117,29 +29,12 @@ def process_video(req: VideoRequest):
     video_path = os.path.join(work_dir, "video.mp4")
     audio_path = os.path.join(work_dir, "audio.mp3")
-    target_url = clean_url(req.url)
     try:
-        # 🚨 FORMAT FIX: Use 'best' to avoid split-stream errors on TikTok/IG
-        ydl_opts = {
-            'format': 'best[height<=720]',
-            'outtmpl': video_path,
-            'quiet': True,
-            'no_warnings': True,
-            'nocheckcertificate': True,
-            'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36',
-            'extractor_args': {'youtube': ['player_client=android,web']}
-        }
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.download([target_url])
-        if not os.path.exists(video_path):
-            files = os.listdir(work_dir)
-            if files: video_path = os.path.join(work_dir, files[0])
-            else: raise ValueError("Download failed - No file saved.")
-        # PROBE DURATION
         probe = subprocess.run([
             "ffprobe", "-v", "error", "-show_entries",
             "format=duration", "-of", "default=noprint_wrappers=1:nokey=1",
@@ -147,26 +42,27 @@ def process_video(req: VideoRequest):
         ], capture_output=True, text=True, check=True)
         duration = float(probe.stdout.strip() or 0)
-        if duration > 120 and not req.is_pro:
             raise ValueError(f"Video ({duration:.1f}s) exceeds 120s limit.")
-        # EXTRACT FRAMES (2 per second)
         subprocess.run([
             "ffmpeg", "-y", "-i", video_path,
-            "-vf", f"fps={req.fps}",
             "-q:v", "4", f"{work_dir}/frame_%04d.jpg"
         ], check=True, capture_output=True)
-        # EXTRACT 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)
         frame_files = sorted([f for f in os.listdir(work_dir) if f.startswith("frame_") and f.endswith(".jpg")])
-        if len(frame_files) > 80: frame_files = frame_files[:80]
-        frames_b64 = [file_to_base64(os.path.join(work_dir, f)) for f in frame_files]
         audio_b64 = file_to_base64(audio_path)
         return {
@@ -182,5 +78,6 @@ def process_video(req: VideoRequest):
         return {"success": False, "error": str(e)}
     finally:
         if os.path.exists(work_dir):
             shutil.rmtree(work_dir, ignore_errors=True)

+from fastapi import FastAPI, HTTPException, UploadFile, File, Form
 import subprocess
 import base64
 import os
 import uuid
 import shutil
 app = FastAPI()
+@app.get("/")
+def greet_json():
+    return {"status": "online", "engine": "Viral Cat Local Processor"}
 def file_to_base64(filepath):
     if not os.path.exists(filepath): return None
     with open(filepath, "rb") as f:
         return base64.b64encode(f.read()).decode('utf-8')
 @app.post("/process-video")
+async def process_video(
+    file: UploadFile = File(...),
+    fps: int = Form(2),
+    is_pro: bool = Form(False)
+):
     job_id = str(uuid.uuid4())
     work_dir = f"/tmp/viralcat_{job_id}"
     os.makedirs(work_dir, exist_ok=True)
     video_path = os.path.join(work_dir, "video.mp4")
     audio_path = os.path.join(work_dir, "audio.mp3")
     try:
+        # 1. Save the uploaded file locally
+        with open(video_path, "wb") as buffer:
+            shutil.copyfileobj(file.file, buffer)
+        # 2. Check Duration (120s limit for free)
         probe = subprocess.run([
             "ffprobe", "-v", "error", "-show_entries",
             "format=duration", "-of", "default=noprint_wrappers=1:nokey=1",
         ], capture_output=True, text=True, check=True)
         duration = float(probe.stdout.strip() or 0)
+        if duration > 120 and not is_pro:
             raise ValueError(f"Video ({duration:.1f}s) exceeds 120s limit.")
+        # 3. Extract Frames (2 per second)
         subprocess.run([
             "ffmpeg", "-y", "-i", video_path,
+            "-vf", f"fps={fps}",
             "-q:v", "4", f"{work_dir}/frame_%04d.jpg"
         ], check=True, capture_output=True)
+        # 4. Extract 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)
+        # 5. Base64 Conversion
         frame_files = sorted([f for f in os.listdir(work_dir) if f.startswith("frame_") and f.endswith(".jpg")])
+        if len(frame_files) > 80: frame_files = frame_files[:80] # Safety cap
+        frames_b64 =[file_to_base64(os.path.join(work_dir, f)) for f in frame_files]
         audio_b64 = file_to_base64(audio_path)
         return {
         return {"success": False, "error": str(e)}
     finally:
+        # CLEANUP
         if os.path.exists(work_dir):
             shutil.rmtree(work_dir, ignore_errors=True)