Update app.py

app.py

@@ -1,35 +1,51 @@
 import gradio as gr
+import tempfile
+from pathlib import Path
 import requests
 
-def upload_to_litterbox(video_file):
-    """
-    Uploads a video file to Litterbox and returns the direct URL.
-    """
-    try:
-        with open(video_file.name, "rb") as f:
-            files = {"fileToUpload": (video_file.name, f, "video/mp4")}
-            data = {"reqtype": "fileupload"}
-
-            resp = requests.post(
-                "https://litter.catbox.moe/resources/internals/api.php",
-                files=files,
-                data=data
-            )
-
-        if resp.status_code == 200 and resp.text.startswith("https://"):
-            return resp.text
-        else:
-            return f"Upload failed: {resp.status_code} - {resp.text}"
-
-    except Exception as e:
-        return f"Error: {e}"
+class LitterboxClient:
 
-with gr.Blocks() as demo:
-    gr.Markdown("## Upload Video to Litterbox / Catbox")
-    video_input = gr.File(label="Upload MP4", file_types=['.mp4'])
-    output_url = gr.Textbox(label="Litterbox URL")
-    upload_btn = gr.Button("Upload")
+    LITTERBOX_API = "https://litterbox.catbox.moe/resources/internals/api.php"
+    VALID_TIMES = ["1h", "12h", "24h", "72h"]
+
+    def __init__(self):
+        pass
+
+    def upload_file(self, filepath: str, expire_time: str = "24h") -> str:
+        if expire_time not in self.VALID_TIMES:
+            raise ValueError(f"Invalid expire_time '{expire_time}'. Must be one of {self.VALID_TIMES}")
+        
+        with open(filepath, "rb") as f:
+            files = {"fileToUpload": f}
+            data = {"reqtype": "fileupload", "time": expire_time}
+
+            response = requests.post(self.LITTERBOX_API, files=files, data=data)
+            response.raise_for_status()  # This is where 405 will show
+            if response.text.startswith("https://"):
+                return response.text.strip()
+            else:
+                raise RuntimeError(f"Unexpected response from Litterbox: {response.text}")
+
+client = LitterboxClient()
+
+def gradio_upload(video_file):
+    if not video_file:
+        return "No file provided"
+    
+    path = Path(video_file.name)
+    # Save the uploaded file temporarily
+    temp_path = tempfile.mktemp(suffix=path.suffix)
+    with open(temp_path, "wb") as f:
+        f.write(video_file.read())
     
-    upload_btn.click(upload_to_litterbox, inputs=video_input, outputs=output_url)
+    # Upload using your exact class
+    return client.upload_file(temp_path)
+
+with gr.Blocks() as demo:
+    gr.Markdown("## Test Litterbox Upload (Pure Script)")
+    video_input = gr.File(label="Upload Video")
+    output = gr.Textbox(label="Uploaded URL / Error")
+    run_btn = gr.Button("Upload")
+    run_btn.click(gradio_upload, inputs=video_input, outputs=output)
 
 demo.launch()