Update app.py

app.py

@@ -37,7 +37,6 @@ def is_http_url(s: str) -> bool:
     return s.startswith("http://") or s.startswith("https://")
 
 def ensure_parent_dir(path: str) -> None:
-    """Create parent directory for a file path if missing."""
     parent = os.path.dirname(path)
     if parent:
         os.makedirs(parent, exist_ok=True)
@@ -86,7 +85,6 @@ def fetch_bytes(url: str, timeout: int = 45) -> bytes:
     return r.content
 
 def parse_torrent(raw: bytes) -> Dict[str, Any]:
-    # Use top-level decode; current bencodepy has no Bencode class
     data = bencodepy.decode(raw)
     if not isinstance(data, dict) or b"info" not in data:
         raise ValueError("Invalid .torrent: missing 'info' dictionary.")
@@ -278,10 +276,8 @@ def _sha256_file(path: pathlib.Path, bufsize: int = 1024 * 1024) -> str:
     return h.hexdigest()
 
 def _supports_range(url: str, timeout: int = 30) -> Tuple[bool, Optional[int]]:
-    # HEAD to learn size and Accept-Ranges
     r = requests.head(url, timeout=timeout, allow_redirects=True)
     if r.status_code >= 400:
-        # Some servers don't allow HEAD; try GET without download
         r = requests.get(url, stream=True, timeout=timeout, allow_redirects=True)
         r.raise_for_status()
         size = int(r.headers.get("Content-Length", "0") or 0)
@@ -296,10 +292,6 @@ def _supports_range(url: str, timeout: int = 30) -> Tuple[bool, Optional[int]]:
     return ("bytes" in accept_ranges.lower() or size > 0, size if size > 0 else None)
 
 def _download_with_resume(url: str, dest_path: pathlib.Path, timeout: int = 120) -> Tuple[int, Optional[int]]:
-    """
-    Download URL to dest_path with simple resume (HTTP Range).
-    Returns (bytes_written, total_expected or None).
-    """
     dest_path.parent.mkdir(parents=True, exist_ok=True)
     tmp_path = dest_path.with_suffix(dest_path.suffix + ".part")
 
@@ -320,22 +312,12 @@ def _download_with_resume(url: str, dest_path: pathlib.Path, timeout: int = 120)
                     f.write(chunk)
                     bytes_written += len(chunk)
 
-    # finalize
     final_size = (existing + bytes_written)
-    # If we know total_size and match or server returned 200 with complete payload, rename
     if total_size is None or final_size >= (total_size or 0):
         tmp_path.rename(dest_path)
-    else:
-        # Keep .part if incomplete
-        pass
     return bytes_written, total_size
 
 def prepare_download(parsed_json: str, base_url_override: str) -> Tuple[str, List[str], List[str], str]:
-    """
-    Build seed list and file choices.
-    - If torrent has web seeds → use them.
-    - Else if base_url_override provided → use that as a 'seed'.
-    """
     if not parsed_json:
         return "Parse something in Inspect first.", [], [], ""
     parsed = json.loads(parsed_json)
@@ -353,10 +335,6 @@ def prepare_download(parsed_json: str, base_url_override: str) -> Tuple[str, Lis
     return msg, seeds, file_list, root
 
 def download_selected(parsed_json: str, seed_url: str, root_dir: str, selected_files: List[str]) -> Tuple[str, List[str]]:
-    """
-    Download selected files into /mnt/data/downloads/<infohash>/...
-    Create .sha256 sidecar after successful completion.
-    """
     if not parsed_json:
         raise gr.Error("Parse a torrent first.")
     if not seed_url:
@@ -378,10 +356,8 @@ def download_selected(parsed_json: str, seed_url: str, root_dir: str, selected_f
             url = _join_url(seed_url, root_dir, rel)
             dest_path = out_root / rel
             bytes_written, total_expected = _download_with_resume(url, dest_path)
-            # Verify size if server told us
             if total_expected is not None and dest_path.exists() and dest_path.stat().st_size != total_expected:
                 logs.append(f"⚠️ Size mismatch for {rel} (got {dest_path.stat().st_size}, expected {total_expected}). Kept .part if incomplete.")
-            # SHA256
             if dest_path.exists():
                 sha = _sha256_file(dest_path)
                 with open(str(dest_path) + ".sha256", "w") as f:
@@ -526,4 +502,8 @@ with gr.Blocks(css=CSS, title="Torrent Inspector + Full HTTP Downloader") as dem
     )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=int(os.environ.get("PORT", 7860)),
+        allowed_paths=["/mnt/data"]  # <-- allow returning files from /mnt/data
+    )