Update app.py

app.py

@@ -1,8 +1,7 @@
 import os
 import uuid
-import subprocess
-import time
 import gradio as gr
+import requests
 from gradio_client import Client, handle_file
 
 # -------------------------
@@ -12,22 +11,7 @@ from gradio_client import Client, handle_file
 SPACE = "pockit-cloud/main"
 client = Client(SPACE)
 
-SPACE_URL = os.getenv("SPACE_URL", "https://pockit-cloud-main.hf.space")
-HF_TOKEN = os.getenv("HF_TOKEN")  # Secret for private dataset access
-
-TMP_DIR = "tmp"
-os.makedirs(TMP_DIR, exist_ok=True)
-
-# -------------------------
-# CLEAN 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)
+HF_TOKEN = os.getenv("HF_TOKEN")  # Secret for private dataset
 
 # -------------------------
 # LOGIN
@@ -64,31 +48,8 @@ def list_files(user_id, password):
         return str(e), []
 
 # -------------------------
-# UPLOAD
-# -------------------------
-def upload_file(user_id, password, file, custom_name):
-
-    if not user_id or not password:
-        return "Username and password required"
-
-    if not file:
-        return "No file selected"
-
-    try:
-        result = client.predict(
-            user_id=user_id,
-            password=password,
-            filepath=handle_file(file),
-            custom_name=custom_name if custom_name else os.path.basename(file),
-            api_name="/upload_file_secure"
-        )
-        return result[0]
-    except Exception as e:
-        return str(e)
-
-# -------------------------
-# DOWNLOAD (PRIVATE SAFE VERSION)
-# Returns file directly to gr.File
+# DOWNLOAD (STREAMED + PRIVATE SAFE)
+# Returns direct file for automatic link generation
 # -------------------------
 def download_file(user_id, password, filename):
 
@@ -99,9 +60,7 @@ def download_file(user_id, password, filename):
         return None
 
     try:
-        cleanup_tmp()
-
-        # Get private download link
+        # Get private download link from backend
         link = client.predict(
             user_id=user_id,
             password=password,
@@ -109,30 +68,49 @@ def download_file(user_id, password, filename):
             api_name="/get_download_link_action"
         )
 
-        temp_name = f"{uuid.uuid4()}_{filename}"
-        temp_path = os.path.join(TMP_DIR, temp_name)
-
-        cmd = [
-            "wget",
-            "--tries=3",
-            "--no-check-certificate",
-            "-O",
-            temp_path,
-            link
-        ]
-
-        # Add HuggingFace token if available
+        headers = {}
         if HF_TOKEN:
-            cmd.insert(1, f"--header=Authorization: Bearer {HF_TOKEN}")
+            headers["Authorization"] = f"Bearer {HF_TOKEN}"
+
+        r = requests.get(link, headers=headers, stream=True)
+        r.raise_for_status()
+
+        temp_name = f"{uuid.uuid4()}_{filename}"
 
-        subprocess.run(cmd, check=True)
+        with open(temp_name, "wb") as f:
+            for chunk in r.iter_content(chunk_size=8192):
+                if chunk:
+                    f.write(chunk)
 
-        return temp_path  # IMPORTANT: return path to gr.File
+        return temp_name  # Gradio auto-generates real link
 
     except Exception as e:
         print(e)
         return None
 
+# -------------------------
+# UPLOAD
+# -------------------------
+def upload_file(user_id, password, file, custom_name):
+
+    if not user_id or not password:
+        return "Username and password required"
+
+    if not file:
+        return "No file selected"
+
+    try:
+        result = client.predict(
+            user_id=user_id,
+            password=password,
+            filepath=handle_file(file),
+            custom_name=custom_name if custom_name else os.path.basename(file),
+            api_name="/upload_file_secure"
+        )
+        return result[0]
+    except Exception as e:
+        return str(e)
+
 # -------------------------
 # DELETE
 # -------------------------
@@ -252,5 +230,4 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=change_status
         )
 
-# IMPORTANT: allow tmp folder for serving
-app.launch(allowed_paths=[TMP_DIR])
+app.launch()