Update app.py

app.py

@@ -5,7 +5,7 @@ import tempfile
 import base64
 import json
 from io import BytesIO
-from typing import List, Tuple, Optional, Set # Import Set
+from typing import List, Tuple, Optional, Set
 import requests
 from PIL import Image, ImageFile, UnidentifiedImageError
 import gradio as gr
@@ -176,19 +176,21 @@ def fetch_bytes(src: str, stream_threshold: int = STREAM_THRESHOLD_BYTES, timeou
                     fd, p = tempfile.mkstemp(suffix=ext_from_src(src) or ".tmp")
                     os.close(fd)
                     try:
-                        with requests.get(src, timeout=timeout, stream=True, headers=DEFAULT_HEADERS) as r:
-                            r.raise_for_status()
-                            total_size = int(r.headers.get("content-length", 0))
-                            downloaded_size = 0
-                            for chunk in r.iter_content(8192):
-                                if chunk:
-                                    fh.write(chunk)
-                                    downloaded_size += len(chunk)
-                                    if progress is not None and total_size > 0:
-                                        # Scale progress from 0.1 to 0.25 for streaming phase
-                                        progress(0.1 + (downloaded_size / total_size) * 0.15)
-                        with open(p, "rb") as fh:
-                            return fh.read()
+                        # FIX: Open file for writing before the streaming loop
+                        with open(p, "wb") as fh_write:
+                            with requests.get(src, timeout=timeout, stream=True, headers=DEFAULT_HEADERS) as r:
+                                r.raise_for_status()
+                                total_size = int(r.headers.get("content-length", 0))
+                                downloaded_size = 0
+                                for chunk in r.iter_content(8192):
+                                    if chunk:
+                                        fh_write.write(chunk)
+                                        downloaded_size += len(chunk)
+                                        if progress is not None and total_size > 0:
+                                            # Scale progress from 0.1 to 0.25 for streaming phase
+                                            progress(0.1 + (downloaded_size / total_size) * 0.15)
+                        with open(p, "rb") as fh_read: # Open again to read full content
+                            return fh_read.read()
                     finally:
                         try: os.remove(p)
                         except Exception as e: print(f"Error during streaming temp file cleanup {p}: {e}")
@@ -243,13 +245,19 @@ def _ffprobe_streams(path: str) -> Optional[dict]:
     """Uses ffprobe to get stream information for a media file."""
     if not FFMPEG_BIN:
         return None
-    # Use FFMPEG_BIN for ffprobe if it's explicitly provided and ends with ffmpeg
-    ffprobe_path = FFMPEG_BIN.replace("ffmpeg", "ffprobe") if "ffmpeg" in os.path.basename(FFMPEG_BIN) else "ffprobe"
-    if not shutil.which(ffprobe_path):
-        # Fallback to just 'ffprobe' if the path replacement didn't find it
-        ffprobe_path = "ffprobe"
-        if not shutil.which(ffprobe_path):
-            return None # ffprobe is not available
+
+    ffprobe_path = None
+    if FFMPEG_BIN:
+        ffmpeg_dir = os.path.dirname(FFMPEG_BIN)
+        potential_ffprobe_in_dir = os.path.join(ffmpeg_dir, "ffprobe")
+        if os.path.exists(potential_ffprobe_in_dir) and os.access(potential_ffprobe_in_dir, os.X_OK):
+            ffprobe_path = potential_ffprobe_in_dir
+
+    if not ffprobe_path: # Fallback to checking PATH if not found next to ffmpeg
+        ffprobe_path = shutil.which("ffprobe")
+
+    if not ffprobe_path:
+        return None # ffprobe is not available
 
     cmd = [
         ffprobe_path, "-v", "error", "-print_format", "json", "-show_streams", "-show_format", path