Delete app.py

app.py

@@ -1,275 +0,0 @@
-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!"
-
-    # TỰ ĐỘNG PHÁT HIỆN LOẠI REPO THEO TÊN (ưu tiên hơn dropdown)
-    try:
-        repo_info = await asyncio.to_thread(api.repo_info, repo_id=repo, token=token)
-        actual_type = repo_info.repo_type
-        rtype = actual_type  # ép dùng đúng loại repo thực tế
-        progress(0, desc=f"Phát hiện repo loại: {actual_type}")
-    except:
-        # Nếu repo chưa tồn tại → dùng loại từ dropdown (giữ nguyên rtype người dùng chọn)
-        pass
-
-    try:
-        if zip_it:
-            git_dir = path / ".git"
-            if git_dir.exists():
-                shutil.rmtree(git_dir)
-                progress(0.1, desc="Đã tự động xóa .git")
-
-            progress(0.2, desc="Đang nén 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), stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
-            await proc.wait()
-
-            progress(0.7, desc="Upload 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() and not any(x in str(f.relative_to(path)) for x in ['.git/', '__pycache__', '.cache/', '.DS_Store'])
-            ]
-            if not files:
-                return "Không có file nào hợp lệ để upload!"
-
-            for i, f in enumerate(files):
-                rel = f.relative_to(path)
-                progress((i + 1) / len(files), desc=f"Upload {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
-)