Update app.py

app.py

@@ -2,56 +2,92 @@ import gradio as gr
 import subprocess
 import os
 
-# Ensure npm dependencies are installed before starting (just in case)
-subprocess.run(["npm", "install"], check=True)
-
 def search_pinterest(query):
     if not query:
-        return None, "Please enter a search term."
+        return None, "⚠️ Please enter a search keyword."
     
-    if "video" not in query.lower():
-        query += " video"
-
-    print(f"Calling Node.js for: {query}")
+    # 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 script
-        # We capture stdout (standard output) to get the URL
+        # 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", query],
+            ["node", "scraper.js", search_query],
             capture_output=True,
             text=True,
-            timeout=60 # 60 second timeout
+            timeout=90
         )
         
-        output = result.stdout.strip()
-        error = result.stderr.strip()
+        # Capture all output from Node.js
+        stdout_content = result.stdout.strip()
+        stderr_content = result.stderr.strip()
         
-        print("Node Output:", output)
+        # 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'
+        lines = stdout_content.split('\n')
+        video_url = None
+        for line in lines:
+            clean_line = line.strip()
+            if clean_line.startswith("http"):
+                video_url = clean_line
+                break
         
-        if "http" in output:
-            return output, f"Found video for: {query}"
-        elif "ERROR" in output:
-            return None, output
+        if video_url:
+            print(f"✅ Success! URL Found: {video_url}")
+            return video_url, f"Found video for: {query}"
         else:
-            return None, "No video found or scraper failed. Try a different keyword."
+            print("❌ No URL found in Node output.")
+            return None, f"Could not find a video. Output received: {stdout_content[:100]}..."
 
+    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:
-        return None, str(e)
+        print(f"❌ System Error: {str(e)}")
+        return None, f"System Error: {str(e)}"
 
-# Build the Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# 📌 Pinterest Video Finder")
-    gr.Markdown("Type a keyword (e.g., 'Aesthetic Sunset') and the system will find a downloadable video.")
+# Define the Gradio UI Theme and Layout
+with gr.Blocks(theme=gr.themes.Soft()) 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.")
     
     with gr.Row():
-        txt_input = gr.Textbox(label="Search Query", placeholder="cat funny video")
-        btn_submit = gr.Button("Search Video")
+        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")
     
-    # Video Output
-    video_output = gr.Video(label="Result Video")
-    status_output = gr.Textbox(label="Status Log", interactive=False)
+    with gr.Row():
+        # The Video component handles the .mp4 URL directly
+        video_display = gr.Video(label="Result Video")
+    
+    status_msg = gr.Textbox(label="Status Log", interactive=False)
 
-    btn_submit.click(search_pinterest, inputs=txt_input, outputs=[video_output, status_output])
+    # 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]
+    )
 
-demo.launch()
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()