app.py

import o
import io
import json
import urllib.parse
import requests
from fastapi import FastAPI, File, UploadFile, Form, Request, BackgroundTasks
from fastapi.responses import HTMLResponse
from PIL import Image
import imagehash
from playwright.sync_api import sync_playwright

app = FastAPI()

DATABASE_DIR = "scam_database"
VOTES_FILE = "votes.json"

# ฟังก์ชันโหลดข้อมูลโหวตจากไฟล์ JSON
def load_votes():
    if os.path.exists(VOTES_FILE):
        try:
            with open(VOTES_FILE, "r", encoding="utf-8") as f:
                return json.load(f)
        except Exception:
            return {}
    return {}

# ฟังก์ชันบันทึกข้อมูลโหวตลงไฟล์ JSON
def save_votes(votes_data):
    with open(VOTES_FILE, "w", encoding="utf-8") as f:
        json.dump(votes_data, f, ensure_ascii=False, indent=2)

# ฟังก์ชันดึง Hash ของรูปทั้งหมดในฐานข้อมูล
def get_database_hashes():
    hashes = {}
    if not os.path.exists(DATABASE_DIR):
        os.makedirs(DATABASE_DIR)
        
    for filename in os.listdir(DATABASE_DIR):
        if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.webp')):
            filepath = os.path.join(DATABASE_DIR, filename)
            try:
                img = Image.open(filepath)
                h = str(imagehash.phash(img))
                hashes[filename] = h
            except Exception as e:
                print(f"Error loading {filename}: {e}")
    return hashes

# ฟังก์ชันดึง IP ของผู้ใช้
def get_client_ip(request: Request):
    forwarded_for = request.headers.get("x-forwarded-for")
    if forwarded_for:
        return forwarded_for.split(",")[0].strip()
    return request.client.host if request.client else "127.0.0.1"

# ---------------------------------------------------------
# ระบบบอทสแกนรูปภาพจาก TikTok (เพิ่มระบบกันตรวจจับบอท)
# ---------------------------------------------------------
def run_tiktok_bot_task(keyword: str):
    db_hashes = get_database_hashes()
    existing_hash_objs = [imagehash.hex_to_hash(h) for h in db_hashes.values()]
    MATCH_THRESHOLD = 12

    encoded_keyword = urllib.parse.quote(keyword)
    search_url = f"https://www.tiktok.com/search?q={encoded_keyword}"

    try:
        with sync_playwright() as p:
            # เพิ่ม args ปลอมตัวไม่ให้ TikTok บล็อกบอท
            browser = p.chromium.launch(
                headless=True,
                args=[
                    "--no-sandbox",
                    "--disable-setuid-sandbox",
                    "--disable-blink-features=AutomationControlled",
                    "--disable-dev-shm-usage"
                ]
            )
            context = browser.new_context(
                user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36",
                viewport={"width": 1280, "height": 800},
                locale="th-TH"
            )

            page = context.new_page()
            # ซ่อนสถานะ webdriver
            page.add_init_script("Object.defineProperty(navigator, 'webdriver', {get: () => undefined})")

            print(f"[Bot] กำลังเปิด TikTok ค้นหาคำว่า: '{keyword}'")
            page.goto(search_url, wait_until="domcontentloaded", timeout=45000)
            page.wait_for_timeout(5000)

            # เลื่อนหน้าจอลงหลายๆ ครั้งกระตุ้นให้คลิปโหลด
            for _ in range(4):
                page.mouse.wheel(0, 1000)
                page.wait_for_timeout(2000)

            # ค้นหาลิงก์คลิปทั้งหมดในหน้าค้นหา
            video_links = set()
            links = page.query_selector_all("a")
            for a in links:
                href = a.get_attribute("href")
                if href and ("/video/" in href or "/photo/" in href):
                    if href.startswith("/"):
                        href = "https://www.tiktok.com" + href
                    clean_url = href.split("?")[0]
                    video_links.add(clean_url)

            video_list = list(video_links)
            print(f"[Bot] เจอวิดีโอจากหน้าค้นหาทั้งหมด {len(video_list)} คลิป")

            # สแกนทีละคลิป (สูงสุด 5 คลิปแรก)
            for v_idx, v_url in enumerate(video_list[:5]):
                try:
                    print(f"[Bot] ({v_idx+1}/{min(len(video_list), 5)}) กำลังเข้าไปสแกนคอมเมนต์คลิป: {v_url}")
                    page.goto(v_url, wait_until="domcontentloaded", timeout=30000)
                    page.wait_for_timeout(4000)

                    # สกรอลล์ลงไปส่วนคอมเมนต์
                    page.mouse.wheel(0, 1000)
                    page.wait_for_timeout(2500)

                    # ดึงเฉพาะรูปภาพที่แนบมาในคอมเมนต์
                    comment_imgs = page.query_selector_all(
                        "[data-e2e='comment-list'] img, [class*='comment'] img, div[class*='Comment'] img"
                    )

                    print(f"[Bot] พบรูปในคอมเมนต์จำนวน {len(comment_imgs)} รูป")

                    for img_el in comment_imgs:
                        src = img_el.get_attribute("src")
                        if not src or not src.startswith("http"):
                            continue

                        try:
                            resp = requests.get(src, timeout=5)
                            if resp.status_code == 200:
                                img = Image.open(io.BytesIO(resp.content))

                                # กรองรูปโปรไฟล์ขนาดเล็กออก (รูปในคอมเมนต์มักจะใหญ่กว่า 120px)
                                if img.width < 120 or img.height < 120:
                                    continue

                                new_hash_obj = imagehash.phash(img)

                                is_dup = False
                                for existing_h in existing_hash_objs:
                                    if (new_hash_obj - existing_h) <= MATCH_THRESHOLD:
                                        is_dup = True
                                        break

                                if not is_dup:
                                    filename = f"bot_{str(new_hash_obj)}.jpg"
                                    filepath = os.path.join(DATABASE_DIR, filename)
                                    img.convert("RGB").save(filepath, "JPEG")
                                    existing_hash_objs.append(new_hash_obj)
                                    print(f"[Bot] ✅ บันทึกรูปใหม่จากคอมเมนต์: {filename}")
                                else:
                                    print(f"[Bot] ❌ รูปในคอมเมนต์ซ้ำแล้ว - ข้าม")
                        except Exception:
                            continue
                except Exception as ve:
                    print(f"[Bot Error] เข้าดูคลิปไม่ได้: {ve}")
                    continue

            browser.close()
            print(f"[Bot] สแกนเสร็จสิ้นสำหรับคำว่า '{keyword}'")
    except Exception as e:
        print(f"[Bot Error] {e}")

# Endpoint สำหรับสั่งให้บอทเริ่มทำงาน
@app.post("/bot/run")
async def trigger_bot(
    keyword: str = Form(...), 
    background_tasks: BackgroundTasks = BackgroundTasks()
):
    background_tasks.add_task(run_tiktok_bot_task, keyword)
    return {
        "success": True, 
        "message": f"บอทเริ่มค้นหาคำว่า '{keyword}' และสแกนรูปในคอมเมนต์เรียบร้อยแล้ว"
    }

# ---------------------------------------------------------
# Endpoint เดิม (100% ตามโค้ดเดิม)
# ---------------------------------------------------------
@app.get("/", response_class=HTMLResponse)
def serve_index():
    if os.path.exists("index.html"):
        with open("index.html", "r", encoding="utf-8") as f:
            return f.read()
    return "<h1>index.html not found</h1>"

@app.post("/check")
async def check_image(file: UploadFile = File(...)):
    try:
        contents = await file.read()
        uploaded_img = Image.open(io.BytesIO(contents))
        uploaded_hash_obj = imagehash.phash(uploaded_img)
        uploaded_hash_str = str(uploaded_hash_obj)

        db_hashes = get_database_hashes()
        votes_data = load_votes()

        MATCH_THRESHOLD = 12
        matched_hash_key = None
        is_in_db = False

        for filename, db_hash_str in db_hashes.items():
            db_hash_obj = imagehash.hex_to_hash(db_hash_str)
            distance = uploaded_hash_obj - db_hash_obj
            if distance <= MATCH_THRESHOLD:
                matched_hash_key = db_hash_str
                is_in_db = True
                break

        if not matched_hash_key:
            matched_hash_key = uploaded_hash_str

        image_vote = votes_data.get(matched_hash_key, {"likes": 0, "dislikes": 0, "users": {}})
        likes = image_vote.get("likes", 0)
        dislikes = image_vote.get("dislikes", 0)

        if likes >= 100:
            is_scam = False
            result_text = "ปลอดภัย"
            detail_text = f"รูปภาพนี้ได้รับการยืนยันว่าปลอดภัยจากผู้ใช้งาน (กดไลก์ {likes} คน)"
        elif is_in_db:
            is_scam = True
            result_text = "มิจแน่นอน100%"
            detail_text = "พบรูปภาพตรงกับฐานข้อมูลรูปภาพโปรโมทหลอกลวง"
        else:
            is_scam = False
            result_text = "ไม่สามารถตรวจสอบได้"
            detail_text = "ยังไม่พบรูปภาพนี้ในฐานข้อมูลมิจฉาชีพ"

        return {
            "is_scam": is_scam,
            "result": result_text,
            "detail": detail_text,
            "image_hash": matched_hash_key,
            "likes": likes,
            "dislikes": dislikes
        }
    except Exception as e:
        return {"error": f"เกิดข้อผิดพลาดในการประมวลผล: {str(e)}"}

@app.post("/vote")
async def vote_image(
    request: Request,
    image_hash: str = Form(...),
    vote_type: str = Form(...),
    user_id: str = Form(None)
):
    try:
        votes_data = load_votes()
        if image_hash not in votes_data:
            votes_data[image_hash] = {"likes": 0, "dislikes": 0, "users": {}}
        elif "users" not in votes_data[image_hash]:
            votes_data[image_hash]["users"] = {}

        client_ip = get_client_ip(request)
        user_key = f"{client_ip}_{user_id}" if user_id else client_ip

        if user_key in votes_data[image_hash]["users"]:
            return {
                "success": False,
                "error": "คุณเคยโหวตรูปภาพนี้ไปแล้ว สามารถโหวตได้เพียง 1 ครั้งเท่านั้น"
            }

        votes_data[image_hash]["users"][user_key] = vote_type

        if vote_type == "like":
            votes_data[image_hash]["likes"] += 1
        elif vote_type == "dislike":
            votes_data[image_hash]["dislikes"] += 1

        save_votes(votes_data)

        likes = votes_data[image_hash]["likes"]
        dislikes = votes_data[image_hash]["dislikes"]

        db_hashes = get_database_hashes()
        is_in_db = image_hash in db_hashes.values()

        if likes >= 100:
            is_scam = False
            result_text = "ปลอดภัย"
            detail_text = f"รูปภาพนี้ได้รับการยืนยันว่าปลอดภัยจากผู้ใช้งาน (กดไลก์ {likes} คน)"
        elif is_in_db:
            is_scam = True
            result_text = "มิจแน่นอน100%"
            detail_text = "พบรูปภาพตรงกับฐานข้อมูลรูปภาพโปรโมทหลอกลวง"
        else:
            is_scam = False
            result_text = "ไม่สามารถตรวจสอบได้"
            detail_text = "ยังไม่พบรูปภาพนี้ในฐานข้อมูลมิจฉาชีพ"

        return {
            "success": True,
            "likes": likes,
            "dislikes": dislikes,
            "is_scam": is_scam,
            "result": result_text,
            "detail": detail_text
        }
    except Exception as e:
        return {"error": f"เกิดข้อผิดพลาดในการโหวต: {str(e)}"}