import os
import subprocess
import json

SESSION_FILE = r"c:\Projects\Historical\runpod_session.json"
SSH_KEY = os.path.expanduser(r"~/.ssh/id_ed25519")

with open(SESSION_FILE, "r") as f:
    session = json.load(f)

HOST = session["host"]
PORT = str(session["port"])

ssh_cmd = [
    "ssh", "-o", "StrictHostKeyChecking=no", "-i", SSH_KEY, "-p", PORT, f"root@{HOST}",
    "python3 -c 'import gdown; gdown.download(\"https://drive.google.com/uc?id=1iVnVxnFHeXQTfNRDAmnD_HSuPLVZhhNm\", \"/workspace/inputs/ep3.mp4\", quiet=False)'"
]
p = subprocess.run(ssh_cmd, capture_output=True, text=True, encoding="utf-8")
print(p.stdout)
if p.stderr:
    print("STDERR:", p.stderr)