Update app.py

app.py

@@ -8,9 +8,9 @@ import uuid
 import shutil
 import yt_dlp
 import socket
+import re
 
-# ── 1. SOCKET HACK: FORCE IPv4 ──
-# Prevents "[Errno -5] No address associated with hostname" on cloud hosts
+# Force IPv4
 orig_getaddrinfo = socket.getaddrinfo
 def hooked_getaddrinfo(*args, **kwargs):
     res = orig_getaddrinfo(*args, **kwargs)
@@ -21,7 +21,7 @@ app = FastAPI()
 
 class VideoRequest(BaseModel):
     url: str
-    platform: str # 'tiktok' or 'instagram'
+    platform: str
     is_pro: bool = False
     fps: int = 2
 
@@ -30,7 +30,11 @@ def file_to_base64(filepath):
     with open(filepath, "rb") as f:
         return base64.b64encode(f.read()).decode('utf-8')
 
-# ── 2. TEST UI ROUTE ──
+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 """
@@ -70,7 +74,7 @@ def index():
                 const results = document.getElementById('results');
                 
                 btn.disabled = true;
-                btn.innerText = "Processing... (may take 20s)";
+                btn.innerText = "Processing...";
                 status.innerText = "Downloading and extracting frames...";
                 results.innerHTML = "";
 
@@ -87,11 +91,11 @@ def index():
                     const data = await res.json();
                     if(data.success) {
                         status.innerText = "Success! Extracted " + data.total_frames + " frames.";
-                        data.frames.slice(0, 8).forEach(img => {
+                        data.frames.slice(0, 12).forEach(img => {
                             results.innerHTML += `<img src="data:image/jpeg;base64,${img}" class="rounded-lg">`;
                         });
                     } else {
-                        status.innerText = "Error: " + data.detail;
+                        status.innerText = "Error: " + data.error;
                     }
                 } catch(e) {
                     status.innerText = "Network Error: " + e.message;
@@ -104,7 +108,6 @@ def index():
     </html>
     """
 
-# ── 3. MEDIA PROCESSING ROUTE ──
 @app.post("/process-video")
 def process_video(req: VideoRequest):
     job_id = str(uuid.uuid4())
@@ -114,28 +117,24 @@ 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:
-        # TikTok/IG require specific headers to not get 403 Forbidden
+        # 🚨 FORMAT FIX: Use 'best' to avoid split-stream errors on TikTok/IG
         ydl_opts = {
-            'format': 'bestvideo[height<=720]+bestaudio/best[height<=720]',
+            'format': 'best[height<=720]',
             'outtmpl': video_path,
             'quiet': True,
             'no_warnings': True,
             'nocheckcertificate': True,
-            # Disguise as a standard mobile browser
-            'user_agent': 'Mozilla/5.0 (iPhone; CPU iPhone OS 16_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.0 Mobile/15E148 Safari/604.1',
-            'add_header': [
-                'Accept: text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
-                'Accept-Language: en-us',
-                'Connection: keep-alive',
-            ]
+            '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([req.url])
+            ydl.download([target_url])
 
         if not os.path.exists(video_path):
-            # Find the actual file if extension differed
             files = os.listdir(work_dir)
             if files: video_path = os.path.join(work_dir, files[0])
             else: raise ValueError("Download failed - No file saved.")
@@ -164,10 +163,7 @@ def process_video(req: VideoRequest):
             "-q:a", "0", "-map", "a", "-ac", "1", "-b:a", "64k", audio_path
         ], check=True, capture_output=True)
 
-        # COLLECT DATA
         frame_files = sorted([f for f in os.listdir(work_dir) if f.startswith("frame_") and f.endswith(".jpg")])
-        
-        # Max limit for AI context windows
         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]
@@ -186,6 +182,5 @@ def process_video(req: VideoRequest):
         return {"success": False, "error": str(e)}
         
     finally:
-        # 🚨 GARBAGE COLLECTION
         if os.path.exists(work_dir):
             shutil.rmtree(work_dir, ignore_errors=True)