Create app.py

app.py

@@ -0,0 +1,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)