Upload 5 files

config.py

@@ -0,0 +1,25 @@
+import os
+
+# Cấu hình chung
+DRIVE_EXCEL_FILE_ID = "ID_FILE_EXCEL_CUA_BAN"  # Thay ID file input.xlsx trên Drive vào đây
+LOCAL_EXCEL_PATH = "input.xlsx"
+TEMP_VIDEO_DIR = "temp_videos"
+
+# Cấu hình các Page (Thêm Page mới thì thêm 1 block vào đây)
+PAGES_CONFIG = [
+    {
+        "sheet_name": "BeYeu",           # Tên Sheet trong Excel
+        "page_id": os.getenv("PAGE_ID_BEYEU"), 
+        "access_token": os.getenv("TOKEN_BEYEU"),
+        "schedule_times": ["07:00", "19:00"], # Giờ đăng bài
+        "active": True
+    },
+    {
+        "sheet_name": "BeYeu2",
+        "page_id": os.getenv("PAGE_ID_BEYEU2"),
+        "access_token": os.getenv("TOKEN_BEYEU2"),
+        "schedule_times": ["08:00", "20:00"],
+        "active": True
+    },
+    # Muốn thêm BeYeu3, chỉ cần copy block trên và đổi thông tin
+]

facebook_utils.py

@@ -0,0 +1,50 @@
+import httpx
+import asyncio
+import os
+
+async def upload_reel(page_id, access_token, video_path, description):
+    api_version = "v19.0"
+    base_url = f"https://graph.facebook.com/{api_version}"
+    
+    async with httpx.AsyncClient(timeout=120.0) as client:
+        # 1. Init
+        init_res = await client.post(
+            f"{base_url}/{page_id}/video_reels",
+            data={"upload_phase": "start", "access_token": access_token}
+        )
+        init_data = init_res.json()
+        video_id = init_data.get("video_id")
+        upload_url = init_data.get("upload_url")
+        
+        if not video_id: raise Exception(f"Init Error: {init_data}")
+
+        # 2. Upload
+        file_size = os.path.getsize(video_path)
+        with open(video_path, "rb") as f:
+            video_data = f.read()
+        
+        headers = {"Authorization": f"OAuth {access_token}", "offset": "0", "file_size": str(file_size)}
+        upload_res = await client.post(upload_url, content=video_data, headers=headers)
+        if upload_res.status_code != 200: raise Exception(f"Upload Error: {upload_res.text}")
+
+        # 3. Publish
+        pub_res = await client.post(
+            f"{base_url}/{page_id}/video_reels",
+            data={
+                "access_token": access_token,
+                "video_id": video_id,
+                "upload_phase": "finish",
+                "video_state": "PUBLISHED",
+                "description": description
+            }
+        )
+        if not pub_res.json().get("success"): raise Exception(f"Publish Error: {pub_res.text}")
+        
+        return video_id
+
+async def comment_on_video(video_id, access_token, message):
+    # Đợi 1 chút để video xử lý
+    await asyncio.sleep(15) 
+    url = f"https://graph.facebook.com/v19.0/{video_id}/comments"
+    async with httpx.AsyncClient() as client:
+        await client.post(url, data={"message": message, "access_token": access_token})

google_utils.py

@@ -0,0 +1,63 @@
+import pandas as pd
+import google.generativeai as genai
+import random
+import asyncio
+import os
+from googleapiclient.discovery import build
+from googleapiclient.http import MediaFileUpload, MediaIoBaseDownload
+import io
+import pickle
+
+# --- Biến toàn cục cho Gemini Keys ---
+GEMINI_KEYS = []
+
+async def load_gemini_keys(excel_path):
+    """Đọc list API Key từ sheet APIKey"""
+    global GEMINI_KEYS
+    try:
+        df = pd.read_excel(excel_path, sheet_name='APIKey', header=None)
+        # Giả sử key nằm ở cột A, từ dòng 2
+        GEMINI_KEYS = df.iloc[1:, 0].dropna().astype(str).tolist()
+        print(f"✅ Đã load {len(GEMINI_KEYS)} Gemini Keys.")
+    except Exception as e:
+        print(f"❌ Lỗi load Gemini Keys: {e}")
+
+async def generate_content_with_gemini(product_desc):
+    """Viết lại nội dung bằng Gemini với cơ chế xoay vòng Key"""
+    if not GEMINI_KEYS:
+        return product_desc # Fallback nếu lỗi
+
+    prompt = (
+        f"Viết một caption Facebook Reels ngắn gọn, hấp dẫn, bắt trend "
+        f"cho sản phẩm sau. Có icon sinh động và kêu gọi mua hàng khéo léo.\n"
+        f"Sản phẩm: {product_desc}"
+    )
+
+    # Thử ngẫu nhiên key cho đến khi thành công
+    for _ in range(3):
+        key = random.choice(GEMINI_KEYS)
+        genai.configure(api_key=key)
+        try:
+            model = genai.GenerativeModel('gemini-1.5-flash')
+            response = await model.generate_content_async(prompt)
+            return response.text.strip()
+        except Exception:
+            continue
+    return product_desc
+
+# --- Hàm Drive (Tái sử dụng code của bạn) ---
+# (Bạn giữ nguyên các hàm get_drive_credentials, download_file_from_drive, upload_file_to_drive cũ)
+# Chỉ thêm hàm download video theo ID:
+
+async def download_video_by_id(drive_service, file_id, save_path):
+    if os.path.exists(save_path):
+        os.remove(save_path) # Xóa file cũ nếu có
+    request = drive_service.files().get_media(fileId=file_id)
+    fh = io.BytesIO()
+    downloader = MediaIoBaseDownload(fh, request)
+    done = False
+    while not done:
+        status, done = downloader.next_chunk()
+    with open(save_path, 'wb') as f:
+        f.write(fh.getvalue())
+    return save_path

main.py

@@ -0,0 +1,60 @@
+from fastapi import FastAPI
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+from apscheduler.triggers.cron import CronTrigger
+import asyncio
+from contextlib import asynccontextmanager
+import config
+import processor
+import google_utils
+
+# --- Khởi tạo Scheduler ---
+scheduler = AsyncIOScheduler()
+
+async def job_runner(sheet_name):
+    """Hàm wrapper để chạy job"""
+    # Tìm config dựa trên sheet_name
+    page_cfg = next((p for p in config.PAGES_CONFIG if p['sheet_name'] == sheet_name), None)
+    if not page_cfg: return
+
+    # Lấy credentials (dùng lại hàm cũ của bạn)
+    # Lưu ý: cần truyền chat_id nếu dùng hàm cũ, hoặc sửa hàm get_drive_credentials để không cần chat_id
+    credentials = await google_utils.get_drive_credentials_no_chat_id() 
+    if credentials:
+        await google_utils.load_gemini_keys(config.LOCAL_EXCEL_PATH) # Load key mới nhất
+        await processor.process_single_page(page_cfg, credentials)
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    print("🚀 Bot khởi động...")
+    
+    # 1. Tải file Excel lần đầu để lấy API Key
+    # (Thực hiện logic tải file và load key ở đây)
+    
+    # 2. Đăng ký lịch chạy từ Config
+    for page in config.PAGES_CONFIG:
+        if page['active']:
+            for time_str in page['schedule_times']:
+                hour, minute = time_str.split(':')
+                scheduler.add_job(
+                    job_runner, 
+                    CronTrigger(hour=int(hour), minute=int(minute)),
+                    args=[page['sheet_name']],
+                    name=f"{page['sheet_name']}_{time_str}"
+                )
+                print(f"📅 Đã hẹn giờ: {page['sheet_name']} lúc {time_str}")
+
+    scheduler.start()
+    yield
+    print("🛑 Bot dừng.")
+
+app = FastAPI(lifespan=lifespan)
+
+@app.get("/")
+def health_check():
+    return {"status": "Running", "jobs": [str(j) for j in scheduler.get_jobs()]}
+
+# Endpoint để test thủ công 1 page
+@app.post("/trigger/{sheet_name}")
+async def manual_trigger(sheet_name: str):
+    await job_runner(sheet_name)
+    return {"message": f"Triggered {sheet_name}"}

processor.py

@@ -0,0 +1,103 @@
+import pandas as pd
+import random
+import os
+import asyncio
+from googleapiclient.discovery import build
+from config import LOCAL_EXCEL_PATH, DRIVE_EXCEL_FILE_ID, TEMP_VIDEO_DIR
+import google_utils
+import facebook_utils
+
+# Lock để tránh 2 page cùng ghi file Excel 1 lúc
+excel_lock = asyncio.Lock()
+
+async def process_single_page(page_config, drive_credentials):
+    sheet_name = page_config['sheet_name']
+    print(f"🚀 [Processor] Bắt đầu xử lý cho Sheet: {sheet_name}")
+    
+    drive_service = build('drive', 'v3', credentials=drive_credentials)
+    
+    async with excel_lock:
+        # 1. Tải file Excel mới nhất về
+        await google_utils.download_file_from_drive(drive_service, DRIVE_EXCEL_FILE_ID, LOCAL_EXCEL_PATH)
+        
+        # 2. Đọc dữ liệu
+        try:
+            df = pd.read_excel(LOCAL_EXCEL_PATH, sheet_name=sheet_name)
+        except ValueError:
+            print(f"❌ Không tìm thấy sheet {sheet_name}")
+            return
+
+        # 3. Lọc các dòng NYS (Giả sử cột Status là cột E - index 4)
+        # Cấu trúc: A:STT, B:Desc, C:Shopee, D:DriveID, E:Status
+        # Pandas index cột: 0, 1, 2, 3, 4
+        
+        # Chuẩn hóa tên cột để dễ gọi
+        df.columns = ['STT', 'Desc', 'ShopeeLink', 'DriveVideoID', 'Status', 'NewContent'] # Thêm cột NewContent nếu cần
+        
+        nys_rows = df[df['Status'] == 'NYS']
+        
+        if nys_rows.empty:
+            print(f"⚠️ Sheet {sheet_name} đã hết bài (không còn NYS).")
+            return
+
+        # 4. Chọn ngẫu nhiên
+        selected_row_index = random.choice(nys_rows.index)
+        row_data = df.loc[selected_row_index]
+        
+        print(f"--- Đã chọn dòng {selected_row_index + 2} ---")
+
+    # (Nhả lock Excel ra để xử lý nặng)
+
+    try:
+        # 5. Gemini viết lại nội dung
+        original_desc = str(row_data['Desc'])
+        new_content = await google_utils.generate_content_with_gemini(original_desc)
+        
+        # 6. Tải video
+        video_drive_id = str(row_data['DriveVideoID'])
+        if not os.path.exists(TEMP_VIDEO_DIR): os.makedirs(TEMP_VIDEO_DIR)
+        local_video_path = os.path.join(TEMP_VIDEO_DIR, f"{sheet_name}_{video_drive_id}.mp4")
+        
+        await google_utils.download_video_by_id(drive_service, video_drive_id, local_video_path)
+        
+        # 7. Upload Facebook
+        video_id = await facebook_utils.upload_reel(
+            page_config['page_id'],
+            page_config['access_token'],
+            local_video_path,
+            new_content
+        )
+        print(f"✅ Upload thành công. ID: {video_id}")
+        
+        # 8. Comment
+        cta = "🔥 Săn ngay tại đây cả nhà ơi: "
+        comment_msg = f"{cta} {row_data['ShopeeLink']}"
+        await facebook_utils.comment_on_video(video_id, page_config['access_token'], comment_msg)
+        
+        # 9. Cập nhật Excel (Cần Lock lại)
+        async with excel_lock:
+            # Đọc lại file để đảm bảo không đè dữ liệu của luồng khác
+            # (Trong thực tế nên dùng Google Sheets API update cell trực tiếp sẽ tốt hơn download/upload)
+            # Nhưng ở đây ta làm theo logic file Excel local
+            df_update = pd.read_excel(LOCAL_EXCEL_PATH, sheet_name=None) # Đọc tất cả sheet
+            
+            # Cập nhật sheet hiện tại
+            sheet_df = df_update[sheet_name]
+            sheet_df.at[selected_row_index, 'Status'] = 'Uploaded'
+            # Giả sử cột F là nội dung mới
+            if len(sheet_df.columns) > 5:
+                sheet_df.iloc[selected_row_index, 5] = new_content 
+            
+            # Lưu và Upload
+            with pd.ExcelWriter(LOCAL_EXCEL_PATH, engine='openpyxl') as writer:
+                for s_name, s_df in df_update.items():
+                    s_df.to_excel(writer, sheet_name=s_name, index=False)
+            
+            await google_utils.upload_file_to_drive(drive_service, LOCAL_EXCEL_PATH, DRIVE_EXCEL_FILE_ID)
+            print(f"✅ Đã cập nhật Excel lên Drive.")
+
+        # Dọn dẹp
+        if os.path.exists(local_video_path): os.remove(local_video_path)
+
+    except Exception as e:
+        print(f"❌ Lỗi xử lý Page {sheet_name}: {e}")