Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,257 @@
+import gradio as gr
+import asyncio
+import shutil
+import os
+import tempfile
+from pathlib import Path
+from huggingface_hub import HfApi, login, create_repo, upload_file
+import git
+
+# ========================== GLOBAL ==========================
+api = HfApi()
+temp_dir = Path(tempfile.mkdtemp())
+token = None
+current_user = None
+last_download_path = None
+
+# ====================== ĐĂNG NHẬP / ĐĂNG XUẤT ======================
+def login_hf(t):
+    global token, current_user
+    t = t.strip()
+    if not t:
+        return "Token không được để trống!", gr.update(value="Chưa đăng nhập")
+
+    try:
+        login(t)
+        info = api.whoami(token=t)
+        token = t
+        current_user = info["name"]
+
+        status = f"Đã đăng nhập: **{current_user}**"
+        return "Đăng nhập thành công! Sẵn sàng sử dụng tất cả chức năng", gr.update(value=status)
+    except Exception as e:
+        current_user = None
+        token = None
+        return f"Login thất bại: {str(e)}", gr.update(value="Login thất bại")
+
+def logout_hf():
+    global token, current_user
+    from huggingface_hub import logout
+    logout()
+    token = None
+    current_user = None
+    return "Đã đăng xuất thành công!", gr.update(value="Chưa đăng nhập")
+
+# ====================== CÁC HÀM CHÍNH ======================
+async def download(url, type_, progress=gr.Progress()):
+    global last_download_path
+    if not url.strip():
+        return "Vui lòng nhập URL!", None
+
+    out = temp_dir / "downloaded"
+    if out.exists():
+        shutil.rmtree(out)
+    out.mkdir(parents=True, exist_ok=True)
+
+    try:
+        progress(0, desc="Chuẩn bị...")
+        if type_ == "git":
+            progress(0.2, desc="Đang clone bằng git...")
+            await asyncio.to_thread(git.Repo.clone_from, url.strip(), out, depth=1)
+        else:
+            progress(0.2, desc="Đang tải bằng wget...")
+            proc = await asyncio.create_subprocess_exec(
+                "wget", "-r", "-np", "-nH", "--cut-dirs=10", "-P", str(out), url.strip(),
+                stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE
+            )
+            await proc.communicate()
+
+        count = len(list(out.rglob("*")))
+        last_download_path = str(out)
+        return f"Tải thành công! Tìm thấy {count} file/thư mục", last_download_path
+    except Exception as e:
+        shutil.rmtree(out, ignore_errors=True)
+        return f"Lỗi tải xuống: {e}", None
+
+async def create_repo(name, rtype, private):
+    if not token:
+        return "Chưa đăng nhập Hugging Face!"
+    name = name.strip()
+    if "/" not in name:
+        return "Tên repo phải có định dạng username/repo-name!"
+
+    try:
+        create_repo(repo_id=name, repo_type=rtype, private=private, token=token, exist_ok=True)
+        return f"Tạo repo thành công!\nhttps://huggingface.co/{name}"
+    except Exception as e:
+        return f"Lỗi tạo repo: {e}"
+
+async def upload(folder, repo, rtype, zip_it, password, use_dl, dl_path, progress=gr.Progress()):
+    if not token:
+        return "Chưa đăng nhập!"
+
+    repo = repo.strip()
+    path = Path(dl_path if use_dl and dl_path else folder)
+    if not path or not path.exists():
+        return "Thư mục không tồn tại hoặc chưa tải!"
+
+    try:
+        if zip_it:
+            progress(0, desc="Đang nén thành ZIP...")
+            zip_name = f"{repo.split('/')[-1]}_archive.zip"
+            zip_path = temp_dir / zip_name
+            cmd = ["zip", "-r"] + (["-P", password] if password else []) + [str(zip_path), "."]
+            proc = await asyncio.create_subprocess_exec(*cmd, cwd=str(path))
+            await proc.wait()
+
+            progress(0.7, desc="Đang upload file ZIP...")
+            await asyncio.to_thread(upload_file,
+                                    path_or_fileobj=str(zip_path),
+                                    path_in_repo=zip_name,
+                                    repo_id=repo,
+                                    repo_type=rtype,
+                                    token=token)
+            return f"Upload ZIP thành công!\nhttps://huggingface.co/{repo}/blob/main/{zip_name}"
+
+        else:
+            files = [f for f in path.rglob("*") if f.is_file()]
+            if not files:
+                return "Không có file nào để upload!"
+
+            for i, f in enumerate(files):
+                rel = f.relative_to(path)
+                progress((i + 1) / len(files), desc=f"Upload {i+1}/{len(files)}: {rel}")
+                await asyncio.to_thread(upload_file,
+                                        path_or_fileobj=str(f),
+                                        path_in_repo=str(rel),
+                                        repo_id=repo,
+                                        repo_type=rtype,
+                                        token=token)
+
+            return f"Upload thành công {len(files)} file!\nhttps://huggingface.co/{repo}"
+    except Exception as e:
+        return f"Lỗi upload: {e}"
+
+async def delete_files(local, repo, rtype, dry, progress=gr.Progress()):
+    if not token:
+        return "Chưa đăng nhập!"
+
+    try:
+        local_path = Path(local)
+        if not local_path.exists():
+            return "Thư mục local không tồn tại!"
+
+        repo_files = await asyncio.to_thread(api.list_repo_files, repo_id=repo, repo_type=rtype, token=token)
+        to_del = [str(p.relative_to(local_path)) for p in local_path.rglob("*") if p.is_file() and str(p.relative_to(local_path)) in repo_files]
+
+        if not to_del:
+            return "Không tìm thấy file nào trùng để xóa trên repo."
+
+        if dry:
+            preview = "\n".join(f"- {f}" for f in to_del[:20])
+            more = f"\n... và {len(to_del)-20} file khác" if len(to_del) > 20 else ""
+            return f"**Dry-run**: Sẽ xóa {len(to_del)} file:\n{preview}{more}"
+
+        for i, f in enumerate(to_del):
+            progress((i + 1) / len(to_del), desc=f"Đang xóa {i+1}/{len(to_del)}")
+            await asyncio.to_thread(api.delete_file, path_in_repo=f, repo_id=repo, repo_type=rtype, token=token)
+
+        return f"Đã xóa thành công {len(to_del)} file trên repo!"
+    except Exception as e:
+        return f"Lỗi khi xóa: {e}"
+
+# ========================== GIAO DIỆN GRADIO ==========================
+with gr.Blocks(title="HF Manager Pro") as app:
+    gr.Markdown("# HF Manager Pro\nTải • Tạo Repo • Upload • Xóa file trên HF - Siêu tiện & nhanh!")
+
+    # ===== TRẠNG THÁI ĐĂNG NHẬP Ở ĐẦU =====
+    with gr.Row():
+        status_login = gr.Markdown("**Chưa đăng nhập**", elem_classes="status")
+        logout_btn = gr.Button("Đăng xuất", variant="stop", visible=False, size="sm")
+
+    with gr.Tabs():
+        # ==================== TAB LOGIN ====================
+        with gr.Tab("Đăng nhập HF"):
+            gr.Markdown("Dán token từ: https://huggingface.co/settings/tokens")
+            tk = gr.Textbox(label="Hugging Face Token", type="password", placeholder="hf_xxxxxxxxxxxxxxxxxxxxxx")
+            login_btn = gr.Button("Đăng nhập", variant="primary")
+            login_out = gr.Textbox(label="Kết quả")
+
+            login_btn.click(
+                login_hf,
+                inputs=tk,
+                outputs=[login_out, status_login]
+            ).then(
+                lambda: gr.update(visible=True),
+                outputs=logout_btn
+            )
+
+            logout_btn.click(
+                logout_hf,
+                outputs=[login_out, status_login]
+            ).then(
+                lambda: gr.update(visible=False),
+                outputs=logout_btn
+            )
+
+        # ==================== TAB TẢI XUỐNG ====================
+        with gr.Tab("Tải xuống"):
+            url = gr.Textbox(label="URL (GitHub, Google Drive public, git repo...)", placeholder="https://...")
+            typ = gr.Radio(["wget", "git"], label="Phương thức tải", value="wget")
+            dl_btn = gr.Button("Bắt đầu tải", variant="primary")
+            dl_out = gr.Textbox(label="Trạng thái tải")
+            dl_path = gr.Textbox(visible=False)
+
+            dl_btn.click(download, [url, typ], [dl_out, dl_path])
+
+        # ==================== TAB TẠO REPO ====================
+        with gr.Tab("Tạo Repo mới"):
+            name = gr.Textbox(label="Tên repo", placeholder="your-username/my-awesome-model")
+            rt = gr.Dropdown(["model", "dataset", "space"], value="model", label="Loại repo")
+            priv = gr.Checkbox(label="Repo riêng tư (Private)", value=True)
+            create_out = gr.Textbox(label="Kết quả", lines=3)
+            gr.Button("Tạo repo ngay", variant="primary").click(
+                create_repo, [name, rt, priv], create_out
+            )
+
+        # ==================== TAB UPLOAD ====================
+        with gr.Tab("Upload lên HF"):
+            folder = gr.Textbox(label="Thư mục local (bỏ qua nếu dùng tải sẵn)", placeholder="/path/to/your/folder")
+            use_dl = gr.Checkbox(label="Dùng thư mục vừa tải xuống (khuyên dùng)", value=True)
+            repo = gr.Textbox(label="Repo đích", placeholder="your-username/my-awesome-model")
+            rt2 = gr.Dropdown(["model", "dataset", "space"], value="model", label="Loại")
+            zipc = gr.Checkbox(label="Nén thành ZIP (có thể đặt mật khẩu)")
+            pw = gr.Textbox(label="Mật khẩu ZIP (nếu có)", type="password", visible=False)
+            zipc.change(lambda x: gr.update(visible=x), zipc, pw)
+
+            up_out = gr.Textbox(label="Kết quả upload", lines=5)
+            gr.Button("Upload ngay", variant="primary").click(
+                upload, [folder, repo, rt2, zipc, pw, use_dl, dl_path], up_out
+            )
+
+        # ==================== TAB XÓA FILE TRÊN REPO ====================
+        with gr.Tab("Xóa file trên Repo"):
+            gr.Markdown("So sánh thư mục local → xóa các file trùng trên repo")
+            loc = gr.Textbox(label="Thư mục local để đối chiếu", placeholder="/path/to/local/folder")
+            repo2 = gr.Textbox(label="Repo cần dọn dẹp", placeholder="username/repo")
+            rt3 = gr.Dropdown(["model", "dataset", "space"], value="model")
+            dry = gr.Checkbox(label="Chỉ xem trước (Dry-run) - an toàn", value=True)
+            del_out = gr.Textbox(label="Kết quả", lines=8)
+            gr.Button("Thực hiện xóa", variant="stop").click(
+                delete_files, [loc, repo2, rt3, dry], del_out
+            )
+
+    gr.Markdown("**Mẹo:** Sau khi tải xong → tick “Dùng thư mục vừa tải xuống” → Upload ngay không cần nhập đường dẫn!")
+
+    # CSS nhỏ cho đẹp
+    app.css = """
+    .status { font-size: 1.3em; font-weight: bold; margin: 10px 0; }
+    """
+
+# ========================== KHỞI CHẠY ==========================
+app.queue(max_size=20).launch(
+    share=True,
+    debug=True,
+    server_name="0.0.0.0",
+    server_port=7860
+)