Spaces:
Sleeping
Sleeping
File size: 2,368 Bytes
da04961 |
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 63 64 |
import streamlit as st
import openai
import pandas as pd
import datetime
# Initialize Streamlit app
st.set_page_config(page_title="OpenAI Usage Dashboard", layout="centered")
# Set up OpenAI API key
st.sidebar.title("OpenAI API Usage Dashboard")
api_key = st.sidebar.text_input("Enter your OpenAI API Key:", type="password")
openai.api_key = api_key
# Title and description
st.title("OpenAI Usage Tracker")
st.write("This app retrieves and displays your OpenAI API usage data for a specified date range.")
# Date selection
start_date = st.sidebar.date_input("Select Start Date:", datetime.date.today() - datetime.timedelta(days=30))
end_date = st.sidebar.date_input("Select End Date:", datetime.date.today())
# Ensure end date is not before start date
if start_date > end_date:
st.error("End Date must be after Start Date")
# Function to retrieve usage data
def get_usage_data(api_key, start_date, end_date):
try:
openai.api_key = api_key
usage = openai.api_usage.get(start_date=start_date, end_date=end_date)
usage_data = usage['data']
# Create a DataFrame from the usage data
data = pd.DataFrame(usage_data)
data['date'] = pd.to_datetime(data['date'])
data.set_index('date', inplace=True)
return data[['n_tokens_total']]
except Exception as e:
st.error(f"An error occurred: {e}")
return None
# Retrieve and display usage data if API key is provided
if api_key:
usage_data = get_usage_data(api_key, start_date.strftime("%Y-%m-%d"), end_date.strftime("%Y-%m-%d"))
# Display usage data as a table and line chart
if usage_data is not None:
st.subheader("Usage Data")
st.write("Displaying usage from", start_date, "to", end_date)
st.dataframe(usage_data)
# Display line chart for usage over time
st.subheader("Token Usage Over Time")
st.line_chart(usage_data['n_tokens_total'])
# Display statistics
total_usage = usage_data['n_tokens_total'].sum()
avg_usage = usage_data['n_tokens_total'].mean()
st.subheader("Statistics")
st.write(f"Total Token Usage: {total_usage}")
st.write(f"Average Daily Token Usage: {avg_usage:.2f}")
else:
st.info("Enter your API key to view usage data.")
else:
st.warning("Please enter your OpenAI API key to proceed.") |