Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import requests
 import gradio as gr
 from gradio_client import Client, handle_file
 
@@ -64,15 +63,15 @@ def upload_file(user_id, password, file, custom_name):
         return str(e)
 
 # -------------------------
-# DOWNLOAD FILE
+# GET DOWNLOAD LINK ONLY
 # -------------------------
 def download_file(user_id, password, filename):
 
     if not user_id or not password:
-        return None
+        return "Username and password required"
 
     if not filename:
-        return None
+        return "No file selected"
 
     try:
         link = client.predict(
@@ -82,18 +81,10 @@ def download_file(user_id, password, filename):
             api_name="/get_download_link_action"
         )
 
-        r = requests.get(link)
-
-        os.makedirs("downloads", exist_ok=True)
-        path = f"downloads/{filename}"
+        return link  # just display the link
 
-        with open(path, "wb") as f:
-            f.write(r.content)
-
-        return path
     except Exception as e:
-        print(e)
-        return None
+        return str(e)
 
 # -------------------------
 # PASSWORD CHANGE
@@ -146,8 +137,8 @@ with gr.Blocks(title="Pockit Cloud Client") as app:
             outputs=[status_box, file_list]
         )
 
-        download_btn = gr.Button("Download Selected File")
-        download_output = gr.File(label="Downloaded File")
+        download_btn = gr.Button("Get Download Link")
+        download_output = gr.Textbox(label="Download Link")
 
         download_btn.click(
             download_file,