Create app.py

app.py

@@ -0,0 +1,133 @@
+import os
+import time
+import shutil
+import zipfile
+import gradio as gr
+from fastapi import FastAPI
+import uvicorn
+
+# --- CẤU HÌNH HỆ THỐNG ---
+BASE_UPLOAD_DIR = "bins"
+EXPIRATION_HOURS = 24
+
+if not os.path.exists(BASE_UPLOAD_DIR):
+    os.makedirs(BASE_UPLOAD_DIR)
+
+# --- CÁC HÀM XỬ LÝ (BACKEND) ---
+
+def clean_old_bins():
+    """Tự động xóa các Bin cũ hơn 24h"""
+    now = time.time()
+    cutoff = now - (EXPIRATION_HOURS * 3600)
+    if os.path.exists(BASE_UPLOAD_DIR):
+        for bin_id in os.listdir(BASE_UPLOAD_DIR):
+            bin_path = os.path.join(BASE_UPLOAD_DIR, bin_id)
+            if os.path.getmtime(bin_path) < cutoff:
+                shutil.rmtree(bin_path)
+
+def upload_to_bin(bin_id, files, progress=gr.Progress()):
+    clean_old_bins()
+    if not bin_id or not files:
+        return "⚠️ Thiếu mã Bin hoặc File!", None
+
+    bin_id = bin_id.strip()
+    bin_path = os.path.join(BASE_UPLOAD_DIR, bin_id)
+    if not os.path.exists(bin_path):
+        os.makedirs(bin_path)
+    
+    total_files = len(files)
+    uploaded_list = []
+    
+    for i, file in enumerate(files):
+        # Hiển thị % xử lý trên Server
+        progress((i + 1) / total_files, desc=f"Đang lưu file {i+1}/{total_files}...")
+        dest_path = os.path.join(bin_path, os.path.basename(file.name))
+        shutil.copy(file.name, dest_path)
+        uploaded_list.append(dest_path)
+        
+    os.utime(bin_path, None) # Reset thời gian tồn tại (gia hạn 24h)
+    return f"✅ Đã tải lên thành công {total_files} file vào Bin: {bin_id}", uploaded_list
+
+def get_bin_files(bin_id):
+    clean_old_bins()
+    if not bin_id: return "⚠️ Vui lòng nhập mã Bin!", None
+    
+    bin_path = os.path.join(BASE_UPLOAD_DIR, bin_id.strip())
+    if not os.path.exists(bin_path):
+        return f"❌ Bin '{bin_id}' không tồn tại hoặc đã hết hạn.", None
+    
+    files = [os.path.join(bin_path, f) for f in os.listdir(bin_path) if f != "all_files.zip"]
+    return f"📂 Tìm thấy {len(files)} file trong Bin: {bin_id}", files
+
+def download_all_zip(bin_id, progress=gr.Progress()):
+    if not bin_id: return None
+    bin_path = os.path.join(BASE_UPLOAD_DIR, bin_id.strip())
+    if not os.path.exists(bin_path): return None
+    
+    files = [f for f in os.listdir(bin_path) if f != "all_files.zip"]
+    if not files: return None
+    
+    zip_path = os.path.join(bin_path, "all_files.zip")
+    with zipfile.ZipFile(zip_path, 'w') as zipf:
+        for i, f in enumerate(files):
+            progress((i + 1) / len(files), desc="Đang nén file...")
+            zipf.write(os.path.join(bin_path, f), f)
+            
+    return zip_path
+
+def delete_bin(bin_id):
+    if not bin_id: return "⚠️ Nhập mã Bin cần xóa!", None
+    bin_path = os.path.join(BASE_UPLOAD_DIR, bin_id.strip())
+    if os.path.exists(bin_path):
+        shutil.rmtree(bin_path)
+        return f"🗑️ Đã xóa vĩnh viễn Bin: {bin_id}", None
+    return "❌ Bin không tồn tại.", None
+
+# --- GIAO DIỆN GRADIO ---
+
+with gr.Blocks(theme=gr.themes.Soft(), title="HF FileBin Pro") as demo:
+    gr.Markdown("# 🚀 HF FileBin Pro (Bản Full)")
+    gr.Markdown(f"Dịch vụ lưu trữ file tạm thời (Tự xóa sau {EXPIRATION_HOURS}h không hoạt động)")
+    
+    with gr.Row():
+        bin_id_input = gr.Textbox(
+            label="Mã Bin (Private ID)", 
+            placeholder="Nhập mã bí mật để tạo hoặc truy cập...",
+            info="Ai có mã này mới có thể xem/tải file."
+        )
+        
+    with gr.Row():
+        with gr.Column(scale=1):
+            file_input = gr.File(label="Chọn Files (Hỗ trợ kéo thả)", file_count="multiple")
+            btn_upload = gr.Button("📤 Tải lên & Xử lý", variant="primary")
+            
+        with gr.Column(scale=1):
+            status_msg = gr.Textbox(label="Thông báo hệ thống", interactive=False)
+            file_output = gr.File(label="Danh sách file khả dụng", file_count="multiple")
+            
+            with gr.Row():
+                btn_access = gr.Button("🔍 Kiểm tra Bin")
+                btn_zip = gr.Button("📦 Tải ZIP (Tất cả)")
+                btn_delete = gr.Button("🗑️ Xóa Bin", variant="stop")
+
+    # Gán sự kiện
+    btn_upload.click(fn=upload_to_bin, inputs=[bin_id_input, file_input], outputs=[status_msg, file_output])
+    btn_access.click(fn=get_bin_files, inputs=[bin_id_input], outputs=[status_msg, file_output])
+    btn_zip.click(fn=download_all_zip, inputs=[bin_id_input], outputs=[file_output])
+    btn_delete.click(fn=delete_bin, inputs=[bin_id_input], outputs=[status_msg, file_output])
+
+# --- TÍCH HỢP FASTAPI CHO UPTIME ROBOT ---
+
+fastapi_app = FastAPI()
+
+@fastapi_app.get("/health")
+def health_check():
+    """Endpoint dành riêng cho Uptime Robot ping"""
+    return {"status": "online", "active_bins": len(os.listdir(BASE_UPLOAD_DIR)) if os.path.exists(BASE_UPLOAD_DIR) else 0}
+
+# Mount Gradio vào FastAPI
+app = gr.mount_gradio_app(fastapi_app, demo, path="/")
+
+if __name__ == "__main__":
+    # Chạy trên port 7860 (Port mặc định của Hugging Face)
+    uvicorn.run(app, host="0.0.0.0", port=7860)