Update app.py

app.py

@@ -1,10 +1,27 @@
 import os
+import uuid
+import subprocess
+import time
 import gradio as gr
 from gradio_client import Client, handle_file
 
 SPACE = "pockit-cloud/main"
 client = Client(SPACE)
 
+TMP_DIR = "tmp"
+os.makedirs(TMP_DIR, exist_ok=True)
+
+# -------------------------
+# CLEAN OLD TEMP FILES
+# -------------------------
+def cleanup_tmp():
+    now = time.time()
+    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
+                os.remove(path)
+
 # -------------------------
 # LOGIN
 # -------------------------
@@ -63,7 +80,7 @@ def upload_file(user_id, password, file, custom_name):
         return str(e)
 
 # -------------------------
-# GET DOWNLOAD LINK
+# DOWNLOAD USING WGET TEMP
 # -------------------------
 def download_file(user_id, password, filename):
 
@@ -74,6 +91,8 @@ def download_file(user_id, password, filename):
         return "No file selected"
 
     try:
+        cleanup_tmp()
+
         link = client.predict(
             user_id=user_id,
             password=password,
@@ -81,7 +100,16 @@ def download_file(user_id, password, filename):
             api_name="/get_download_link_action"
         )
 
-        return link
+        temp_name = f"{uuid.uuid4()}_{filename}"
+        temp_path = os.path.join(TMP_DIR, temp_name)
+
+        subprocess.run(
+            ["wget", "-q", "-O", temp_path, link],
+            check=True
+        )
+
+        return f"/file={temp_path}"
+
     except Exception as e:
         return str(e)
 
@@ -103,7 +131,6 @@ def delete_file(user_id, password, filename):
             filename=filename,
             api_name="/delete_file_secure"
         )
-
         return result
     except Exception as e:
         return str(e)
@@ -165,9 +192,9 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[status_box, file_list]
         )
 
-        gr.Markdown("### Get Download Link")
+        gr.Markdown("### Download File")
 
-        download_btn = gr.Button("Get Download Link")
+        download_btn = gr.Button("Download (Temporary Mirror)")
         download_output = gr.Textbox(label="Download Link")
 
         download_btn.click(