Quick Start

# 1. Creates user account
new_user = User(id=new_user_id, name=fullname, email=email)
new_user.set_password(password)

# 2. Creates default settings
db.session.add(UserSetting(user_id=new_user_id))

# 3. Creates default cash wallet with 0 balance
db.session.add(Wallet(
    id=str(uuid.uuid4())[:8],
    user_id=new_user_id,
    name="Ví tiền mặt",
    type="Tiền mặt",
    balance=0
))

# 1. Creates transaction record
new_trans = Transaction(
    id=str(uuid.uuid4())[:8],
    user_id=user_id,
    wallet_id=final_wallet_id,
    category_id=data.get('category_id'),
    type='chi',  # expense
    amount=amount,
    description=data.get('description'),
    date=datetime.now(),
    ai_category_id=data.get('ai_category_id'),
    ai_confidence=data.get('ai_confidence')
)

# 2. Updates wallet balance
wallet = Wallet.query.get(final_wallet_id)
wallet.balance -= amount  # Deduct for expense

db.session.commit()

def chat_with_data(self, user_question, context_data):
    prompt = f"""
    Bạn là FinAI – một trợ lý tài chính cá nhân thông minh.

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

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

    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í → 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.
    """
    
    response_stream = self.client.models.generate_content_stream(
        model='gemini-2.5-flash',
        contents=prompt
    )
    return response_stream

# AI receives your custom category list
def predict(self, text, user_categories):
    prompt = f"""
    Danh mục hiện có: {', '.join(user_categories)}
    Giao dịch cần phân loại: "{text}"
    """