Spaces:
Sleeping
Sleeping
File size: 2,024 Bytes
8ee72dd d0539ef 8ee72dd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import pandas as pd
from openai import OpenAI
import os
import json
import altair as alt
import sqlite3
import ast
import streamlit as st
from streamlit_navigation_bar import st_navbar
from reportlab.lib.pagesizes import letter
from reportlab.lib import colors
from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Image
from altair_saver import save
from utils.menu import menu
from loguru import logger
from datetime import datetime
st.set_page_config( page_title = 'Insight Lab', page_icon="chart_with_upwards_trend", layout = 'wide')
if "logger" not in st.session_state:
logger.add("Logger.txt", format="{time:YYYY-MM-DD HH:mm:ss} {level} {message}", level="INFO", enqueue=True)
st.session_state.logger = logger
def log_data(data):
current_date = datetime.now().strftime('%Y-%m-%d')
current_time = datetime.now().strftime('%H:%M:%S')
logger.info(f"Date: {current_date}\n========================================\nTime: {current_time}\nLogger Data: {data}\n----------------------------------------\n")
data = "This is Insight's lab log data."
log_data(data)
else:
logger = st.session_state.logger
# logger.add("file_{time}.log")
# Initialize token storage
token_file = "token_usage.json"
if not os.path.exists(token_file):
with open(token_file, 'w') as f:
json.dump({}, f)
def store_token_usage(token_usage):
# current_month = "2025-01"
current_month = datetime.now().strftime('%Y-%m')
with open(token_file, 'r') as f:
token_data = json.load(f)
if current_month in token_data:
token_data[current_month] += token_usage
else:
token_data[current_month] = token_usage
with open(token_file, 'w') as f:
json.dump(token_data, f)
def get_monthly_token_usage():
with open(token_file, 'r') as f:
token_data = json.load(f)
return token_data
# Example usage of get_monthly_token_usage function
monthly_token_usage = get_monthly_token_usage()
# print(monthly_token_usage)
menu()
|