server.py

from dotenv import load_dotenv
load_dotenv()

from fastapi import FastAPI, HTTPException, Body
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel, EmailStr
from typing import Optional
import os, logging, uuid, re
from datetime import datetime, timedelta

# LangChain & Agent
from langchain_groq import ChatGroq
from langchain.agents import AgentExecutor, create_tool_calling_agent
from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain.memory import ConversationBufferMemory

# Database
from db import (
    get_connection,
    add_potential_customer,
    save_consultation_to_db,
    init_tables,
)

# Tools & Prompt
from promtt import custom_prompt
from tools import (
    get_company_info,
    classify_customer,
    save_consultation_schedule_tool,
)

# ==============================
# LOGGING & APP INIT
# ==============================
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
logger = logging.getLogger(__name__)

app = FastAPI(title="AT Shop API", version="1.0")

# ==============================
# CORS
# ==============================
origins = ["http://127.0.0.1:5500", "http://localhost:5500"]
app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# ==============================
# INIT DATABASE
# ==============================
init_tables()

# ==============================
# SESSION CACHE
# ==============================
session_cache = {}

# ==============================
# LLM CONFIG
# ==============================
api_key = os.getenv("API_KEY")
model = os.getenv("MODEL")
if not api_key or not model:
    raise ValueError("❌ Thiếu API_KEY hoặc MODEL trong .env")

llm = ChatGroq(api_key=api_key, model=model)
prompt = ChatPromptTemplate.from_messages([
    ("system", custom_prompt),
    MessagesPlaceholder(variable_name="chat_history"),
    ("human", "{input}"),
    MessagesPlaceholder(variable_name="agent_scratchpad"),
])

memory = ConversationBufferMemory(k=20, memory_key="chat_history", return_messages=True)
tools = [get_company_info, classify_customer, save_consultation_schedule_tool]
agent = create_tool_calling_agent(llm=llm, prompt=prompt, tools=tools)
agent_executor = AgentExecutor(agent=agent, tools=tools, memory=memory, handle_parsing_errors=True)

# ==============================
# MODELS
# ==============================
class Question(BaseModel):
    question: str

class PotentialCustomer(BaseModel):
    name: str
    phone: str
    email: Optional[EmailStr] = None
    address: Optional[str] = None

# ==============================
# HELPERS
# ==============================
VN_PHONE = re.compile(r"^0\d{9,10}$")

def is_valid_name(name: Optional[str]) -> bool:
    if not name:
        return False
    n = name.strip().lower()
    return len(n) >= 2 and n not in {"ban", "bạn", "toi", "tôi", "minh", "mình"}

def is_valid_phone(phone: Optional[str]) -> bool:
    return bool(phone and VN_PHONE.match(phone))

def non_empty(s: Optional[str]) -> bool:
    return bool(s and s.strip())

def parse_customer_info(text: str):
    """Tách số điện thoại/email khi người dùng nhập thẳng trong câu chat."""
    text = text.strip()
    phone_match = re.search(r"\b0\d{9,10}\b", text)
    email_match = re.search(r'[\w\.-]+@[\w\.-]+', text)
    parts = [p.strip() for p in text.split(',') if p.strip()]
    name = parts[0] if len(parts) > 0 else None
    address = parts[-1] if len(parts) > 1 else None
    return {"name": name, "phone": phone_match.group(0) if phone_match else None, "email": email_match.group(0) if email_match else None, "address": address}

# ==============================
# 🕒 Xử lý cụm thời gian tiếng Việt tự nhiên
# ==============================
def parse_vietnamese_time_phrase(raw_time: str) -> str:
    raw_time = raw_time.strip().lower()
    now = datetime.now()

    # Nếu có "mai" → cộng 1 ngày
    if "mai" in raw_time:
        now = now + timedelta(days=1)

    # Mặc định buổi sáng
    hour = 9
    if "chiều" in raw_time:
        hour = 15
    elif "tối" in raw_time:
        hour = 19
    elif "trưa" in raw_time:
        hour = 12

    # Nếu có dạng số giờ
    match_h = re.search(r"(\d{1,2})h", raw_time)
    if match_h:
        hour = int(match_h.group(1))

    return now.replace(hour=hour, minute=0, second=0, microsecond=0).strftime("%Y-%m-%d %H:%M:%S")

# ==============================
# CHAT ENDPOINT
# ==============================
@app.post("/chat")
async def chat_endpoint(q: Question):
    user_input = q.question.strip()
    if not user_input:
        raise HTTPException(status_code=400, detail="Câu hỏi không được để trống.")

    # Gọi Agent
    customer_type = classify_customer.run(user_input)
    response = agent_executor.invoke({"input": user_input})
    final_output = response.get("output", "").strip()

    # 🧹 Ẩn tool call nội bộ
    if "save_consultation_schedule_tool" in final_output or "function_call" in final_output:
        final_output = re.split(r"save_consultation_schedule_tool|function_call", final_output, 1)[0].strip()
    final_output = re.sub(
        r"(save_consultation_schedule_tool|create_order|classify_customer).*",
        "",
        final_output,
        flags=re.DOTALL
    ).strip()

    # Phân tích dữ liệu khách hàng
    info = parse_customer_info(user_input)
    name = info.get("name")
    phone = info.get("phone")
    email = info.get("email")
    address = info.get("address")

    has_schedule = bool(re.search(r"\b(hẹn|lịch|tư\s*vấn|gọi\s*lại)\b", user_input, re.I))
    has_time = bool(re.search(r"\d{1,2}[:h]\d{0,2}", user_input))

    try:
        consultation_time = None

        # Parse thời gian cụ thể (vd: "10h 21/10/2025")
        match_full = re.search(r"(\d{1,2})h\s*(\d{1,2})/(\d{1,2})/(\d{4})", user_input)
        if match_full:
            hour, day, month, year = match_full.groups()
            consultation_time = f"{year}-{month.zfill(2)}-{day.zfill(2)} {hour.zfill(2)}:00:00"
        else:
            match_hour = re.search(r"(\d{1,2})[:h](\d{0,2})", user_input)
            if match_hour:
                hour, minute = match_hour.groups()
                minute = minute if minute else "00"
                consultation_time = f"{datetime.now().strftime('%Y-%m-%d')} {hour.zfill(2)}:{minute.zfill(2)}:00"

        # Nếu là cụm 'sáng mai', 'chiều nay' ...
        if not consultation_time and re.search(r"(sáng|chiều|tối|mai|nay)", user_input, re.I):
            consultation_time = parse_vietnamese_time_phrase(user_input)

        # Nếu có thông tin hợp lệ thì lưu
        if phone and name and (has_schedule or has_time):
            save_consultation_to_db(
                name=name,
                phone=phone,
                email=email or "",
                consultation_time=consultation_time or datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                note="AI tự động ghi nhận từ hội thoại",
                address=address or ""
            )
            logger.info(f"✅ [AI] Lưu lịch tư vấn: {name} - {phone} ({consultation_time})")

        elif name and phone and address:
            add_potential_customer(name, phone, email, address)
            logger.info(f"✅ [AI] Lưu khách hàng tiềm năng: {name} - {phone}")

    except Exception as err:
        logger.warning(f"⚠️ Không thể tự động lưu thông tin: {err}")

    return {"responses": [final_output], "customer_type": customer_type}

# ==============================
# KHÁCH HÀNG TIỀM NĂNG
# ==============================
@app.get("/potential_customer")
async def get_potential_customers():
    conn = get_connection()
    if not conn:
        raise HTTPException(status_code=500, detail="Không thể kết nối DB.")
    cursor = conn.cursor()
    cursor.execute("""
        SELECT id, name, phone, email, address, created_at
        FROM potential_customers ORDER BY created_at DESC
    """)
    rows = cursor.fetchall()
    cursor.close()
    conn.close()

    customers = [
        {"id": r[0], "name": r[1], "phone": r[2], "email": r[3], "address": r[4], "created_at": r[5]} for r in rows
    ]
    return {"customers": customers}

@app.delete("/potential_customer/{id}")
async def delete_potential_customer(id: int):
    conn = get_connection()
    if not conn:
        raise HTTPException(status_code=500, detail="Không thể kết nối DB.")
    cursor = conn.cursor()
    cursor.execute("DELETE FROM potential_customers WHERE id = %s", (id,))
    conn.commit()
    cursor.close()
    conn.close()
    return {"success": True, "message": "Đã xóa khách hàng thành công!"}

# ==============================
# LỊCH TƯ VẤN
# ==============================
@app.get("/consultations")
async def get_consultations():
    conn = get_connection()
    if not conn:
        raise HTTPException(status_code=500, detail="Không thể kết nối DB.")
    cursor = conn.cursor()
    cursor.execute("""
        SELECT id, name, email, phone, consultation_time, created_at
        FROM consultations ORDER BY created_at DESC
    """)
    rows = cursor.fetchall()
    cursor.close()
    conn.close()

    consultations = []
    for row in rows:
        consultation_time = row[4]
        created_at = row[5]
        if isinstance(consultation_time, datetime):
            consultation_time = consultation_time.strftime("%Y-%m-%d %H:%M")
        if isinstance(created_at, datetime):
            created_at = created_at.strftime("%Y-%m-%d %H:%M:%S")
        consultations.append({
            "id": row[0],
            "name": row[1],
            "email": row[2],
            "phone": row[3],
            "consultation_time": consultation_time,
            "created_at": created_at
        })
    return {"consultations": consultations}

# ==============================
# HEALTH CHECK
# ==============================
@app.get("/health")
async def health_check():
    return {"status": "ok"}