Update app.py

app.py

@@ -5,16 +5,12 @@ from gradio_client import Client
 
 SPACE = "pockit-cloud/main"
 
-HF_TOKEN = os.getenv("HF_TOKEN")
-if HF_TOKEN is None:
-    raise Exception("HF_TOKEN environment variable not set")
+# Connect to the backend Space
+client = Client(SPACE)
 
-client = Client(SPACE, hf_token=HF_TOKEN)
-
-# --------------------
+# -------------------------
 # Login
-# --------------------
-
+# -------------------------
 def login(user_id, password):
     try:
         files, status = client.predict(
@@ -24,72 +20,100 @@ def login(user_id, password):
         )
 
         return status, user_id, password, files
+
+    except Exception as e:
+        return f"Login error: {e}", "", "", []
+
+
+# -------------------------
+# List files
+# -------------------------
+def list_files(user_id, password):
+    if not user_id or not password:
+        return "Not logged in", []
+
+    try:
+        files, status = client.predict(
+            user_id=user_id,
+            password=password,
+            api_name="/get_files_secure"
+        )
+
+        return status, files
+
     except Exception as e:
-        return str(e), "", "", []
+        return str(e), []
 
-# --------------------
-# Change Password
-# --------------------
 
+# -------------------------
+# Change password
+# -------------------------
 def change_password(user_id, stored_pw, old_pw, new_pw):
+
+    if not user_id:
+        return "Not logged in"
+
     if old_pw != stored_pw:
         return "Old password incorrect"
 
-    result = client.predict(
-        user_id=user_id,
-        new_password=new_pw,
-        api_name="/update_password"
-    )
+    try:
+        result = client.predict(
+            user_id=user_id,
+            new_password=new_pw,
+            api_name="/update_password"
+        )
 
-    return result
+        return result
 
-# --------------------
-# List Files
-# --------------------
+    except Exception as e:
+        return str(e)
 
-def list_files(user_id, password):
-    files, status = client.predict(
-        user_id=user_id,
-        password=password,
-        api_name="/get_files_secure"
-    )
 
-    return status, files
+# -------------------------
+# Download file
+# -------------------------
+def download_file(user_id, password, filename):
 
-# --------------------
-# Download File
-# --------------------
+    if not filename:
+        return None
 
-def download_file(user_id, password, filename):
-    link = client.predict(
-        user_id=user_id,
-        password=password,
-        filename=filename,
-        api_name="/get_download_link_action"
-    )
+    try:
+        link = client.predict(
+            user_id=user_id,
+            password=password,
+            filename=filename,
+            api_name="/get_download_link_action"
+        )
+
+        r = requests.get(link)
 
-    r = requests.get(link)
+        os.makedirs("downloads", exist_ok=True)
+        path = f"downloads/{filename}"
 
-    path = f"downloads/{filename}"
-    os.makedirs("downloads", exist_ok=True)
+        with open(path, "wb") as f:
+            f.write(r.content)
 
-    with open(path, "wb") as f:
-        f.write(r.content)
+        return path
 
-    return path
+    except Exception as e:
+        print(e)
+        return None
 
 
-# --------------------
+# -------------------------
 # UI
-# --------------------
+# -------------------------
 
 with gr.Blocks(title="Pockit Cloud Client") as app:
 
     user_state = gr.State("")
     pass_state = gr.State("")
 
-    gr.Markdown("# Pockit Cloud Client")
+    gr.Markdown("# ☁️ Pockit Cloud Client")
 
+    # ---------------------
+    # LOGIN TAB
+    # ---------------------
     with gr.Tab("Login"):
 
         user = gr.Textbox(label="User ID")
@@ -105,6 +129,9 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[login_status, user_state, pass_state]
         )
 
+    # ---------------------
+    # FILES TAB
+    # ---------------------
     with gr.Tab("Files"):
 
         refresh_btn = gr.Button("Refresh File List")
@@ -119,7 +146,7 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[refresh_status, file_list]
         )
 
-        download_btn = gr.Button("Download")
+        download_btn = gr.Button("Download Selected File")
 
         download_output = gr.File(label="Downloaded File")
 
@@ -129,6 +156,9 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=download_output
         )
 
+    # ---------------------
+    # PASSWORD TAB
+    # ---------------------
     with gr.Tab("Password"):
 
         old_pw = gr.Textbox(label="Old Password", type="password")