Update app.py

app.py

@@ -7,60 +7,48 @@ import json
 if not os.path.exists("node_modules"):
     subprocess.run(["npm", "install"], check=True)
 
-def check_orientation(url):
+def get_video_info(url):
     try:
-        cmd = [
-            "ffprobe", "-v", "error", "-select_streams", "v:0",
-            "-show_entries", "stream=width,height", "-of", "json", url
-        ]
+        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)
         w, h = data['streams'][0]['width'], data['streams'][0]['height']
-        if h > w: return "Portrait"
-        if w > h: return "Landscape"
-        return "Square"
+        return "Portrait" if h > w else "Landscape"
     except:
         return "Unknown"
 
 def process_video(query, target_orientation, start, end, mute, progress=gr.Progress()):
     if not query: return None, "⚠️ Enter a query"
     
-    progress(0, desc="🔍 Searching across platforms...")
+    progress(0, desc=f"🔍 Searching for {target_orientation} videos...")
     try:
-        result = subprocess.run(["node", "scraper.js", query], capture_output=True, text=True, timeout=300)
+        # Pass both query and orientation to Node
+        result = subprocess.run(["node", "scraper.js", query, target_orientation], capture_output=True, text=True, timeout=300)
         raw_lines = result.stdout.split('\n')
         
-        # Separate results by source
         pinterest_links = [l.split('|')[1] for l in raw_lines if l.startswith("SOURCE_PINTEREST")]
         pexels_links = [l.split('|')[1] for l in raw_lines if l.startswith("SOURCE_PEXELS")]
 
         final_url = None
-        current_source = ""
+        source_name = ""
 
-        # --- STEP 1: Try Pinterest ---
-        progress(0.2, desc=f"Checking Pinterest for {target_orientation}...")
+        # 1. Try Pinterest Results (Manual check)
         for url in pinterest_links:
-            ori = check_orientation(url)
-            if target_orientation == "Any" or ori == target_orientation:
-                final_url = url
-                current_source = "Pinterest"
+            if target_orientation == "Any" or check_orientation(url) == target_orientation:
+                final_url, source_name = url, "Pinterest"
                 break
         
-        # --- STEP 2: Automatic Fallback to Pexels ---
+        # 2. Fallback to Pexels (Guaranteed orientation via URL filter)
         if not final_url:
-            progress(0.5, desc=f"❌ No {target_orientation} on Pinterest. Falling back to Pexels...")
-            for url in pexels_links:
-                ori = check_orientation(url)
-                if target_orientation == "Any" or ori == target_orientation:
-                    final_url = url
-                    current_source = "Pexels (Fallback)"
-                    break
+            progress(0.5, desc="Pinterest failed. Checking Pexels...")
+            if pexels_links:
+                final_url, source_name = pexels_links[0], "Pexels (Fallback)"
 
         if not final_url:
-            return None, f"❌ Checked both Pinterest and Pexels. No {target_orientation} videos found for '{query}'."
+            return None, f"❌ No {target_orientation} videos found on Pinterest or Pexels for '{query}'."
 
-        # --- STEP 3: Process the winner ---
-        progress(0.9, desc=f"⚙️ Found on {current_source}! Processing...")
+        # 3. Process
+        progress(0.9, desc=f"⚙️ Found on {source_name}! Processing...")
         out_file = f"final_{uuid.uuid4().hex[:5]}.mp4"
         ffmpeg_cmd = [
             "ffmpeg", "-ss", str(start), "-to", str(end) if end > 0 else "999",
@@ -71,31 +59,27 @@ def process_video(query, target_orientation, start, end, mute, progress=gr.Progr
         else: ffmpeg_cmd.insert(-1, "-c:a"); ffmpeg_cmd.insert(-1, "aac")
 
         subprocess.run(ffmpeg_cmd, check=True)
-        return out_file, f"✅ Success! | Source: {current_source} | Mode: {target_orientation}"
+        return out_file, f"✅ Found on {source_name}!"
 
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
 
+def check_orientation(url):
+    return get_video_info(url)
+
 # --- UI ---
-with gr.Blocks(theme=gr.themes.Default()) as demo:
-    gr.Markdown("# 🚀 AI Video Scraper & Editor")
-    gr.Markdown("**Smart Fallback:** If Pinterest doesn't have your Portrait/Landscape choice, it auto-switches to Pexels.")
-    
+with gr.Blocks() as demo:
+    gr.Markdown("# 🚀 Ultimate AI Video Scraper")
     with gr.Row():
         with gr.Column():
-            q = gr.Textbox(label="Search Query", placeholder="e.g. Cyberpunk City")
-            o = gr.Dropdown(choices=["Any", "Portrait", "Landscape"], value="Portrait", label="Orientation Preference")
+            q = gr.Textbox(label="Query", placeholder="crowded city aerial view")
+            o = gr.Dropdown(choices=["Portrait", "Landscape"], value="Portrait", label="Orientation")
             with gr.Row():
-                s = gr.Number(label="Start Sec", value=0)
-                e = gr.Number(label="End Sec", value=5)
+                s, e = gr.Number(label="Start", value=0), gr.Number(label="End", value=5)
             mute = gr.Checkbox(label="Mute Audio")
             btn = gr.Button("Find & Download", variant="primary")
-        
         with gr.Column():
-            out_v = gr.Video(label="Result Video")
-            out_s = gr.Textbox(label="Status / Source Info")
-
+            out_v, out_s = gr.Video(), gr.Textbox(label="Status")
     btn.click(process_video, [q, o, s, e, mute], [out_v, out_s])
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()