Update app.py

app.py

@@ -1,50 +1,36 @@
-import gradio as gr
-import requests
-import re
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-
-# Gradio + FastAPI backend
-app = FastAPI(title="Catbox Raw File Extractor API")
-
-# Allow all origins (for frontend)
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-def get_direct_link(catbox_url: str):
-    """Extract direct raw file from a Catbox/Litterbox URL"""
-    if not catbox_url.startswith("http"):
-        return {"ok": False, "error": "Invalid URL"}
-
-    # If it already ends with a raw file extension, return as-is
-    if re.search(r'\.(mp4|png|jpg|jpeg|gif|webm)$', catbox_url, re.IGNORECASE):
-        return {"ok": True, "direct_link": catbox_url}
 
-    # Otherwise, try to fetch page and extract
-    try:
-        r = requests.get(catbox_url, timeout=10)
-        r.raise_for_status()
-        text = r.text
-
-        match = re.search(r'src="([^"]+\.(mp4|png|jpg|jpeg|gif|webm))"', text, re.IGNORECASE)
-        if match:
-            return {"ok": True, "direct_link": match.group(1)}
-        else:
-            return {"ok": False, "error": "Could not find raw file link"}
-    except Exception as e:
-        return {"ok": False, "error": str(e)}
-
-# Gradio API interface
-with gr.Blocks(server_name="0.0.0.0") as demo:
-    gr.Markdown("## Catbox / Litterbox Raw File Extractor API")
-    input_url = gr.Textbox(label="Paste Catbox/Litterbox URL")
-    output_json = gr.JSON(label="API Response")
-    extract_btn = gr.Button("Get Direct Link")
-    extract_btn.click(fn=get_direct_link, inputs=input_url, outputs=output_json)
-
-# Mount Gradio app onto FastAPI
-app = gr.mount_gradio_app(app, demo, path="/api")
+# app.py
+import gradio as gr
+import yt_dlp
+import os
+
+def download_anime(url, ep_start=1, ep_end=None):
+    ydl_opts = {
+        'format': 'best',
+        'outtmpl': '%(title)s - Episode %(episode_num)02d.%(ext)s',
+        'quiet': False,
+        'no_warnings': False,
+    }
+    
+    if ep_end:
+        ydl_opts['playliststart'] = ep_start
+        ydl_opts['playlistend'] = ep_end
+    
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        try:
+            info = ydl.extract_info(url, download=True)
+            title = info.get('title', 'Unknown')
+            return f"Downloaded: {title}"
+        except Exception as e:
+            return f"Error: {str(e)}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Anime Downloader (yt-dlp)")
+    url = gr.Textbox(label="Anime Page URL (e.g., animewave episode or series)")
+    start = gr.Number(label="Start Episode", value=1, precision=0)
+    end = gr.Number(label="End Episode (optional)", value=None)
+    btn = gr.Button("Download")
+    output = gr.Textbox(label="Status")
+    btn.click(download_anime, inputs=[url, start, end], outputs=output)
+
+demo.launch()