Add permanent Bear video example with direct URL from Hugging Face

app.py

@@ -34,26 +34,8 @@ load_dotenv()
 # Get default API key from environment (will be '' if not set)
 DEFAULT_API_KEY = os.getenv("OPENAI_API_KEY", "")
 
-def download_video_from_url(url):
-    try:
-        # Create a temporary file
-        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-        temp_filename = temp_file.name
-        temp_file.close()
-        
-        # Download the video
-        response = requests.get(url, stream=True)
-        if response.status_code == 200:
-            with open(temp_filename, 'wb') as f:
-                for chunk in response.iter_content(chunk_size=8192):
-                    f.write(chunk)
-            return temp_filename
-        else:
-            print(f"Failed to download video: {response.status_code}")
-            return None
-    except Exception as e:
-        print(f"Error downloading video: {e}")
-        return None
+# Embed the Bear.mp4 video directly from a guaranteed source
+BEAR_VIDEO_URL = "https://huggingface.co/spaces/gradio/video_component/resolve/main/files/Bear.mp4"
 
 def process_frame(frame_path, style_prompt, api_key):
     """Process a single frame with GPT-4o analysis and DALL-E 3 generation"""
@@ -136,7 +118,13 @@ def stylize_video(video_path, style_prompt, api_key):
         
         # Save the input video to a temporary file
         if isinstance(video_path, str):
-            if os.path.exists(video_path):
+            if video_path.startswith('http'):
+                # It's a URL, download it
+                response = requests.get(video_path, stream=True)
+                with open(input_filename, 'wb') as f:
+                    for chunk in response.iter_content(chunk_size=8192):
+                        f.write(chunk)
+            elif os.path.exists(video_path):
                 # It's a file path, copy it
                 shutil.copy(video_path, input_filename)
             else:
@@ -219,28 +207,22 @@ def stylize_video(video_path, style_prompt, api_key):
         print(f"Error: {str(e)}\n{traceback_str}")
         return None, f"Error: {str(e)}"
 
-def use_sample_bear_video():
-    """Function to use a sample bear video"""
-    # Try using a shorter, more reliable sample video URL
-    bear_url = "https://commondatastorage.googleapis.com/gtv-videos-bucket/sample/ForBiggerBlazes.mp4"
-    
-    # Download the video to a temporary file
-    temp_video_path = download_video_from_url(bear_url)
-    if temp_video_path:
-        return temp_video_path, "Studio Ghibli animation with Hayao Miyazaki's distinctive hand-drawn art style"
-    else:
-        # Return a more descriptive error message
-        print("Failed to download sample video. Using fallback.")
-        # If download fails, try a direct URL
-        return bear_url, "Studio Ghibli animation with Hayao Miyazaki's distinctive hand-drawn art style"
+def use_bear_video():
+    """Loads the Bear.mp4 sample video"""
+    return BEAR_VIDEO_URL, "Studio Ghibli animation with Hayao Miyazaki's distinctive hand-drawn art style"
 
 with gr.Blocks(title="Video-to-Ghibli Style Converter") as iface:
     gr.Markdown("# Video-to-Ghibli Style Converter")
     gr.Markdown("Upload a video and convert it to Studio Ghibli animation style using GPT-4o and DALL-E 3.")
     
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=2):
+            # Main input column
             video_input = gr.Video(label="Upload Video (up to 15 seconds)")
+            with gr.Row():
+                # Add example button in its own row
+                example_btn = gr.Button("Use Bear Video Example", variant="secondary")
+            
             api_key = gr.Textbox(
                 label="OpenAI API Key (requires GPT-4o and DALL-E 3 access)", 
                 type="password",
@@ -251,13 +233,12 @@ with gr.Blocks(title="Video-to-Ghibli Style Converter") as iface:
                 value="Studio Ghibli animation with Hayao Miyazaki's distinctive hand-drawn art style"
             )
             
-            with gr.Row():
-                submit_btn = gr.Button("Stylize Video", variant="primary")
-                example_btn = gr.Button("Use Sample Bear Video")
+            submit_btn = gr.Button("Stylize Video", variant="primary")
         
-        with gr.Column():
+        with gr.Column(scale=2):
+            # Output column
             video_output = gr.Video(label="Stylized Video")
-            status_output = gr.Textbox(label="Status", value="Ready. Upload a video or use the sample video.")
+            status_output = gr.Textbox(label="Status", value="Ready. Upload a video or use the Bear video example.")
     
     submit_btn.click(
         fn=stylize_video,
@@ -265,19 +246,19 @@ with gr.Blocks(title="Video-to-Ghibli Style Converter") as iface:
         outputs=[video_output, status_output]
     )
     
-    def load_example_and_update_status():
-        video, prompt = use_sample_bear_video()
-        return video, prompt, "Sample video loaded. Enter your OpenAI API key and click 'Stylize Video'."
+    def load_bear_and_update_status():
+        video, prompt = use_bear_video()
+        return video, prompt, "Bear video loaded. Enter your OpenAI API key and click 'Stylize Video'."
     
     example_btn.click(
-        fn=load_example_and_update_status,
+        fn=load_bear_and_update_status,
         inputs=None,
         outputs=[video_input, style_prompt, status_output]
     )
     
     gr.Markdown("""
     ## Instructions
-    1. Upload a video up to 15 seconds long or use the sample bear video
+    1. Upload a video up to 15 seconds long or use the Bear video example
     2. Enter your OpenAI API key with GPT-4o and DALL-E 3 access
     3. Customize the style prompt if desired
     4. Click "Stylize Video" and wait for processing