Create app.py

app.py

@@ -0,0 +1,141 @@
+import gradio as gr
+import subprocess
+import os
+
+# --- Configuration ---
+
+# Use persistent storage (/data/) if available.
+if os.path.exists("/data"):
+    SAVE_PATH = "/data/torrents"
+else:
+    SAVE_PATH = "./torrents"
+    print("WARNING: /data directory not found. Using ephemeral storage.")
+    print("Files will be DELETED when the Space restarts.")
+
+# Create the save directory
+os.makedirs(SAVE_PATH, exist_ok=True)
+
+print(f"File save location: {SAVE_PATH}")
+
+# Keep track of running processes (by magnet link)
+processes = {}
+
+def add_torrent(magnet_link):
+    """
+    Starts an aria2c download process in the background.
+    """
+    if not magnet_link or not magnet_link.startswith("magnet:?"):
+        return "Error: Invalid magnet link. Must start with 'magnet:?'"
+    
+    if magnet_link in processes:
+        # Check if the process is still running
+        poll = processes[magnet_link].poll()
+        if poll is None:
+            return f"Download for this link is already running."
+    
+    log_file = os.path.join(SAVE_PATH, "aria2.log")
+    
+    # Command to run aria2
+    # --seed-time=0: Stop seeding immediately after download
+    # --daemon=true: Run in the background
+    # --enable-rpc=false: We don't need the RPC server
+    command = [
+        "aria2c",
+        magnet_link,
+        f"--dir={SAVE_PATH}",
+        "--seed-time=0",
+        "--daemon=true",
+        "--enable-rpc=false",
+        f"--log={log_file}",
+        "--log-level=info",
+        "--follow-torrent=mem" # Don't save .torrent file
+    ]
+    
+    try:
+        # Start the process
+        # We use Popen to run it non-blockingly (in the background)
+        process = subprocess.Popen(command)
+        processes[magnet_link] = process
+        print(f"Started process {process.pid} for: {magnet_link}")
+        return f"Download started. Check status below."
+        
+    except Exception as e:
+        print(f"Error starting aria2: {e}")
+        return f"Error: {e}"
+
+def get_status_report():
+    """
+    Scans the download directory and lists files.
+    .aria2 files are temporary control files; their presence means 
+    a download is incomplete.
+    """
+    try:
+        files = os.listdir(SAVE_PATH)
+    except Exception as e:
+        return f"Could not read directory: {e}"
+
+    if not files:
+        return "No downloads yet. Add a magnet link to start."
+
+    incomplete = []
+    completed = []
+
+    for f in files:
+        if f.endswith(".aria2"):
+            # It's a control file, so the real file is "in progress"
+            incomplete.append(f.replace(".aria2", ""))
+        elif f == "aria2.log":
+            continue # Ignore the log file
+        else:
+            # Check if it has a matching .aria2 file
+            if f"{f}.aria2" not in files:
+                completed.append(f)
+    
+    # Build the report
+    report = ""
+    if incomplete:
+        report += "🔽 IN PROGRESS:\n" + "\n".join(f"  - {f}" for f in incomplete) + "\n\n"
+    
+    if completed:
+        report += "✅ COMPLETED:\n" + "\n".join(f"  - {f}" for f in completed)
+    
+    if not incomplete and not completed:
+        return "Directory is empty (or only has log file)."
+
+    return report
+
+# --- Build the Gradio App ---
+with gr.Blocks(title="Torrent Downloader") as demo:
+    gr.Markdown(
+        "# Hugging Face Torrent Downloader (aria2)\n"
+        f"Add a magnet link to download files. Files are saved in: `{SAVE_PATH}`"
+    )
+    
+    with gr.Row():
+        magnet_input = gr.Textbox(
+            label="Magnet Link", 
+            placeholder="magnet:?xt=urn:btih:...",
+            scale=4
+        )
+        add_button = gr.Button("Add Torrent", variant="primary", scale=1)
+
+    # This Textbox will automatically refresh every 5 seconds
+    status_output = gr.Textbox(
+        label="Download Status (refreshes every 5s)", 
+        lines=15, 
+        interactive=False,
+        value=get_status_report,
+        every=5 
+    )
+    
+    add_result_output = gr.Textbox(label="System Message", interactive=False)
+
+    add_button.click(
+        fn=add_torrent, 
+        inputs=magnet_input, 
+        outputs=add_result_output 
+    )
+    add_button.click(lambda: "", outputs=magnet_input) # Clear input on click
+
+# Launch the app
+demo.launch()