Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import subprocess
 import os
 import uuid
+import json
 
 def initialize_env():
     if not os.path.exists("node_modules"):
@@ -9,86 +10,99 @@ def initialize_env():
 
 initialize_env()
 
+def get_video_dimensions(url):
+    """Checks video dimensions using ffprobe"""
+    try:
+        cmd = [
+            "ffprobe", "-v", "error", "-select_streams", "v:0",
+            "-show_entries", "stream=width,height", "-of", "json", url
+        ]
+        result = subprocess.run(cmd, capture_output=True, text=True, timeout=10)
+        data = json.loads(result.stdout)
+        width = data['streams'][0]['width']
+        height = data['streams'][0]['height']
+        return width, height
+    except:
+        return 0, 0
+
 def process_video(query, orientation, start_time, end_time, progress=gr.Progress()):
     if not query:
         return None, "⚠️ Please enter a keyword."
     
-    progress(0, desc="🔍 Searching for High Quality Pins...")
+    progress(0, desc="🔍 Searching Pinterest...")
     search_query = query if "video" in query.lower() else f"{query} video"
     
     try:
+        # 1. Run Node Scraper (collects many links)
         result = subprocess.run(
             ["node", "scraper.js", search_query],
-            capture_output=True, text=True, timeout=180
+            capture_output=True, text=True, timeout=180 # Increased timeout for deep search
         )
         
-        # Parse the custom output format: URL|WIDTH|HEIGHT
-        raw_output = result.stdout.strip().split('\n')
-        video_pool = []
-        for line in raw_output:
-            if '|' in line:
-                parts = line.split('|')
-                video_pool.append({"url": parts[0], "w": int(parts[1]), "h": int(parts[2])})
-
-        if not video_pool:
-            return None, "❌ No high-quality videos found."
+        all_urls = [line.strip() for line in result.stdout.split('\n') if line.strip().startswith("http")]
+        
+        if not all_urls:
+            return None, "❌ No videos found in the search results."
 
-        # Filter by orientation using Metadata (Faster than ffprobe!)
-        selected_video = None
-        for i, vid in enumerate(video_pool):
-            progress((i+1)/len(video_pool), desc=f"Filtering for {orientation}...")
+        # 2. Loop through every found URL until the orientation matches
+        selected_url = None
+        total = len(all_urls)
+        
+        for i, url in enumerate(all_urls):
+            progress((i+1)/total, desc=f"Checking video {i+1}/{total} for {orientation}...")
             
             if orientation == "Any":
-                selected_video = vid
+                selected_url = url
                 break
-            if orientation == "Portrait" and vid['h'] > vid['w']:
-                selected_video = vid
+            
+            w, h = get_video_dimensions(url)
+            if orientation == "Portrait" and h > w:
+                selected_url = url
                 break
-            if orientation == "Landscape" and vid['w'] > vid['h']:
-                selected_video = vid
+            if orientation == "Landscape" and w > h:
+                selected_url = url
                 break
+            # If no match, the loop continues to the next URL automatically
         
-        if not selected_video:
-            return None, f"❌ Found {len(video_pool)} videos, but none were {orientation}."
+        if not selected_url:
+            return None, f"❌ Searched {total} videos, but none were {orientation}. Pinterest is mostly Portrait content."
 
-        # 3. HIGH QUALITY TRIMMING
-        progress(0.9, desc="💎 Processing High Quality Output...")
-        output_filename = f"hd_clip_{uuid.uuid4().hex[:8]}.mp4"
-        
-        # FFmpeg Pro Settings:
-        # -crf 18: Visually lossless quality
-        # -preset slow: Better compression/clarity
-        # -c:v libx264: Industry standard HD codec
+        # 3. Trim and return the video
+        progress(0.9, desc="✂️ Trimming video...")
+        output_filename = f"final_{uuid.uuid4().hex[:8]}.mp4"
         ffmpeg_cmd = [
             "ffmpeg", "-ss", str(start_time), "-to", str(end_time),
-            "-i", selected_video['url'],
-            "-c:v", "libx264", "-crf", "18", "-preset", "slow",
-            "-c:a", "aac", "-b:a", "192k", "-y", output_filename
+            "-i", selected_url, "-c:v", "libx264", "-c:a", "aac", "-y", output_filename
         ]
         
         if end_time <= 0:
             ffmpeg_cmd.pop(3); ffmpeg_cmd.pop(3)
 
         subprocess.run(ffmpeg_cmd, check=True)
-        return output_filename, f"✅ HD Video Processed! (Res: {selected_video['w']}x{selected_video['h']})"
+        return output_filename, f"✅ Success! Found {orientation} video after checking {i+1} results."
 
     except Exception as e:
-        return None, f"❌ Error: {str(e)}"
+        return None, f"❌ System Error: {str(e)}"
 
-# UI remains same but works faster and better
+# --- UI Layout ---
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 💎 Pinterest HD Search & Trim")
+    gr.Markdown("# 📌 Pinterest Deep Search & Filter")
+    gr.Markdown("This app will scroll and search deep into Pinterest until it finds the orientation you want.")
+    
     with gr.Row():
         with gr.Column():
-            query_input = gr.Textbox(label="Topic", placeholder="e.g. Minecraft Gameplay")
-            filter_input = gr.Dropdown(choices=["Any", "Portrait", "Landscape"], value="Landscape", label="Orientation")
+            query_input = gr.Textbox(label="Topic", placeholder="e.g. Ocean Nature")
+            filter_input = gr.Dropdown(choices=["Any", "Portrait", "Landscape"], value="Landscape", label="Required Orientation")
+            
             with gr.Row():
                 start_input = gr.Number(label="Start (sec)", value=0)
                 end_input = gr.Number(label="End (sec, 0=full)", value=5)
-            btn = gr.Button("🚀 Get HD Clip", variant="primary")
+            
+            btn = gr.Button("🚀 Deep Search & Process", variant="primary")
+        
         with gr.Column():
-            video_output = gr.Video(label="HD Result")
-            status_output = gr.Textbox(label="Status")
+            video_output = gr.Video(label="Trimmed Result")
+            status_output = gr.Textbox(label="Search Progress / Status")
 
     btn.click(process_video, [query_input, filter_input, start_input, end_input], [video_output, status_output])