Update app.py

app.py

@@ -1,233 +1,72 @@
 import os
-import uuid
+from fastapi import FastAPI
+from fastapi.responses import FileResponse
 import gradio as gr
-import requests
-from gradio_client import Client, handle_file
 
 # -------------------------
 # CONFIG
 # -------------------------
 
-SPACE = "pockit-cloud/main"
-client = Client(SPACE)
+SPACE_NAME = os.getenv("SPACE_ID", "your-username/your-space")
+BASE_URL = f"https://{SPACE_NAME.split('/')[-1]}.hf.space"
 
-HF_TOKEN = os.getenv("HF_TOKEN")  # Secret for private dataset
+FILE_DIR = "files"
+os.makedirs(FILE_DIR, exist_ok=True)
 
 # -------------------------
-# LOGIN
+# FASTAPI SERVER
 # -------------------------
-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,
-            password=password,
-            api_name="/get_files_secure"
-        )
-        return status, files
-    except Exception as e:
-        return f"Login error: {e}", []
 
-# -------------------------
-# 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,
-            password=password,
-            api_name="/get_files_secure"
-        )
-        return status, files
-    except Exception as e:
-        return str(e), []
+server = FastAPI()
 
-# -------------------------
-# DOWNLOAD (STREAMED + PRIVATE SAFE)
-# Returns direct file for automatic link generation
-# -------------------------
-def download_file(user_id, password, filename):
+@server.get("/download/{filename}")
+def download_file(filename: str):
+    file_path = os.path.join(FILE_DIR, filename)
 
-    if not user_id or not password:
-        return None
-
-    if not filename:
-        return None
-
-    try:
-        # Get private download link from backend
-        link = client.predict(
-            user_id=user_id,
-            password=password,
-            filename=filename,
-            api_name="/get_download_link_action"
+    if os.path.exists(file_path):
+        return FileResponse(
+            path=file_path,
+            filename=filename
         )
 
-        headers = {}
-        if HF_TOKEN:
-            headers["Authorization"] = f"Bearer {HF_TOKEN}"
-
-        r = requests.get(link, headers=headers, stream=True)
-        r.raise_for_status()
-
-        temp_name = f"{uuid.uuid4()}_{filename}"
-
-        with open(temp_name, "wb") as f:
-            for chunk in r.iter_content(chunk_size=8192):
-                if chunk:
-                    f.write(chunk)
-
-        return temp_name  # Gradio auto-generates real link
-
-    except Exception as e:
-        print(e)
-        return None
+    return {"error": "File not found"}
 
 # -------------------------
-# UPLOAD
+# LINK GENERATOR
 # -------------------------
-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"
-
-    try:
-        result = client.predict(
-            user_id=user_id,
-            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[0]
-    except Exception as e:
-        return str(e)
-
-# -------------------------
-# DELETE
-# -------------------------
-def delete_file(user_id, password, filename):
-
-    if not user_id or not password:
-        return "Username and password required"
+def generate_link(filename):
+    file_path = os.path.join(FILE_DIR, filename)
 
     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)
+        return "Enter filename"
 
-# -------------------------
-# 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"
+    if not os.path.exists(file_path):
+        return "File does not exist"
 
-    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)
+    return f"{BASE_URL}/download/{filename}"
 
 # -------------------------
-# UI
+# SIMPLE UI
 # -------------------------
-with gr.Blocks(title="Pockit Cloud Client") as app:
 
-    gr.Markdown("# ☁️ Pockit Cloud Client")
+with gr.Blocks() as app:
 
-    user = gr.Textbox(label="User ID")
-    pw = gr.Textbox(label="Password", type="password")
-
-    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, login_files]
-        )
+    gr.Markdown("# 🔗 HF Spaces Link Generator")
 
-    with gr.Tab("Files"):
+    filename_input = gr.Textbox(label="Filename (must exist in /files)")
+    link_output = gr.Textbox(label="Direct Link")
 
-        refresh_btn = gr.Button("Refresh File List")
-        file_list = gr.Dropdown(label="Your Files", choices=[])
-        status_box = gr.Textbox(label="Status")
+    btn = gr.Button("Generate Link")
 
-        refresh_btn.click(
-            list_files,
-            inputs=[user, pw],
-            outputs=[status_box, file_list]
-        )
-
-        gr.Markdown("### Download")
-
-        download_btn = gr.Button("Download File")
-        download_output = gr.File(label="Downloaded File")
-
-        download_btn.click(
-            download_file,
-            inputs=[user, pw, file_list],
-            outputs=download_output
-        )
+    btn.click(
+        generate_link,
+        inputs=filename_input,
+        outputs=link_output
+    )
 
-        gr.Markdown("### Upload")
-
-        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
-        )
-
-        gr.Markdown("### Delete")
-
-        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
-        )
-
-    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")
-
-        change_btn.click(
-            change_password,
-            inputs=[user, pw, new_pw],
-            outputs=change_status
-        )
+# IMPORTANT: mount FastAPI
+app = gr.mount_gradio_app(server, app)
 
+# Launch for Spaces
 app.launch()