Update app.py

app.py

@@ -0,0 +1,148 @@
+import os
+import requests
+import gradio as gr
+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")
+
+client = Client(SPACE, hf_token=HF_TOKEN)
+
+# --------------------
+# Login
+# --------------------
+
+def login(user_id, password):
+    try:
+        files, status = client.predict(
+            user_id=user_id,
+            password=password,
+            api_name="/get_files_secure"
+        )
+
+        return status, user_id, password, files
+    except Exception as e:
+        return str(e), "", "", []
+
+# --------------------
+# Change Password
+# --------------------
+
+def change_password(user_id, stored_pw, old_pw, new_pw):
+    if old_pw != stored_pw:
+        return "Old password incorrect"
+
+    result = client.predict(
+        user_id=user_id,
+        new_password=new_pw,
+        api_name="/update_password"
+    )
+
+    return result
+
+# --------------------
+# List Files
+# --------------------
+
+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):
+    link = client.predict(
+        user_id=user_id,
+        password=password,
+        filename=filename,
+        api_name="/get_download_link_action"
+    )
+
+    r = requests.get(link)
+
+    path = f"downloads/{filename}"
+    os.makedirs("downloads", exist_ok=True)
+
+    with open(path, "wb") as f:
+        f.write(r.content)
+
+    return path
+
+
+# --------------------
+# UI
+# --------------------
+
+with gr.Blocks(title="Pockit Cloud Client") as app:
+
+    user_state = gr.State("")
+    pass_state = gr.State("")
+
+    gr.Markdown("# Pockit Cloud Client")
+
+    with gr.Tab("Login"):
+
+        user = gr.Textbox(label="User ID")
+        pw = gr.Textbox(label="Password", type="password")
+
+        login_btn = gr.Button("Login")
+
+        login_status = gr.Textbox(label="Status")
+
+        login_btn.click(
+            login,
+            inputs=[user, pw],
+            outputs=[login_status, user_state, pass_state]
+        )
+
+    with gr.Tab("Files"):
+
+        refresh_btn = gr.Button("Refresh File List")
+
+        file_list = gr.Dropdown(label="Your Files", choices=[])
+
+        refresh_status = gr.Textbox(label="Status")
+
+        refresh_btn.click(
+            list_files,
+            inputs=[user_state, pass_state],
+            outputs=[refresh_status, file_list]
+        )
+
+        download_btn = gr.Button("Download")
+
+        download_output = gr.File(label="Downloaded File")
+
+        download_btn.click(
+            download_file,
+            inputs=[user_state, pass_state, file_list],
+            outputs=download_output
+        )
+
+    with gr.Tab("Password"):
+
+        old_pw = gr.Textbox(label="Old Password", type="password")
+        new_pw = gr.Textbox(label="New Password", type="password")
+
+        change_btn = gr.Button("Change Password")
+
+        change_status = gr.Textbox(label="Status")
+
+        change_btn.click(
+            change_password,
+            inputs=[user_state, pass_state, old_pw, new_pw],
+            outputs=change_status
+        )
+
+
+app.launch()