Update app.py

app.py

@@ -1,18 +1,16 @@
 import streamlit as st
 import numpy as np
 import plotly.graph_objects as go
-# import time
 
 st.set_page_config()
-st.title('RoI on Openai Chatgpt vs API plans') #
-
+st.title('Cost Comparison Chatgpt Plus vs Custom-Chatgpt CoPilot')
 
 @st.dialog("Assumptions")
 def note():
     st.markdown('''
                 # ChatGPT plans
-                - Plus - $20/month with 320 and 640 msgs per day cap for GPT4 and GPT-4o models resp
-                - Teams - $30/month with 800 and 1600 msgs per day cap for GPT4 and GPT-4o models resp (minimum 2 users)  
+                - Plus - $20/month with 320 and 640 msgs per day cap for GPT-4 and GPT-4o models resp
+                - Teams - $30/month with 800 and 1600 msgs per day cap for GPT-4 and GPT-4o models resp (minimum 2 users)  
                 [link](https://openai.com/chatgpt/pricing)
 
                 # API plans
@@ -21,60 +19,77 @@ def note():
                 [link](https://openai.com/api/pricing)
                 
                 # Assumptions
-                - 1 token = 0.75 words (1.33 token $\\approx$ 1 word)
-                - For Chatgpt API plans, we consider 1 input prompt = 250 words and same word count for output messages 
-                
+                - 1 token = 0.75 words (1.33 tokens ≈ 1 word)
+                - For ChatGPT API plans, we consider 1 input prompt = 250 words and same word count for output messages 
                 ''')
 
-# st.button("Note")
-if st.button("Show Assumptions"):
-    note()
+# if st.button("Show Assumptions"):
+#     note()
 
 with st.sidebar:
-  st.title("Model Parameters")
-  max_months = st.select_slider("No. of months to show on plot (x-axis)", 
-                                  options = np.arange(0, 37, 1, dtype=int), value=13)
+    st.title("Model Parameters")
+    max_months = st.select_slider("No. of months to show on plot (x-axis)", 
+                                  options=np.arange(0, 37, 1, dtype=int), value=13)
   
-  no_of_users = st.select_slider("No. of users", 
-                                  options = np.arange(0, 100, 1, dtype=int), value=2)
+    no_of_users = st.select_slider("No. of users", 
+                                    options=np.arange(0, 100, 1, dtype=int), value=2)
   
-  st.subheader("Rest params app. for API plans only", divider="gray")
+    st.subheader("One-Time Development Cost (Custom App)", divider="gray")
+    development_cost = st.select_slider("Development Cost for Custom App ($)", 
+                                         options=np.arange(3000, 5001, 100, dtype=int), value=3000)
 
-  daily_no_of_prompts = st.select_slider("No. of prompts expected per day per user", 
-                                  options = np.arange(0, 1520, 20, dtype=int), value=100)
-  
-  input_prompts_word_cnt = st.select_slider("No. of words given as prompt to llm", 
-                                  options = np.arange(0, 1050, 50, dtype=int), value=200)
-  
-  output_prompts_word_cnt = st.select_slider("No. of words in the output response from llm", 
-                                  options = np.arange(0, 1050, 50, dtype=int), value=200)
-    
-    
-# fixed
+# Fixed parameters
 plan_limits = {"Plus": {"GPT-4o": 640, "GPT4": 320, "price": 20}, # 40 messages/3hrs
                "Team": {"GPT-4o": 1600, "GPT4": 800, "price": 30}} # 100 messages/3hrs, minimum 2 users price 25 pm if billed annually
 
-api_limits = {"GPT-4o": {"input": 5, "output": 15}, "GPT4": {"input": 30, "output": 60}} #usd per 1M tokens
-
-# assumptions
-token_to_word_ratio = 0.75
-word_to_token_ratio = 1/token_to_word_ratio
+api_limits = {"GPT-4o": {"input": 0.0100, "output": 0.0300}, "GPT4": {"input": 0.0100, "output": 0.0300}} #usd per 1K tokens
 
+# Timeline
 x = np.arange(0, max_months, dtype=int) # in months timeline
 
-api_price_per_month = {k: x*(v["input"] * daily_no_of_prompts * input_prompts_word_cnt * word_to_token_ratio + 
-                           v["output"] * daily_no_of_prompts * output_prompts_word_cnt * word_to_token_ratio)*30/1_000_000 for k, v in api_limits.items()}
+# Accumulated cost for ChatGPT-4o API
+api_price_per_month = x * (api_limits["GPT4"]["input"] * 85000 *0.001  + 
+                           api_limits["GPT4"]["output"] * 85000*0.001) * no_of_users 
+
+# Accumulated cost for ChatGPT Team
+team_price_per_month = x * no_of_users * plan_limits["Plus"]["price"]
+# # Calculate breakeven month
 
+savings_per_month=(team_price_per_month-api_price_per_month)/x
+if savings_per_month[1] > 0:
+    breakeven_month = development_cost / savings_per_month[1]
+else:
+    breakeven_month = None
 
+# Plotting
 fig = go.Figure()
-fig.add_trace(go.Scatter(x=x, y=api_price_per_month['GPT4'], name='GPT-4 API', fillcolor="darkkhaki"))
-fig.add_trace(go.Scatter(x=x, y=api_price_per_month['GPT-4o'], name='GPT-4o API', fillcolor="darkgreen"))
-fig.add_trace(go.Scatter(x=x, y=x*no_of_users*plan_limits["Team"]['price'], name='Chatgpt Team', fillcolor="firebrick"))
-fig.add_trace(go.Scatter(x=x, y=x*no_of_users*plan_limits["Plus"]['price'], name='Chatgpt Plus', fillcolor="dodgerblue"))
+fig.add_trace(go.Scatter(x=x, y=team_price_per_month, name='ChatGPT Plus', fillcolor="firebrick"))
+fig.add_trace(go.Scatter(x=x, y=api_price_per_month + development_cost, name='Custom-Chatgpt', fillcolor="darkgreen"))
 
+fig.update_layout(title="Accumulated Monthly Costs Over Time",
+                  xaxis_title="Time in Months",
+                  yaxis_title="Accumulated Cost in $")
 
-fig.update_layout(title="Accumulated monthly costs over time",
-                      xaxis_title="time in months",
-                      yaxis_title="accu. cost in $")
+
+# Add breakeven annotation if applicable
+if breakeven_month is not None and breakeven_month < max_months:
+    fig.add_vline(x=breakeven_month, line_dash="dash", line_color="black")
+    fig.add_annotation(x=breakeven_month, y=api_price_per_month[int(breakeven_month)] + development_cost,
+                       text=f"Breakeven at {breakeven_month:.1f} months",
+                       showarrow=True, arrowhead=1)
 
 st.plotly_chart(fig)
+
+# Description
+st.info('''
+This plot compares the accumulated costs of two options over time: ChatGPT Teams and GPT-4o API with a Custom Application. 
+
+1. **ChatGPT Teams**: A fixed cost per user with no upfront development costs. 
+2. **GPT-4o API with Custom Application**: Includes a one-time development cost and usage-based charges for input and output tokens.
+
+#### Key Insights:
+- The black dashed line indicates the **breakeven point**, where the total cost of the custom application becomes equal to or less than the cost of ChatGPT Teams.
+- Beyond this point, the custom application provides **cost savings** compared to ChatGPT Teams.
+
+Use the sliders to adjust parameters like the number of users and the one-time development cost to see how they affect the breakeven point and accumulated costs.
+''')