Update app.py

app.py

@@ -1,19 +1,33 @@
 import gradio as gr
 import subprocess
 import os
+import sys
+
+# --- NEW STARTUP BLOCK: Install Node Dependencies ---
+def install_node_deps():
+    if not os.path.exists("node_modules"):
+        print("--- Node.js dependencies not found. Installing... ---")
+        try:
+            # Run npm install
+            subprocess.run(["npm", "install"], check=True)
+            print("--- Node.js dependencies installed successfully! ---")
+        except Exception as e:
+            print(f"--- Error installing Node dependencies: {e} ---")
+
+# Run the installation
+install_node_deps()
+# ----------------------------------------------------
 
 def search_pinterest(query):
     if not query:
         return None, "⚠️ Please enter a search keyword."
     
-    # Auto-add 'video' to help Pinterest search results
     search_query = query if "video" in query.lower() else f"{query} video"
     
     print(f"--- Starting Scraper for: {search_query} ---")
     
     try:
         # Run the Node.js scraper
-        # We set a 90-second timeout because Puppeteer can be slow on cloud servers
         result = subprocess.run(
             ["node", "scraper.js", search_query],
             capture_output=True,
@@ -21,16 +35,10 @@ def search_pinterest(query):
             timeout=90
         )
         
-        # Capture all output from Node.js
         stdout_content = result.stdout.strip()
         stderr_content = result.stderr.strip()
         
-        # Log to Hugging Face console for debugging
-        if stderr_content:
-            print(f"Debug Logs:\n{stderr_content}")
-            
-        # Extract the URL from stdout
-        # We split by lines and look for the one starting with 'http'
+        # Look for the URL in stdout
         lines = stdout_content.split('\n')
         video_url = None
         for line in lines:
@@ -43,51 +51,31 @@ def search_pinterest(query):
             print(f"✅ Success! URL Found: {video_url}")
             return video_url, f"Found video for: {query}"
         else:
-            print("❌ No URL found in Node output.")
-            return None, f"Could not find a video. Output received: {stdout_content[:100]}..."
+            # Print stderr to the console so you can see if Puppeteer crashed
+            if stderr_content:
+                print(f"Scraper Error Log:\n{stderr_content}")
+            return None, f"Could not find a video. Error: {stderr_content[:200]}"
 
-    except subprocess.TimeoutExpired:
-        print("❌ Scraper timed out.")
-        return None, "The search took too long. Please try again with a simpler keyword."
     except Exception as e:
         print(f"❌ System Error: {str(e)}")
         return None, f"System Error: {str(e)}"
 
-# Define the Gradio UI Theme and Layout
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+# Define the Gradio UI
+with gr.Blocks() as demo:
     gr.Markdown("# 📌 Pinterest Video Downloader")
-    gr.Markdown("Enter a topic below. The system will search Pinterest and return the best quality video found.")
+    gr.Markdown("Searching Pinterest via Puppeteer (Node.js) on Hugging Face.")
     
     with gr.Row():
-        with gr.Column(scale=4):
-            input_text = gr.Textbox(
-                label="What are you looking for?", 
-                placeholder="e.g. Luxury Car, Funny Cat, Nature Aesthetic...",
-                lines=1
-            )
-        with gr.Column(scale=1):
-            submit_btn = gr.Button("🔍 Search Video", variant="primary")
-    
-    with gr.Row():
-        # The Video component handles the .mp4 URL directly
-        video_display = gr.Video(label="Result Video")
+        input_text = gr.Textbox(label="Query", placeholder="e.g. Cat funny video")
+        submit_btn = gr.Button("🔍 Search Video", variant="primary")
     
+    video_display = gr.Video(label="Result Video")
     status_msg = gr.Textbox(label="Status Log", interactive=False)
 
-    # Link the button to the function
-    submit_btn.click(
-        fn=search_pinterest, 
-        inputs=input_text, 
-        outputs=[video_display, status_msg]
-    )
-    
-    # Allow pressing "Enter" to search
-    input_text.submit(
-        fn=search_pinterest, 
-        inputs=input_text, 
-        outputs=[video_display, status_msg]
-    )
+    submit_btn.click(fn=search_pinterest, inputs=input_text, outputs=[video_display, status_msg])
+    input_text.submit(fn=search_pinterest, inputs=input_text, outputs=[video_display, status_msg])
 
 # Launch the app
 if __name__ == "__main__":
+    # Note: theme moved to launch() for Gradio 6.0 compatibility as per your logs
     demo.launch()