Update app.py

app.py

@@ -1,57 +1,28 @@
 import gradio as gr
 import subprocess
 import os
+import uuid
 
-# --- Force Install Node Modules ---
 def initialize_env():
-    print("Checking Node.js environment...")
-    
-    # বর্তমান ডিরেক্টরি চেক করা
-    current_dir = os.getcwd()
-    print(f"Current Directory: {current_dir}")
-    
-    # package.json আছে কিনা চেক করা
-    if not os.path.exists("package.json"):
-        print("❌ ERROR: package.json not found! Creating one...")
-        with open("package.json", "w") as f:
-            f.write('{"dependencies": {"puppeteer-core": "^21.0.0"}}')
-
-    # node_modules না থাকলে install করা
     if not os.path.exists("node_modules"):
-        print("Installing node_modules... this may take a minute.")
-        try:
-            # --no-audit এবং --no-fund দিলে ইন্সটলেশন দ্রুত হয়
-            subprocess.run(["npm", "install", "--no-audit", "--no-fund"], check=True)
-            print("✅ node_modules installed successfully.")
-        except Exception as e:
-            print(f"❌ NPM Install Failed: {e}")
+        print("Installing node_modules...")
+        subprocess.run(["npm", "install"], check=True)
 
-# অ্যাপ শুরু হওয়ার আগেই এনভায়রনমেন্ট রেডি করা
 initialize_env()
 
-def search_pinterest(query):
+def process_video(query, start_time, end_time):
     if not query:
         return None, "⚠️ Please enter a keyword."
     
+    # 1. Scrape the URL using Node.js
     search_query = query if "video" in query.lower() else f"{query} video"
-    
     try:
-        # Scraper চালানো
         result = subprocess.run(
             ["node", "scraper.js", search_query],
-            capture_output=True,
-            text=True,
-            timeout=90
+            capture_output=True, text=True, timeout=90
         )
         
         stdout = result.stdout.strip()
-        stderr = result.stderr.strip()
-        
-        # ডিবাগিং এর জন্য লগ দেখা
-        if stderr:
-            print(f"Node Debug: {stderr}")
-
-        # URL খুঁজে বের করা
         lines = stdout.split('\n')
         video_url = None
         for line in lines:
@@ -59,27 +30,68 @@ def search_pinterest(query):
                 video_url = line.strip()
                 break
         
-        if video_url:
-            return video_url, f"✅ Found video for: {query}"
-        else:
-            return None, f"❌ No video URL found. Logs: {stderr[:100]}"
+        if not video_url:
+            return None, f"❌ No video found. Logs: {result.stderr[:100]}"
 
+        # 2. Trim the video using FFmpeg
+        # We create a unique filename for the trimmed video
+        output_filename = f"trimmed_{uuid.uuid4().hex[:8]}.mp4"
+        
+        print(f"Trimming video: {start_time}s to {end_time}s")
+        
+        # FFmpeg Command Explanation:
+        # -ss: Start time
+        # -to: End time
+        # -i: Input URL (FFmpeg can stream directly from the URL)
+        # -c:v libx264: Re-encode to ensure precision for decimals like 3.79
+        
+        ffmpeg_cmd = [
+            "ffmpeg",
+            "-ss", str(start_time),
+            "-to", str(end_time),
+            "-i", video_url,
+            "-c:v", "libx264",
+            "-c:a", "aac",
+            "-strict", "experimental",
+            "-y", # Overwrite if exists
+            output_filename
+        ]
+        
+        # If end_time is 0, we assume the user wants the whole video from the start_time
+        if end_time <= 0:
+            ffmpeg_cmd.pop(3) # Remove "-to"
+            ffmpeg_cmd.pop(3) # Remove the end_time value
+
+        subprocess.run(ffmpeg_cmd, check=True)
+
+        return output_filename, f"✅ Success! Trimmed from {start_time}s to {end_time}s"
+
+    except subprocess.CalledProcessError as e:
+        return None, f"❌ FFmpeg Error: Maybe the start/end time is longer than the video?"
     except Exception as e:
         return None, f"❌ System Error: {str(e)}"
 
-# Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# 📌 Pinterest Video Downloader")
+# --- Gradio UI ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 📌 Pinterest Video Search & Trim")
     
     with gr.Row():
-        inp = gr.Textbox(label="Query", placeholder="e.g. Funny cat")
-        btn = gr.Button("Search", variant="primary")
-    
-    out_video = gr.Video(label="Result")
-    out_log = gr.Textbox(label="Status")
+        with gr.Column():
+            query_input = gr.Textbox(label="Search Query", placeholder="e.g. Minecraft Gameplay")
+            with gr.Row():
+                start_input = gr.Number(label="Start Second (e.g. 1.5)", value=0)
+                end_input = gr.Number(label="End Second (0 = Full video)", value=0)
+            btn = gr.Button("🚀 Search & Trim", variant="primary")
+        
+        with gr.Column():
+            video_output = gr.Video(label="Trimmed Result")
+            status_output = gr.Textbox(label="Status")
 
-    btn.click(search_pinterest, inputs=inp, outputs=[out_video, out_log])
-    inp.submit(search_pinterest, inputs=inp, outputs=[out_video, out_log])
+    btn.click(
+        fn=process_video, 
+        inputs=[query_input, start_input, end_input], 
+        outputs=[video_output, status_output]
+    )
 
 if __name__ == "__main__":
     demo.launch()