utils_logging.py

import pymongo
import streamlit as st
import datetime


def make_log(session_state):
    log = {
        "user_token": session_state.user_token,
        "date": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
        "id": session_state.id,
        "job_ad": session_state.job_ad,
        "metrics": session_state.metrics,
        "indices": session_state.indices,
        "job_ad_split": session_state.job_ad_split,
        "ai": {
            "params": session_state.openai_params,
            # "openai_response": [
            #     choice.message.content for choice in session_state.response
            # ],
            # "cleaned_response": session_state.cleaned_response,
            "usage": session_state.usage,
        },
    }
    return log


def send_log(log):
    try:
        client = pymongo.MongoClient(st.secrets["mongo_login"])
        col = client["Fairplay"]["logging"]
        col.insert_one(log)
    except Exception as e:
        st.toast(f"Had an issue contacting our server... {e}")