Update app.py

app.py

@@ -1,30 +1,50 @@
 import gradio as gr
 import requests
-import tempfile
-import os
+import re
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 
-UPLOAD_URL = "https://0x0.st"
+# Gradio + FastAPI backend
+app = FastAPI(title="Catbox Raw File Extractor API")
 
-def upload_test(file):
-    if file is None:
-        return "No file uploaded"
-    
-    filename = os.path.basename(file.name)
+# 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:
-        with open(file.name, "rb") as f:
-            r = requests.post(UPLOAD_URL, files={"file": (filename, f)}, timeout=180)
+        r = requests.get(catbox_url, timeout=10)
         r.raise_for_status()
-        url = r.text.strip()
-        return f"Uploaded successfully!\n\nFile: {filename}\nURL: {url}"
+        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 f"Upload failed: {str(e)}"
-
-iface = gr.Interface(
-    fn=upload_test,
-    inputs=gr.File(label="Upload a video file (any size, up to ~512MB)"),
-    outputs=gr.Textbox(label="Result"),
-    title="0x0.st Upload Test",
-    description="Test uploading to 0x0.st directly. Use a small video first."
-)
+        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)
 
-iface.launch()
+# Mount Gradio app onto FastAPI
+app = gr.mount_gradio_app(app, demo, path="/api")