from pydantic import BaseModel
from typing import Optional
from typing import List
import time

class NER_Request (BaseModel):
    text: str
    entities: List[str]
    
class NER_Response (BaseModel):
    success: int
    result: str
    description: Optional[str] = ""
    errorCode: Optional[int] = 0
    errorDescriptin: Optional[str] = ""

entity_labels_sample = [
    "team",
    "developer",
    "technology",
    "tool",
    "amount",
    "duration",
    "capacity",
    "company",
    "currency"
]

def entities_list_to_dict(entitiesList: List[str]):
    return {key: '' for key in entitiesList}

def set_start () -> time:
    return time.time()

def audit_elapsedtime(function: str, start: time):
    end = time.time()
    elapsedtime = end-start
    print("------------------")
    print(f"[{function}] Elapsed time: {elapsedtime}")
    print("------------------")
    return elapsedtime