Update app.py

app.py

@@ -63,7 +63,7 @@ def upload_file(user_id, password, file, custom_name):
         return str(e)
 
 # -------------------------
-# GET DOWNLOAD LINK ONLY
+# GET DOWNLOAD LINK
 # -------------------------
 def download_file(user_id, password, filename):
 
@@ -81,8 +81,30 @@ def download_file(user_id, password, filename):
             api_name="/get_download_link_action"
         )
 
-        return link  # just display the link
+        return link
+    except Exception as e:
+        return str(e)
 
+# -------------------------
+# DELETE FILE
+# -------------------------
+def delete_file(user_id, password, filename):
+
+    if not user_id or not password:
+        return "Username and password required"
+
+    if not filename:
+        return "No file selected"
+
+    try:
+        result = client.predict(
+            user_id=user_id,
+            password=password,
+            filename=filename,
+            api_name="/delete_file_secure"
+        )
+
+        return result
     except Exception as e:
         return str(e)
 
@@ -114,6 +136,9 @@ 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")
@@ -125,6 +150,9 @@ 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")
@@ -137,6 +165,8 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[status_box, file_list]
         )
 
+        gr.Markdown("### Get Download Link")
+
         download_btn = gr.Button("Get Download Link")
         download_output = gr.Textbox(label="Download Link")
 
@@ -159,6 +189,20 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=upload_status
         )
 
+        gr.Markdown("### Delete File")
+
+        delete_btn = gr.Button("Delete Selected File")
+        delete_status = gr.Textbox(label="Delete Status")
+
+        delete_btn.click(
+            delete_file,
+            inputs=[user, pw, file_list],
+            outputs=delete_status
+        )
+
+    # -------------------------
+    # PASSWORD TAB
+    # -------------------------
     with gr.Tab("Password"):
         new_pw = gr.Textbox(label="New Password", type="password")
         change_btn = gr.Button("Change Password")