src/utils.py

import re
from datetime import datetime


def clean_model_name(model_id: str) -> str:
    if "/" in model_id:
        model_id = model_id.split("/")[-1]
    if ":free" in model_id:
        model_id = model_id.replace(":free", "")
    return model_id.strip()


def format_timestamp(dt: datetime = None) -> str:
    if dt is None:
        dt = datetime.now()
    return dt.strftime("%Y-%m-%d %H:%M:%S")