AI Financial Chatbot

# From app/ai_service.py:101-104
response_stream = self.client.models.generate_content_stream(
    model=self.model_name,
    contents=prompt
)

# From app/routes/ai.py:80-83
wallets = Wallet.query.filter_by(user_id=user_id).all()
context_text = "--- TỔNG QUAN TÀI CHÍNH ---\nSỐ DƯ CÁC VÍ:\n"
for w in wallets: 
    context_text += f"- {w.name}: {int(w.balance):,} VND\n"

SỐ DƯ CÁC VÍ:
- Tiền mặt: 5,000,000 VND
- Techcombank: 25,000,000 VND
- Momo: 1,500,000 VND

# From app/routes/ai.py:85-99
expense_summary = db.session.query(
    Category.name, func.sum(Transaction.amount)
).join(Transaction, Transaction.category_id == Category.id).filter(
    Transaction.user_id == user_id, 
    Transaction.date >= last_30_days,
    Transaction.type == 'chi'
).group_by(Category.name).all()

THỐNG KÊ CHI TIÊU 30 NGÀY QUA:
- Ăn uống: 3,500,000đ
- Xăng xe: 1,200,000đ
- Giải trí: 800,000đ

# From app/routes/ai.py:101-107
recent_transactions = Transaction.query.filter_by(user_id=user_id)
    .order_by(Transaction.date.desc()).limit(5).all()

for t in recent_transactions:
    cat_name = t.category.name if t.category else "Khác"
    type_label = "CHI" if t.type == 'chi' else "THU" if t.type == 'thu' else "CHUYỂN KHOẢN"
    context_text += f"- {t.date.strftime('%d/%m')}: {int(t.amount):,}đ ({type_label}) - {cat_name} ({t.description})\n"

5 GIAO DỊCH GẦN NHẤT:
- 07/03: 85,000đ (CHI) - Ăn uống (Cafe sáng với team)
- 06/03: 350,000đ (CHI) - Xăng xe (Đổ xăng Petrolimex)
- 05/03: 2,000,000đ (CHUYỂN KHOẢN) - (Gửi tiết kiệm)

# From app/ai_service.py:76-98
prompt = f"""
Bạn là FinAI – một trợ lý tài chính cá nhân thông minh, thân thiện, nói chuyện tự nhiên như một người bạn hiểu về tiền bạc.

Dữ liệu tài chính gần đây của người dùng:
{context_data}

Câu hỏi: "{user_question}"

Cách trả lời:
- Trả lời kết quả chính ngay câu đầu tiên (ngắn gọn, rõ ràng).
- Viết như đang nói chuyện thật, KHÔNG mang văn phong báo cáo.
- Không dùng các câu mở đầu kiểu: "Dựa trên dữ liệu...", "Theo thông tin tôi có..."
- Nếu có lời khuyên hoặc cảnh báo, hãy xuống dòng 1 lần trước khi viết.
- Dùng emoji nhẹ nhàng (1–2 cái là đủ).
- Giữ câu trả lời trong 3–6 câu, tránh lan man.

Nếu thiếu dữ liệu, hỏi lại một cách tự nhiên như đang trò chuyện.

QUY TẮC KẾ TOÁN (BẮT BUỘC):
- "CHI TIÊU" = tiền ra khỏi hệ thống.
- "CHUYỂN KHOẢN" = tiền chuyển giữa ví của bạn → KHÔNG phải chi tiêu.
- Khi tính tổng chi tiêu: TUYỆT ĐỐI không cộng chuyển khoản.
"""

# From app/routes/ai.py:117-141
@stream_with_context
def generate():
    full_answer = ""
    try:
        for chunk in response_stream:
            if chunk.text:
                full_answer += chunk.text
                yield chunk.text  # Send immediately to frontend
        
        # Save to database only after complete response
        log_id = str(uuid.uuid4())[:8]
        db.session.add(ChatbotLog(
            id=log_id, 
            user_id=user_id, 
            question=user_question, 
            answer=full_answer, 
            created_at=datetime.now()
        ))
        db.session.commit()
    except Exception as e:
        print(f"Lỗi Stream/Database: {e}")
        db.session.rollback()

return Response(generate(), mimetype='text/plain')

# From app/models.py:152-158
class ChatbotLog(db.Model):
    __tablename__ = 'chatbot_lichsu'
    id = db.Column('MaHoiThoai', db.String(8), primary_key=True)
    user_id = db.Column('MaNguoiDung', db.String(8))
    question = db.Column('NoiDungHoi', db.Text)
    answer = db.Column('NoiDungTraLoi', db.Text)
    created_at = db.Column('NgayTao', db.DateTime, default=datetime.now)

GET /api/chat/history

[
  {
    "role": "user",
    "content": "Tháng này tôi chi tiêu bao nhiêu?"
  },
  {
    "role": "ai",
    "content": "Tháng này bạn chi tiêu tổng cộng 5,500,000đ. Phần lớn tiền đi vào Ăn uống (3.5 triệu) và Xăng xe (1.2 triệu). Có vẻ tháng này bạn ăn uống nhiều hơn bình thường đấy!"
  }
]

# From app/ai_service.py:94-97
QUY TẮC KẾ TOÁN (BẮT BUỘC):
- "CHI TIÊU" = tiền ra khỏi hệ thống.
- "CHUYỂN KHOẢN" = tiền chuyển giữa ví của bạn → KHÔNG phải chi tiêu.
- Khi tính tổng chi tiêu: TUYỆT ĐỐI không cộng chuyển khoản.

POST /api/chat
Content-Type: application/json
Authentication: Required (session-based)

{
  "message": "Tôi chi tiêu nhiều nhất vào danh mục nào?"
}

{
  "response": "Xin lỗi, hệ thống AI đang bận kết nối. Vui lòng thử lại sau vài giây nhé!"
}

GET /api/chat/history
Authentication: Required (session-based)

# From app/ai_service.py:107-109
except Exception as e:
    print(f"Chat Error: {e}")
    return "Xin lỗi, hệ thống AI đang bận kết nối. Vui lòng thử lại sau vài giây nhé!"

# From app/routes/ai.py:86-92
# Uses GROUP BY to aggregate spending by category in one query
expense_summary = db.session.query(
    Category.name, func.sum(Transaction.amount)
).join(...).filter(...).group_by(Category.name).all()

# From app/routes/ai.py:171-177
user_setting = UserSetting.query.get(user_id)
if user_setting and user_setting.ai_suggestions == 0:
    return jsonify({
        'status': 'disabled',
        'message': 'Tính năng AI đang bị tắt.'
    })