Update app.py

app.py

@@ -4,14 +4,15 @@ import gradio as gr
 from gradio_client import Client, handle_file
 
 SPACE = "pockit-cloud/main"
-
-# Connect to backend Space
 client = Client(SPACE)
 
 # -------------------------
-# LOGIN (required each time)
+# LOGIN
 # -------------------------
 def login(user_id, password):
+    if not user_id or not password:
+        return "Username and password required", []
+
     try:
         files, status = client.predict(
             user_id=user_id,
@@ -26,6 +27,9 @@ def login(user_id, password):
 # LIST FILES
 # -------------------------
 def list_files(user_id, password):
+    if not user_id or not password:
+        return "Username and password required", []
+
     try:
         files, status = client.predict(
             user_id=user_id,
@@ -41,6 +45,9 @@ def list_files(user_id, password):
 # -------------------------
 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"
 
@@ -52,7 +59,7 @@ def upload_file(user_id, password, file, custom_name):
             custom_name=custom_name if custom_name else os.path.basename(file),
             api_name="/upload_file_secure"
         )
-        return result[0]  # Status
+        return result[0]
     except Exception as e:
         return str(e)
 
@@ -61,6 +68,9 @@ def upload_file(user_id, password, file, custom_name):
 # -------------------------
 def download_file(user_id, password, filename):
 
+    if not user_id or not password:
+        return None
+
     if not filename:
         return None
 
@@ -86,10 +96,13 @@ def download_file(user_id, password, filename):
         return None
 
 # -------------------------
-# CHANGE PASSWORD
+# PASSWORD CHANGE
 # -------------------------
 def change_password(user_id, password, new_password):
 
+    if not user_id or not password or not new_password:
+        return "All fields required"
+
     try:
         result = client.predict(
             user_id=user_id,
@@ -107,11 +120,9 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
 
     gr.Markdown("# ☁️ Pockit Cloud Client")
 
-    # Shared credential inputs (required everywhere)
-    user = gr.Textbox(label="User ID", required=True)
-    pw = gr.Textbox(label="Password", type="password", required=True)
+    user = gr.Textbox(label="User ID")
+    pw = gr.Textbox(label="Password", type="password")
 
-    # ---------------- LOGIN ----------------
     with gr.Tab("Login"):
         login_btn = gr.Button("Login")
         login_status = gr.Textbox(label="Status")
@@ -123,7 +134,6 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[login_status, login_files]
         )
 
-    # ---------------- FILES ----------------
     with gr.Tab("Files"):
 
         refresh_btn = gr.Button("Refresh File List")
@@ -158,9 +168,7 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=upload_status
         )
 
-    # ---------------- PASSWORD ----------------
     with gr.Tab("Password"):
-
         new_pw = gr.Textbox(label="New Password", type="password")
         change_btn = gr.Button("Change Password")
         change_status = gr.Textbox(label="Status")