Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import re
|
|
| 10 |
|
| 11 |
app = FastAPI(title="Neon Anime Blur & Upload")
|
| 12 |
|
| 13 |
-
UPLOAD_URL = "https://
|
| 14 |
RENDER_UPDATE_ENDPOINT = "https://nt-anime-api.onrender.com/update"
|
| 15 |
HF_AYANO_BASE = "https://a-y-a-n-o-k-o-j-i-dnd-api.hf.space"
|
| 16 |
|
|
@@ -41,7 +41,6 @@ def download_video(anime_id: str, episode: int, quality: str) -> str | None:
|
|
| 41 |
if data.get("status") == 422:
|
| 42 |
raise EpisodeExceedsAvailableCount()
|
| 43 |
if data.get("status") != 200:
|
| 44 |
-
log(f"No link available ep {episode} {quality}")
|
| 45 |
return None
|
| 46 |
video_url = data["direct_link"]
|
| 47 |
log(f"Got direct link ep {episode} {quality}")
|
|
@@ -70,14 +69,16 @@ def get_filename(anime_name: str, ep: int, quality: str) -> str:
|
|
| 70 |
slug = re.sub(r'[^a-z0-9-]+', '-', anime_name.lower()).strip('-')
|
| 71 |
return f"nt-animes_{slug}_ep{ep}_{quality}.mp4"
|
| 72 |
|
| 73 |
-
def
|
| 74 |
log(f"Uploading {file_name}")
|
| 75 |
try:
|
| 76 |
with open(file_path, "rb") as f:
|
| 77 |
-
|
|
|
|
|
|
|
| 78 |
r.raise_for_status()
|
| 79 |
url = r.text.strip()
|
| 80 |
-
log(f"Uploaded
|
| 81 |
return url
|
| 82 |
except Exception as e:
|
| 83 |
log(f"Upload failed: {e}")
|
|
@@ -108,7 +109,7 @@ def blur_video(input_path: str) -> str:
|
|
| 108 |
cmd = [
|
| 109 |
"ffmpeg", "-y", "-i", input_path,
|
| 110 |
"-filter_complex", filter_graph,
|
| 111 |
-
"-c:v", "libx264", "-preset", "
|
| 112 |
"-c:a", "copy", output
|
| 113 |
]
|
| 114 |
log("Blurring video")
|
|
@@ -133,7 +134,7 @@ def process_anime(anime_id: str, anime_name: str):
|
|
| 133 |
blurred_file = blur_video(local_file)
|
| 134 |
os.remove(local_file)
|
| 135 |
file_name = get_filename(anime_name, episode, quality)
|
| 136 |
-
file_url =
|
| 137 |
os.remove(blurred_file)
|
| 138 |
notify_render(anime_id, episode, quality, file_url, file_name, 2)
|
| 139 |
processed = True
|
|
|
|
| 10 |
|
| 11 |
app = FastAPI(title="Neon Anime Blur & Upload")
|
| 12 |
|
| 13 |
+
UPLOAD_URL = "https://litterbox.catbox.moe/resources/internals/api.php"
|
| 14 |
RENDER_UPDATE_ENDPOINT = "https://nt-anime-api.onrender.com/update"
|
| 15 |
HF_AYANO_BASE = "https://a-y-a-n-o-k-o-j-i-dnd-api.hf.space"
|
| 16 |
|
|
|
|
| 41 |
if data.get("status") == 422:
|
| 42 |
raise EpisodeExceedsAvailableCount()
|
| 43 |
if data.get("status") != 200:
|
|
|
|
| 44 |
return None
|
| 45 |
video_url = data["direct_link"]
|
| 46 |
log(f"Got direct link ep {episode} {quality}")
|
|
|
|
| 69 |
slug = re.sub(r'[^a-z0-9-]+', '-', anime_name.lower()).strip('-')
|
| 70 |
return f"nt-animes_{slug}_ep{ep}_{quality}.mp4"
|
| 71 |
|
| 72 |
+
def upload_to_litterbox(file_path: str, file_name: str) -> str:
|
| 73 |
log(f"Uploading {file_name}")
|
| 74 |
try:
|
| 75 |
with open(file_path, "rb") as f:
|
| 76 |
+
files = {"fileToUpload": (file_name, f)}
|
| 77 |
+
data = {"reqtype": "fileupload", "time": "72h"}
|
| 78 |
+
r = requests.post(UPLOAD_URL, data=data, files=files, timeout=180)
|
| 79 |
r.raise_for_status()
|
| 80 |
url = r.text.strip()
|
| 81 |
+
log(f"Uploaded: {url}")
|
| 82 |
return url
|
| 83 |
except Exception as e:
|
| 84 |
log(f"Upload failed: {e}")
|
|
|
|
| 109 |
cmd = [
|
| 110 |
"ffmpeg", "-y", "-i", input_path,
|
| 111 |
"-filter_complex", filter_graph,
|
| 112 |
+
"-c:v", "libx264", "-preset", "fast", "-crf", "23",
|
| 113 |
"-c:a", "copy", output
|
| 114 |
]
|
| 115 |
log("Blurring video")
|
|
|
|
| 134 |
blurred_file = blur_video(local_file)
|
| 135 |
os.remove(local_file)
|
| 136 |
file_name = get_filename(anime_name, episode, quality)
|
| 137 |
+
file_url = upload_to_litterbox(blurred_file, file_name)
|
| 138 |
os.remove(blurred_file)
|
| 139 |
notify_render(anime_id, episode, quality, file_url, file_name, 2)
|
| 140 |
processed = True
|