models/session_model.py

from pydantic import BaseModel, Field
from typing import List, Optional, Dict


'''
input_info:
{
    query:"",
    field_inputs:
    {
        "Reddit":["Pain point analysis", "Competitor analysis"],
        "Twitter":["Competitor analysis"],
    }
}
'''

'''
session_info:
{
    "Reddit":[{"Pain point analysis":-1}, {"Competitor analysis":-1}],
    "Twitter":[{"Competitor analysis":-1}],
}
'''
class InputInfoModel(BaseModel) :
    query: str = ""
    field_inputs: dict = {}

class UserSessionModel(BaseModel):
    user_id: Optional[int] = None
    input_info: Optional[dict] = Field(InputInfoModel)
    session_info: dict
    process_info: Optional[dict] = None
    session_completed: Optional[bool] = False
    is_deleted: Optional[bool] = False

class UserAllSessionResponseModel(BaseModel):
    msg: str
    data: List[UserSessionModel]