Update app.py

app.py

@@ -1,6 +1,7 @@
 import requests
 import tempfile
 import os
+import shutil
 import subprocess
 from fastapi import FastAPI, BackgroundTasks
 from pydantic import BaseModel
@@ -16,8 +17,8 @@ HF_AYANO_BASE = "https://a-y-a-n-o-k-o-j-i-dnd-api.hf.space"
 QUALITIES = ["360p", "720p", "1080p"]
 queue_lock = Lock()
 
-def log_error(msg: str):
-    print(f"[ERROR] {msg}", flush=True)
+def log(msg: str):
+    print(f"[HF] {msg}", flush=True)
 
 class EpisodeExceedsAvailableCount(Exception):
     pass
@@ -43,8 +44,7 @@ def download_video(anime_id: str, episode: int, quality: str) -> str | None:
         video_url = data["direct_link"]
     except EpisodeExceedsAvailableCount:
         raise
-    except Exception as e:
-        log_error(f"Link failed ep {episode} {quality}: {e}")
+    except:
         return None
 
     tmp_path = tempfile.mktemp(suffix=".mp4")
@@ -54,8 +54,7 @@ def download_video(anime_id: str, episode: int, quality: str) -> str | None:
             with open(tmp_path, "wb") as f:
                 shutil.copyfileobj(r.raw, f)
         return tmp_path
-    except Exception as e:
-        log_error(f"Download failed ep {episode} {quality}: {e}")
+    except:
         if os.path.exists(tmp_path):
             os.remove(tmp_path)
         return None
@@ -71,7 +70,7 @@ def upload_to_0x0(file_path: str, file_name: str) -> str:
         r.raise_for_status()
         return r.text.strip()
     except Exception as e:
-        log_error(f"Upload failed: {e}")
+        log(f"[ERROR] Upload failed: {e}")
         raise
 
 def notify_render(anime_id: str, episode: int, quality: str, file_url: str, file_name: str, status: int):
@@ -84,9 +83,9 @@ def notify_render(anime_id: str, episode: int, quality: str, file_url: str, file
         "status": status
     }
     try:
-        requests.post(RENDER_UPDATE_ENDPOINT, json=payload, timeout=10)
+        requests.post(RENDER_UPDATE_ENDPOINT, json=payload, timeout=20)
     except Exception as e:
-        log_error(f"Notify failed: {e}")
+        log(f"[ERROR] Notify failed: {e}")
 
 def blur_video(input_path: str) -> str:
     output = tempfile.mktemp(suffix="_blurred.mp4")
@@ -101,14 +100,11 @@ def blur_video(input_path: str) -> str:
         "-c:v", "libx264", "-preset", "fast", "-crf", "23",
         "-c:a", "copy", output
     ]
-    try:
-        subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-        return output
-    except Exception as e:
-        log_error(f"Blur failed: {e}")
-        raise
+    subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    return output
 
 def process_anime(anime_id: str, anime_name: str):
+    log(f"Started {anime_id} - {anime_name}")
     episode = 1
     while True:
         processed = False
@@ -127,15 +123,17 @@ def process_anime(anime_id: str, anime_name: str):
             except EpisodeExceedsAvailableCount:
                 notify_render(anime_id, 0, "", "", "", 5)
                 return
-            except Exception:
+            except:
                 notify_render(anime_id, episode, quality, "", "", 3)
         if not processed:
             notify_render(anime_id, 0, "", "", "", 5)
             break
         episode += 1
+    log("Complete")
 
 @app.post("/start")
-def start_endpoint(payload: StartPayload, background_tasks: BackgroundTasks):
+def start(payload: StartPayload, bg: BackgroundTasks):
+    log(f"Queueing {payload.anime_id} - {payload.anime_name}")
     with queue_lock:
-        background_tasks.add_task(process_anime, payload.anime_id, payload.anime_name)
-    return {"code": 4, "message": "Job queued and processing in background"}
+        bg.add_task(process_anime, payload.anime_id, payload.anime_name)
+    return {"code": 4, "message": "Queued"}