Update app.py

app.py

@@ -8,8 +8,8 @@ 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")
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 TMP_DIR = "tmp"
 os.makedirs(TMP_DIR, exist_ok=True)
@@ -21,9 +21,8 @@ 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:
-                os.remove(path)
+        if os.path.isfile(path) and now - os.path.getmtime(path) > 3600:
+            os.remove(path)
 
 # -------------------------
 # LOGIN
@@ -106,17 +105,19 @@ def download_file(user_id, password, filename):
         temp_name = f"{uuid.uuid4()}_{filename}"
         temp_path = os.path.join(TMP_DIR, temp_name)
 
-        subprocess.run(
-            [
-                "wget",
-                "--tries=3",
-                "--no-check-certificate",
-                "-O",
-                temp_path,
-                link
-            ],
-            check=True
-        )
+        wget_cmd = [
+            "wget",
+            "--tries=3",
+            "--no-check-certificate",
+            "-O",
+            temp_path,
+            link
+        ]
+
+        if HF_TOKEN:
+            wget_cmd.insert(1, f"--header=Authorization: Bearer {HF_TOKEN}")
+
+        subprocess.run(wget_cmd, check=True)
 
         return f"{SPACE_URL}/file={temp_path}"
 
@@ -173,9 +174,6 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
     user = gr.Textbox(label="User ID")
     pw = gr.Textbox(label="Password", type="password")
 
-    # -------------------------
-    # LOGIN TAB
-    # -------------------------
     with gr.Tab("Login"):
         login_btn = gr.Button("Login")
         login_status = gr.Textbox(label="Status")
@@ -187,9 +185,6 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[login_status, login_files]
         )
 
-    # -------------------------
-    # FILES TAB
-    # -------------------------
     with gr.Tab("Files"):
 
         refresh_btn = gr.Button("Refresh File List")
@@ -202,7 +197,7 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[status_box, file_list]
         )
 
-        gr.Markdown("### Download File (Temporary Mirror)")
+        gr.Markdown("### Download File")
 
         download_btn = gr.Button("Download")
         download_output = gr.Textbox(label="Download Link")
@@ -237,9 +232,6 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=delete_status
         )
 
-    # -------------------------
-    # PASSWORD TAB
-    # -------------------------
     with gr.Tab("Password"):
         new_pw = gr.Textbox(label="New Password", type="password")
         change_btn = gr.Button("Change Password")