Update app.py

app.py

@@ -4,14 +4,11 @@ import os
 import uuid
 import json
 
-def initialize_env():
-    if not os.path.exists("node_modules"):
-        subprocess.run(["npm", "install"], check=True)
-
-initialize_env()
+# Startup: Install Node modules if missing
+if not os.path.exists("node_modules"):
+    subprocess.run(["npm", "install"], check=True)
 
 def get_video_dimensions(url):
-    """Checks video dimensions using ffprobe"""
     try:
         cmd = [
             "ffprobe", "-v", "error", "-select_streams", "v:0",
@@ -19,9 +16,7 @@ def get_video_dimensions(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
+        return data['streams'][0]['width'], data['streams'][0]['height']
     except:
         return 0, 0
 
@@ -29,81 +24,85 @@ 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 Pinterest...")
+    progress(0, desc="🔍 Searching deep on Pinterest...")
     search_query = query if "video" in query.lower() else f"{query} video"
     
     try:
-        # 1. Run Node Scraper (collects many links)
+        # 1. Run Node Scraper
         result = subprocess.run(
             ["node", "scraper.js", search_query],
-            capture_output=True, text=True, timeout=180 # Increased timeout for deep search
+            capture_output=True, text=True, timeout=240
         )
-        
         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."
+            return None, "❌ No videos found in search."
 
-        # 2. Loop through every found URL until the orientation matches
+        # 2. Filter for match
         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}...")
-            
+            progress((i+1)/len(all_urls), desc=f"Checking video {i+1} for {orientation}...")
             if orientation == "Any":
-                selected_url = url
-                break
-            
+                selected_url = url; break
             w, h = get_video_dimensions(url)
             if orientation == "Portrait" and h > w:
-                selected_url = url
-                break
+                selected_url = url; break
             if orientation == "Landscape" and w > h:
-                selected_url = url
-                break
-            # If no match, the loop continues to the next URL automatically
+                selected_url = url; break
         
         if not selected_url:
-            return None, f"❌ Searched {total} videos, but none were {orientation}. Pinterest is mostly Portrait content."
+            return None, f"❌ Checked {len(all_urls)} videos, but no {orientation} found."
+
+        # 3. ULTRA 4K ENHANCEMENT & 9:16 CROP
+        # We use 2160:3840 for 4K Portrait 9:16
+        progress(0.8, desc="💎 Upscaling to 4K & Sharpening... (Takes time)")
+        output_filename = f"enhanced_4k_{uuid.uuid4().hex[:5]}.mp4"
+        
+        # Filter: Scale to fit 4K, Crop to 9:16, Sharpening (Unsharp mask)
+        filter_complex = (
+            "scale=2160:3840:force_original_aspect_ratio=increase," # Upscale
+            "crop=2160:3840,"                                       # Force 9:16
+            "unsharp=5:5:1.5:5:5:0.0"                               # Sharpening
+        )
 
-        # 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_url, "-c:v", "libx264", "-c:a", "aac", "-y", output_filename
+            "ffmpeg",
+            "-ss", str(start_time),
+            "-to", str(end_time) if end_time > 0 else "999",
+            "-i", selected_url,
+            "-vf", filter_complex,
+            "-c:v", "libx264",
+            "-preset", "medium", # Balanced speed/quality for HF CPU
+            "-crf", "18",        # High quality bitrate
+            "-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"✅ Success! Found {orientation} video after checking {i+1} results."
+        return output_filename, f"✅ 4K Sharpened & Cropped (Checked {i+1} pins)"
 
     except Exception as e:
-        return None, f"❌ System Error: {str(e)}"
+        return None, f"❌ Error: {str(e)}"
 
-# --- UI Layout ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 📌 Pinterest Deep Search & Filter")
-    gr.Markdown("This app will scroll and search deep into Pinterest until it finds the orientation you want.")
+# --- Gradio Interface ---
+with gr.Blocks() as demo:
+    gr.Markdown("# 📌 Pinterest Ultra-HD 4K Downloader")
+    gr.Markdown("Deep searches Pinterest, filters by orientation, and enhances video to **4K (2160x3840)** with **Sharpening Filters**.")
     
     with gr.Row():
         with gr.Column():
-            query_input = gr.Textbox(label="Topic", placeholder="e.g. Ocean Nature")
-            filter_input = gr.Dropdown(choices=["Any", "Portrait", "Landscape"], value="Landscape", label="Required Orientation")
-            
+            q = gr.Textbox(label="Topic", placeholder="e.g. Nature Relaxing")
+            o = 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("🚀 Deep Search & Process", variant="primary")
+                s = gr.Number(label="Start (sec)", value=0)
+                e = gr.Number(label="End (sec, 0=Full)", value=5)
+            btn = gr.Button("🚀 Deep Search & Enhance", variant="primary")
         
         with gr.Column():
-            video_output = gr.Video(label="Trimmed Result")
-            status_output = gr.Textbox(label="Search Progress / Status")
+            out_v = gr.Video(label="Enhanced 4K Result")
+            out_s = gr.Textbox(label="Status")
 
-    btn.click(process_video, [query_input, filter_input, start_input, end_input], [video_output, status_output])
+    btn.click(process_video, [q, o, s, e], [out_v, out_s])
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()