Update app.py

app.py

@@ -23,14 +23,20 @@ class StartPayload(BaseModel):
     anime_id: str
 
 def download_video(anime_id: str, episode: int, quality: str) -> str | None:
-    """Downloads video from Ayano API. Returns local path or None if episode doesn't exist."""
+    """Downloads video from Ayano API. Returns local path or None if episode/quality unavailable."""
     url = f"{HF_AYANO_BASE}/anime/download?id={anime_id}&episode={episode}&quality={quality}"
     resp = requests.get(url)
     if resp.status_code != 200:
         return None
     data = resp.json()
+
+    # Stop entirely if episode number exceeds available count
+    if data.get("status") == 422:
+        raise EpisodeExceedsAvailableCount(f"Episode {episode} exceeds available count")
+
+    # Skip if quality is missing
     if data.get("status") != 200:
-        return None  # Episode or quality unavailable
+        return None  
 
     video_url = data["direct_link"]
     tmp_path = tempfile.mktemp(suffix=".mp4")
@@ -68,27 +74,40 @@ def notify_render(anime_id: str, episode: int, quality: str, url: str, filename:
     except Exception as e:
         print("Failed to notify Render:", e)
 
-def process_anime(anime_id: str):
-    """Background task: iterates episodes and qualities, uploads, notifies."""
+def process_anime(anime_id: str, anime_name: str):
     episode = 1
     while True:
-        episode_completed = False
+        episode_processed = False
         for quality in QUALITIES:
             try:
-                local_file = download_video(anime_id, episode, quality)
-                if not local_file:
+                # Download returns None if episode/quality unavailable
+                result = download_video(anime_id, episode, quality)
+                if result is None:
+                    # If the API explicitly says episode exceeds count
                     continue
 
-                new_name = rename_file(local_file, anime_id, episode, quality)
-                catbox_url = upload_to_catbox_public(local_file, new_name)
-                notify_render(anime_id, episode, quality, catbox_url, new_name, code=2)  # success
+                local_file = result
+
+                # Blur + watermark
+                blurred_file = blur_video(local_file)
                 os.remove(local_file)
-                episode_completed = True
+
+                new_name = rename_file(blurred_file, anime_name, episode, quality)
+                catbox_url = upload_to_catbox_public(blurred_file, new_name)
+                os.remove(blurred_file)
+
+                notify_render(anime_id, episode, quality, catbox_url, new_name, code=2)
+                episode_processed = True
+
+            except EpisodeExceedsAvailableCount:
+                # Stop the outer loop entirely if the episode doesn't exist
+                return
             except Exception as e:
                 print(f"Error ep {episode} quality {quality}: {e}")
-                notify_render(anime_id, episode, quality, "", "", code=5)  # incomplete
+                notify_render(anime_id, episode, quality, "", "", code=5)
 
-        if not episode_completed:
+        if not episode_processed:
+            # No qualities were processed → stop
             break
         episode += 1