Update app.py

app.py

@@ -8,6 +8,9 @@ from gradio_client import Client, handle_file
 SPACE = "pockit-cloud/main"
 client = Client(SPACE)
 
+# Set your Space URL
+SPACE_URL = os.getenv("SPACE_URL", "https://pockit-cloud-main.hf.space")
+
 TMP_DIR = "tmp"
 os.makedirs(TMP_DIR, exist_ok=True)
 
@@ -19,7 +22,7 @@ def cleanup_tmp():
     for f in os.listdir(TMP_DIR):
         path = os.path.join(TMP_DIR, f)
         if os.path.isfile(path):
-            if now - os.path.getmtime(path) > 3600:  # 1 hour
+            if now - os.path.getmtime(path) > 3600:
                 os.remove(path)
 
 # -------------------------
@@ -80,7 +83,7 @@ def upload_file(user_id, password, file, custom_name):
         return str(e)
 
 # -------------------------
-# DOWNLOAD USING WGET TEMP
+# DOWNLOAD USING WGET MIRROR
 # -------------------------
 def download_file(user_id, password, filename):
 
@@ -104,11 +107,18 @@ def download_file(user_id, password, filename):
         temp_path = os.path.join(TMP_DIR, temp_name)
 
         subprocess.run(
-            ["wget", "-q", "-O", temp_path, link],
+            [
+                "wget",
+                "--tries=3",
+                "--no-check-certificate",
+                "-O",
+                temp_path,
+                link
+            ],
             check=True
         )
 
-        return f"/file={temp_path}"
+        return f"{SPACE_URL}/file={temp_path}"
 
     except Exception as e:
         return str(e)
@@ -192,9 +202,9 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[status_box, file_list]
         )
 
-        gr.Markdown("### Download File")
+        gr.Markdown("### Download File (Temporary Mirror)")
 
-        download_btn = gr.Button("Download (Temporary Mirror)")
+        download_btn = gr.Button("Download")
         download_output = gr.Textbox(label="Download Link")
 
         download_btn.click(