Update app.py

- add page icon
- remove submit button and automatically calculate cost once the the prompt is entered in the text area

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import pandas as pd
 from transformers import GPT2Tokenizer
 
+st.set_page_config(page_title="LLM Cost Estimator", page_icon=":moneybag:")
+
 # Load the tokenizer
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 
@@ -36,20 +38,23 @@ def calculate_cost(model, input_tokens, output_tokens):
 
 # Streamlit App
 st.title("GPT/LLM Usage Cost Estimator")
+st.markdown("> _A simple tool to estimate the cost of using OpenAI models based on the number of input and output tokens._")
 
 # User input
-user_input = st.text_area("Enter your prompt")
-estimated_output_tokens = st.number_input("Estimated number of output tokens", min_value=0, value=50)
+user_input = st.text_area("", placeholder="Paste your prompt here...")
+estimated_output_tokens = st.number_input("Estimated number of output tokens", min_value=0, value=100)
 selected_model = st.selectbox("Select the model", list(rate_prices.keys()))
 
-if st.button("Calculate Cost"):
+# if st.button("Calculate Cost"):
+if user_input:
     input_tokens = count_tokens(user_input)
     total_cost = calculate_cost(selected_model, input_tokens, estimated_output_tokens)
 
+    st.markdown(f"### Estimated Cost: `${total_cost:.2f}`")
     # Create a DataFrame for displaying results
     results_df = pd.DataFrame({
         "Detail": ["Number of Input Tokens", "Estimated Number of Output Tokens", "Estimated Total Cost"],
-        "Value": [input_tokens, estimated_output_tokens, f"${total_cost:.2f}"]
+        "Value": [input_tokens, estimated_output_tokens, f"${total_cost:.4f}"]
     })
 
     # Display the results in a table
@@ -61,5 +66,5 @@ st.markdown("""
 ---
 <sup>**Note:** The pricing information is based on [OpenAI's pricing page](https://openai.com/pricing) as of 12/14/2023.</sup>
 <br>
-<sub>**Disclaimer:** This application is completely written by GPT-4 from a chat conversation.</sub>
+<sub>**Disclaimer:** This application was completely written by GPT-4 from a chat conversation.</sub>
 """, unsafe_allow_html=True)