Update app.py

app.py

@@ -1,56 +1,30 @@
 import gradio as gr
-import subprocess
-import tempfile
-import time
-import os
-
-def download_and_measure(stream_url):
-    output = tempfile.mktemp(suffix=".mp4")
-
-    start = time.time()
-
-    cmd = [
-        "ffmpeg",
-        "-y",
-        "-i", stream_url,
-        "-c", "copy",
-        "-bsf:a", "aac_adtstoasc",
-        output
-    ]
-
-    subprocess.run(
-        cmd,
-        stdout=subprocess.DEVNULL,
-        stderr=subprocess.DEVNULL,
-        check=True
-    )
-
-    elapsed = round(time.time() - start, 2)
-    size_mb = round(os.path.getsize(output) / 1024 / 1024, 2)
-
-    return {
-        "download_time_seconds": elapsed,
-        "file_size_mb": size_mb,
-        "output_file_path": output
+import requests
+
+def upload_to_litterbox(video_file):
+    """
+    Uploads a video file to Litterbox and returns the direct URL.
+    """
+    files = {
+        'fileToUpload': (video_file.name, open(video_file.name, 'rb'), '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:
+        return resp.text  # This is the direct .mp4 link
+    else:
+        return f"Upload failed with status {resp.status_code}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("## HLS Stream Download Speed Test")
-    gr.Markdown("Fastest possible method — stream copy only")
-
-    url_input = gr.Textbox(
-        label="HLS Stream URL",
-        placeholder="https://example.com/master.m3u8"
-    )
-
-    result = gr.JSON(label="Result")
-
-    run_btn = gr.Button("Download & Measure")
-
-    run_btn.click(
-        download_and_measure,
-        inputs=url_input,
-        outputs=result
-    )
+    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")
+    
+    upload_btn.click(upload_to_litterbox, inputs=video_input, outputs=output_url)
 
 demo.launch()