Update app/data/mock_profiles.py

app/data/mock_profiles.py

@@ -1,250 +1,204 @@
 """
-Authentication router for user login and registration.
-Handles Firebase Authentication integration.
+Mock user profiles for automatic assignment on signup.
+These profiles contain realistic financial data for loan processing demo.
 """
 
-from datetime import datetime
-from typing import Optional
-
-from app.config import settings
-from app.schemas import (
-    ErrorResponse,
-    LoginRequest,
-    LoginResponse,
-    MessageResponse,
-    RegisterRequest,
-)
-from app.services.firebase_service import firebase_service
-from app.utils.logger import default_logger as logger
-from fastapi import APIRouter, Header, HTTPException, status
+import random
+from typing import Any, Dict, List
+
+# Profile 1: Young Professional (Good Credit, High Approval Chance)
+PROFILE_YOUNG_PROFESSIONAL = {
+    "monthly_income": 75000.0,
+    "existing_emi": 8000.0,
+    "mock_credit_score": 750,
+    "segment": "Salaried",
+    "employment_type": "Salaried",
+    "employment_years": 3,
+    "company_category": "Category A",
+    "pan_number": "ABCDE1234F",
+    "aadhar_number": "1234-5678-9012",
+    "address": "123, Tech Park, Bangalore, Karnataka - 560001",
+    "phone": "+91-9876543210",
+    "date_of_birth": "1995-06-15",
+    "gender": "Male",
+    "marital_status": "Single",
+    "current_loan_outstanding": 200000.0,
+    "bank_account_number": "1234567890",
+    "bank_name": "HDFC Bank",
+    "bank_ifsc": "HDFC0001234",
+    # KYC Documents (mocked)
+    "kyc_verified": True,
+    "kyc_status": "VERIFIED",
+    "cibil_score": 750,
+    "cibil_last_updated": "2024-11-15",
+    # Loan eligibility metrics
+    "max_eligible_amount": 500000.0,
+    "risk_category": "Low Risk",
+    "profile_completeness": 100,
+    "description": "Young professional with good credit score and stable income. High loan approval probability.",
+}
+
+# Profile 2: Mid-Career Professional (Average Credit, Moderate EMI)
+PROFILE_MID_CAREER = {
+    "monthly_income": 50000.0,
+    "existing_emi": 12000.0,
+    "mock_credit_score": 680,
+    "segment": "Salaried",
+    "employment_type": "Salaried",
+    "employment_years": 7,
+    "company_category": "Category B",
+    "pan_number": "FGHIJ5678K",
+    "aadhar_number": "9876-5432-1098",
+    "address": "456, Green Avenue, Pune, Maharashtra - 411001",
+    "phone": "+91-9876543211",
+    "date_of_birth": "1990-03-22",
+    "gender": "Female",
+    "marital_status": "Married",
+    "current_loan_outstanding": 350000.0,
+    "bank_account_number": "2345678901",
+    "bank_name": "ICICI Bank",
+    "bank_ifsc": "ICIC0002345",
+    # KYC Documents (mocked)
+    "kyc_verified": True,
+    "kyc_status": "VERIFIED",
+    "cibil_score": 680,
+    "cibil_last_updated": "2024-10-20",
+    # Loan eligibility metrics
+    "max_eligible_amount": 300000.0,
+    "risk_category": "Medium Risk",
+    "profile_completeness": 95,
+    "description": "Mid-career professional with moderate existing EMI. May need loan amount adjustment.",
+}
+
+# Profile 3: Entry-Level Professional (Lower Credit, New to Credit)
+PROFILE_ENTRY_LEVEL = {
+    "monthly_income": 35000.0,
+    "existing_emi": 3000.0,
+    "mock_credit_score": 650,
+    "segment": "New to Credit",
+    "employment_type": "Salaried",
+    "employment_years": 1,
+    "company_category": "Category B",
+    "pan_number": "KLMNO9012P",
+    "aadhar_number": "5555-6666-7777",
+    "address": "789, Lake View, Hyderabad, Telangana - 500001",
+    "phone": "+91-9876543212",
+    "date_of_birth": "1998-09-10",
+    "gender": "Male",
+    "marital_status": "Single",
+    "current_loan_outstanding": 50000.0,
+    "bank_account_number": "3456789012",
+    "bank_name": "SBI",
+    "bank_ifsc": "SBIN0003456",
+    # KYC Documents (mocked)
+    "kyc_verified": True,
+    "kyc_status": "VERIFIED",
+    "cibil_score": 650,
+    "cibil_last_updated": "2024-11-01",
+    # Loan eligibility metrics
+    "max_eligible_amount": 200000.0,
+    "risk_category": "High Risk",
+    "profile_completeness": 90,
+    "description": "Entry-level professional with limited credit history. Eligible for smaller loan amounts.",
+}
+
+# List of all profiles for random selection
+MOCK_PROFILES: List[Dict[str, Any]] = [
+    PROFILE_YOUNG_PROFESSIONAL,
+    PROFILE_MID_CAREER,
+    PROFILE_ENTRY_LEVEL,
+]
+
+
+def get_random_mock_profile() -> Dict[str, Any]:
+    """
+    Get a random mock profile from the available profiles.
 
-router = APIRouter()
+    Returns:
+        Dictionary with mock user financial data
+    """
+    return random.choice(MOCK_PROFILES).copy()
 
 
-@router.post("/login", response_model=LoginResponse)
-async def login(request: LoginRequest):
+def get_profile_by_index(index: int) -> Dict[str, Any]:
     """
-    Login endpoint for user authentication.
-
-    For hackathon: Simple email/password validation.
-    In production: This would validate Firebase ID tokens.
+    Get a specific mock profile by index.
 
     Args:
-        request: Login request with email and password
+        index: Profile index (0-2)
 
     Returns:
-        LoginResponse with access token and user info
+        Dictionary with mock user financial data
     """
-    try:
-        logger.info(f"Login attempt for email: {request.email}")
-
-        # For hackathon: Simple validation
-        if len(request.password) < 6:
-            raise HTTPException(
-                status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid credentials"
-            )
-
-        # Try to find user by email in Firestore
-        # In production, Firebase Auth would handle this
-        user_profile = None
-
-        # For now, we'll check if user exists or create a mock response
-        # In production, you'd verify Firebase ID token here
-
-        # Generate a mock user_id from email (for demo)
-        user_id = request.email.split("@")[0].replace(".", "_")
-
-        # Try to fetch existing user
-        user_profile = firebase_service.get_user_profile(user_id)
-
-        if not user_profile:
-            # For demo: Create a profile with realistic mock data
-            logger.info(f"Creating new user profile with mock data for {request.email}")
-
-            # Assign random mock profile with realistic financial data
-            basic_user_data = {
-                "user_id": user_id,
-                "email": request.email,
-                "full_name": request.email.split("@")[0].title(),
-            }
-            user_profile = assign_mock_profile_to_user(basic_user_data)
-
-            firebase_service.create_user_profile(user_profile)
-            logger.info(
-                f"Assigned mock profile: Credit Score={user_profile.get('mock_credit_score')}, Income=₹{user_profile.get('monthly_income')}"
-            )
-
-        # Generate access token (in production, use proper JWT)
-        access_token = f"finagent_token_{user_id}_{datetime.utcnow().timestamp()}"
-
-        response = LoginResponse(
-            access_token=access_token,
-            token_type="Bearer",
-            user_id=user_profile.get("user_id", user_id),
-            full_name=user_profile.get("full_name", "User"),
-            email=user_profile.get("email", request.email),
-        )
-
-        logger.info(f"Login successful for user: {user_id}")
-        return response
-
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Login error: {str(e)}")
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="Login failed"
-        )
-
-
-@router.post("/register", response_model=LoginResponse)
-async def register(request: RegisterRequest):
-    """
-    Register a new user.
+    if 0 <= index < len(MOCK_PROFILES):
+        return MOCK_PROFILES[index].copy()
+    return MOCK_PROFILES[0].copy()
 
-    Args:
-        request: Registration request with user details
 
-    Returns:
-        LoginResponse with access token and user info
-    """
-    try:
-        logger.info(f"Registration attempt for email: {request.email}")
-
-        # Generate user_id from email
-        user_id = request.email.split("@")[0].replace(".", "_")
-
-        # Check if user already exists
-        existing_user = firebase_service.get_user_profile(user_id)
-        if existing_user:
-            raise HTTPException(
-                status_code=status.HTTP_400_BAD_REQUEST, detail="User already exists"
-            )
-
-        # Create user profile with mock financial data
-        basic_user_data = {
-            "user_id": user_id,
-            "email": request.email,
-            "full_name": request.full_name,
-            "monthly_income": request.monthly_income
-            if hasattr(request, "monthly_income")
-            else None,
-            "existing_emi": request.existing_emi
-            if hasattr(request, "existing_emi")
-            else None,
-        }
-
-        # Assign mock profile with realistic data (KYC, CIBIL, etc.)
-        user_profile = assign_mock_profile_to_user(basic_user_data)
-
-        logger.info(
-            f"Assigning mock profile to new user: Credit Score={user_profile.get('mock_credit_score')}, Income=₹{user_profile.get('monthly_income')}"
-        )
-
-        created_profile = firebase_service.create_user_profile(user_profile)
-
-        # Generate access token
-        access_token = f"finagent_token_{user_id}_{datetime.utcnow().timestamp()}"
-
-        response = LoginResponse(
-            access_token=access_token,
-            token_type="Bearer",
-            user_id=created_profile.get("user_id", user_id),
-            full_name=created_profile.get("full_name"),
-            email=created_profile.get("email"),
-        )
-
-        logger.info(f"Registration successful for user: {user_id}")
-        return response
-
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Registration error: {str(e)}")
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail="Registration failed",
-        )
-
-
-@router.post("/verify-token")
-async def verify_token(authorization: Optional[str] = Header(None)):
+def get_all_profiles() -> List[Dict[str, Any]]:
     """
-    Verify Firebase ID token.
-
-    Args:
-        authorization: Authorization header with Bearer token
+    Get all available mock profiles.
 
     Returns:
-        Token verification result
+        List of all mock profile dictionaries
     """
-    try:
-        if not authorization:
-            raise HTTPException(
-                status_code=status.HTTP_401_UNAUTHORIZED,
-                detail="Authorization header missing",
-            )
-
-        # Extract token from Bearer scheme
-        parts = authorization.split()
-        if len(parts) != 2 or parts[0].lower() != "bearer":
-            raise HTTPException(
-                status_code=status.HTTP_401_UNAUTHORIZED,
-                detail="Invalid authorization header format",
-            )
-
-        token = parts[1]
-
-        # For hackathon: Simple token validation
-        if token.startswith("finagent_token_"):
-            # Extract user_id from token
-            parts = token.split("_")
-            if len(parts) >= 3:
-                user_id = "_".join(parts[2:-1])
-                return MessageResponse(message="Token valid", success=True)
-
-        # In production: Verify Firebase ID token
-        # decoded_token = firebase_service.verify_token(token)
-        # if decoded_token:
-        #     return MessageResponse(message="Token valid", success=True)
-
-        raise HTTPException(
-            status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid or expired token"
-        )
-
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Token verification error: {str(e)}")
-        raise HTTPException(
-            status_code=status.HTTP_401_UNAUTHORIZED, detail="Token verification failed"
-        )
-
-
-@router.post("/logout")
-async def logout(authorization: Optional[str] = Header(None)):
-    """
-    Logout endpoint.
+    return [profile.copy() for profile in MOCK_PROFILES]
+
 
-    For stateless JWT tokens, this is mainly for client-side cleanup.
-    In production with Firebase, you might want to revoke refresh tokens.
+def assign_mock_profile_to_user(user_data: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Assign a random mock profile to a new user, preserving their basic info.
 
     Args:
-        authorization: Authorization header with Bearer token
+        user_data: User's basic info (user_id, email, full_name)
 
     Returns:
-        Success message
+        Complete user profile with mock financial data
     """
-    try:
-        logger.info("Logout request received")
-
-        # In production: Could revoke Firebase refresh tokens here
-        # or add token to blacklist
-
-        return MessageResponse(message="Logged out successfully", success=True)
-
-    except Exception as e:
-        logger.error(f"Logout error: {str(e)}")
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="Logout failed"
-        )
+    # Get random profile
+    mock_profile = get_random_mock_profile()
+
+    # Merge user's actual data with mock profile
+    complete_profile = {
+        **mock_profile,  # Start with mock financial data
+        "user_id": user_data.get("user_id"),
+        "email": user_data.get("email"),
+        "full_name": user_data.get("full_name"),
+    }
+
+    # If user provided any financial data during signup, use that instead
+    if "monthly_income" in user_data and user_data["monthly_income"]:
+        complete_profile["monthly_income"] = user_data["monthly_income"]
+    if "existing_emi" in user_data and user_data["existing_emi"]:
+        complete_profile["existing_emi"] = user_data["existing_emi"]
+
+    return complete_profile
+
+
+# Profile descriptions for admin/display purposes
+PROFILE_DESCRIPTIONS = {
+    "YOUNG_PROFESSIONAL": {
+        "name": "Young Professional",
+        "income_range": "₹70,000 - ₹80,000",
+        "credit_score_range": "740-760",
+        "approval_rate": "95%",
+        "max_loan": "₹5,00,000",
+        "typical_decision": "APPROVED",
+    },
+    "MID_CAREER": {
+        "name": "Mid-Career Professional",
+        "income_range": "₹45,000 - ₹55,000",
+        "credit_score_range": "670-690",
+        "approval_rate": "75%",
+        "max_loan": "₹3,00,000",
+        "typical_decision": "APPROVED or ADJUST",
+    },
+    "ENTRY_LEVEL": {
+        "name": "Entry-Level Professional",
+        "income_range": "₹30,000 - ₹40,000",
+        "credit_score_range": "640-660",
+        "approval_rate": "60%",
+        "max_loan": "₹2,00,000",
+        "typical_decision": "APPROVED (smaller amounts) or ADJUST",
+    },
+}