Hugging Face's logo

Spaces:
MLBench
/
AXION_Deployment_Test
Running

App Files Community
AXION_Deployment_Test
File size: 3,470 Bytes 
28be5ed
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
 
import gradio as gr
import requests
import re
import time
from datetime import datetime, timezone

# Constants
RUNPOD_API_KEY = "rpa_0PN9A3W28RFTK3LDY6M3LRO5HT9SWI441HQV2V6A1jt5ih"
RUNPOD_ENDPOINT = "https://api.runpod.ai/v2/vz476hgvqvzojy"
USER_LOGIN = "Ammariii08"

HEADERS = {
    "Content-Type": "application/json",
    "Authorization": f"Bearer {RUNPOD_API_KEY}"
}

def extract_file_id(url):
    m = re.search(r'/d/([a-zA-Z0-9_-]+)', url) or re.search(r'id=([a-zA-Z0-9_-]+)', url)
    return m.group(1) if m else None

def to_direct_download(link):
    fid = extract_file_id(link)
    return f"https://drive.google.com/uc?export=download&id={fid}" if fid else None

def make_api_request(fid):
    payload = {"input": {"video_url": f"https://drive.google.com/uc?export=download&id={fid}"}}
    resp = requests.post(f"{RUNPOD_ENDPOINT}/run", headers=HEADERS, json=payload, timeout=1800)
    resp.raise_for_status()
    return resp.json()

def process_video(drive_link, progress=gr.Progress()):
    fid = extract_file_id(drive_link)
    if not fid:
        progress(1.0, desc="Invalid URL")
        return "❌ Invalid Drive URL", "", "–", ""
    progress(0.05, desc="Initializing…")
    t0 = time.time()
    try:
        job = make_api_request(fid)
    except Exception as e:
        progress(1.0, desc="Submission Failed")
        return f"❌ Submission error: {e}", "", "–", ""
    progress(0.2, desc="Job Submitted")
    t1 = time.time()
    job_id = job["id"]

    prog = 0.2
    while True:
        st = requests.get(f"{RUNPOD_ENDPOINT}/status/{job_id}", headers=HEADERS).json()
        status = st.get("status", "UNKNOWN")
        if status == "COMPLETED":
            t2 = time.time()
            exec_time = t2 - t1
            out = st["output"]
            dl_link = to_direct_download(out["drive_link"]) or out["drive_link"]
            break
        if status == "FAILED":
            progress(1.0, desc="Job Failed")
            return "❌ Job failed", "", "–", ""
        prog = min(prog + 0.1, 0.9)
        progress(prog, desc=f"Waiting ({status})…")
        time.sleep(5)

    progress(1.0, desc="Completed")
    delay = t1 - t0

    details_md = f"""
**Drive Link:**  
`{dl_link}`

**Job ID:** `{job_id}`  
**Status:** `{status}`  

**Submission Delay:** `{delay:.1f}s`  
**Execution Time:** `{exec_time:.1f}s`  
"""

    # Render a styled HTML link inside Markdown
    button_md = f'<a href="{dl_link}" target="_blank" style="display:inline-block;padding:8px 16px;background:#4CAF50;color:white;border-radius:4px;text-decoration:none;">Open Download</a>'

    return "✅ Completed!", dl_link, details_md, button_md

def create_ui():
    with gr.Blocks(theme=gr.themes.Soft()) as app:
        gr.Markdown(f"""
# AXION 🎾
## RUNPOD Deployment Testing
                    
""")
        link_in  = gr.Textbox(label="Google Drive Link", placeholder="https://drive.google.com/file/d/…")
        btn      = gr.Button("Process Video")
        status   = gr.Textbox(label="Status", interactive=False)
        dl_out   = gr.Textbox(label="Download Link", interactive=False)
        details  = gr.Markdown(label="Details")
        open_btn = gr.Markdown()  # <-- always present

        btn.click(
            fn=process_video,
            inputs=[link_in],
            outputs=[status, dl_out, details, open_btn]
        )

    return app

if __name__ == "__main__":
    create_ui().launch(share=True, server_name="0.0.0.0", server_port=7860)