Update app.py

app.py

@@ -51,18 +51,33 @@ def safe_head(url: str, timeout: int = 6):
     except Exception:
         return None
 
-def fetch_bytes(src: str, timeout: int = 60) -> bytes:
+def fetch_bytes(src: str, stream_threshold: int = STREAM_THRESHOLD, timeout: int = 60, progress=None) -> bytes:
+    if progress: progress(0.05, desc="Checking remote/local source...")
     if is_remote(src):
         head = safe_head(src)
         if head is not None:
-            r = requests.get(src, timeout=timeout)
-            r.raise_for_status()
+            cl = head.headers.get("content-length")
+            if cl and int(cl) > stream_threshold:
+                if progress: progress(0.1, desc="Streaming large remote file...")
+                with requests.get(src, timeout=timeout, stream=True) as r:
+                    r.raise_for_status()
+                    fd, p = tempfile.mkstemp()
+                    os.close(fd)
+                    with open(p, "wb") as fh:
+                        for chunk in r.iter_content(8192):
+                            if chunk: fh.write(chunk)
+                with open(p, "rb") as fh: return fh.read()
+            r = safe_get(src, timeout=timeout)
+            if progress: progress(0.25, desc="Downloaded remote content")
             return r.content
     else:
         if not os.path.exists(src):
-            raise FileNotFoundError(f"Path does not exist: {src}")
+            raise FileNotFoundError(f"Local path does not exist: {src}")
+        if progress: progress(0.05, desc="Reading local file...")
         with open(src, "rb") as f:
-            return f.read()
+            data = f.read()
+        if progress: progress(0.15, desc="Read local file")
+        return data
 
 def save_bytes_to_temp(b: bytes, suffix: str) -> str:
     fd, path = tempfile.mkstemp(suffix=suffix)