Update app.py

app.py

@@ -1,15 +1,15 @@
 import os
 import requests
 import gradio as gr
-from gradio_client import Client
+from gradio_client import Client, handle_file
 
 SPACE = "pockit-cloud/main"
 
-# Connect to the backend Space
+# Connect to backend Space
 client = Client(SPACE)
 
 # -------------------------
-# Login
+# LOGIN (required each time)
 # -------------------------
 def login(user_id, password):
     try:
@@ -18,59 +18,46 @@ def login(user_id, password):
             password=password,
             api_name="/get_files_secure"
         )
-
-        return status, user_id, password, files
-
+        return status, files
     except Exception as e:
-        return f"Login error: {e}", "", "", []
-
+        return f"Login error: {e}", []
 
 # -------------------------
-# List files
+# 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), []
 
-
 # -------------------------
-# Change password
+# UPLOAD FILE
 # -------------------------
-def change_password(user_id, stored_pw, old_pw, new_pw):
-
-    if not user_id:
-        return "Not logged in"
+def upload_file(user_id, password, file, custom_name):
 
-    if old_pw != stored_pw:
-        return "Old password incorrect"
+    if not file:
+        return "No file selected"
 
     try:
         result = client.predict(
             user_id=user_id,
-            new_password=new_pw,
-            api_name="/update_password"
+            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
-
+        return result[0]  # Status
     except Exception as e:
         return str(e)
 
-
 # -------------------------
-# Download file
+# DOWNLOAD FILE
 # -------------------------
 def download_file(user_id, password, filename):
 
@@ -94,85 +81,94 @@ def download_file(user_id, password, filename):
             f.write(r.content)
 
         return path
-
     except Exception as e:
         print(e)
         return None
 
+# -------------------------
+# CHANGE PASSWORD
+# -------------------------
+def change_password(user_id, password, new_password):
+
+    try:
+        result = client.predict(
+            user_id=user_id,
+            new_password=new_password,
+            api_name="/update_password"
+        )
+        return result
+    except Exception as e:
+        return str(e)
 
 # -------------------------
 # UI
 # -------------------------
-
 with gr.Blocks(title="Pockit Cloud Client") as app:
 
-    user_state = gr.State("")
-    pass_state = gr.State("")
-
     gr.Markdown("# ☁️ Pockit Cloud Client")
 
-    # ---------------------
-    # LOGIN TAB
-    # ---------------------
-    with gr.Tab("Login"):
-
-        user = gr.Textbox(label="User ID")
-        pw = gr.Textbox(label="Password", type="password")
+    # Shared credential inputs (required everywhere)
+    user = gr.Textbox(label="User ID", required=True)
+    pw = gr.Textbox(label="Password", type="password", required=True)
 
+    # ---------------- LOGIN ----------------
+    with gr.Tab("Login"):
         login_btn = gr.Button("Login")
-
         login_status = gr.Textbox(label="Status")
+        login_files = gr.Dropdown(label="Files", choices=[])
 
         login_btn.click(
             login,
             inputs=[user, pw],
-            outputs=[login_status, user_state, pass_state]
+            outputs=[login_status, login_files]
         )
 
-    # ---------------------
-    # FILES TAB
-    # ---------------------
+    # ---------------- FILES ----------------
     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")
+        status_box = gr.Textbox(label="Status")
 
         refresh_btn.click(
             list_files,
-            inputs=[user_state, pass_state],
-            outputs=[refresh_status, file_list]
+            inputs=[user, pw],
+            outputs=[status_box, file_list]
         )
 
         download_btn = gr.Button("Download Selected File")
-
         download_output = gr.File(label="Downloaded File")
 
         download_btn.click(
             download_file,
-            inputs=[user_state, pass_state, file_list],
+            inputs=[user, pw, file_list],
             outputs=download_output
         )
 
-    # ---------------------
-    # PASSWORD TAB
-    # ---------------------
+        gr.Markdown("### Upload File")
+
+        upload_input = gr.File(label="Select File")
+        custom_name = gr.Textbox(label="Custom Name (Optional)")
+        upload_btn = gr.Button("Upload")
+        upload_status = gr.Textbox(label="Upload Status")
+
+        upload_btn.click(
+            upload_file,
+            inputs=[user, pw, upload_input, custom_name],
+            outputs=upload_status
+        )
+
+    # ---------------- PASSWORD ----------------
     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],
+            inputs=[user, pw, new_pw],
             outputs=change_status
         )
 
-
 app.launch()