main.py

import dill
import pandas as pd
from fastapi import FastAPI
from pydantic import BaseModel

with open('model/model.pkl', 'rb') as file:
    model = dill.load(file)

app = FastAPI()


class Form(BaseModel):
    session_id: str
    client_id: str
    visit_date: str
    visit_time: str
    visit_nuber: int
    utm_source: str
    utm_medium: str
    utm_campaign: str
    utm_adcontent: str
    utm_keyword: str
    device_category: str
    device_os: str
    device_brand: str
    device_model: str
    device_screen_resolution: str
    device_browser: str
    geo_country: str
    geo_city: str


class Predict(BaseModel):
    session_id: str
    CR: int


@app.get('/status')
def status():
    return {'status': 'ok'}


@app.get('/version')
def version():
    return model['metadata']


@app.post('/predict', response_model=Predict)
def predict(form: Form):
    df = pd.DataFrame.from_dict([form.model_dump()])
    y = model['model'].predict(df)

    return {
        'session_id': form.session_id,
        'CR': y[0]
    }