Update app.py

app.py

@@ -5,9 +5,9 @@ import time
 text = "<h1 style='text-align: center; color: midnightblue; font-size: 30px;'>TCO Comparison Calculator"
 text1 = "<h1 style='text-align: center; color: midnightblue; font-size: 20px;'>First solution"
 text2 = "<h1 style='text-align: center; color: midnightblue; font-size: 20px;'>Second solution"
-text3 = "<h1 style='text-align: center; color: midnightblue; font-size: 20px;'>Compute and compare TCOs"
+text3 = "<h1 style='text-align: center; color: midnightblue; font-size: 20px;'>Compute and compare cost/token for both solutions"
 description=f"""
-<p>In this demo application, we help you compare different solutions for your AI incorporation plans, such as OpenSource or SaaS.</p>
+<p>In this demo application, we help you compare different solutions for your AI integration plans, such as Open source or SaaS.</p>
 <p>First, you'll have to choose the two solutions you'd like to compare. Then, follow the instructions to select your configurations for each solution and we will compute the cost/token accordingly to them. Eventually, both solutions are compared to evaluate which one best suits your needs.</p>
 """
 
@@ -28,7 +28,7 @@ def compute_ratio(tco1, tco2):
             return f"Error: {str(e)}"
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    Models: list[models.BaseTCOModel] = [models.OpenAIModel, models.OpenSourceLlama2Model, models.CohereModel]
+    Models: list[models.BaseTCOModel] = [models.OpenAIModel, models.CohereModel, models.OpenSourceLlama2Model,models.OpenSourceDIY]
     model_names = [Model().get_name() for Model in Models]
     gr.Markdown(value=text)
     gr.Markdown(value=description)
@@ -37,31 +37,31 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column():
             gr.Markdown(value=text1)
             page1 = models.ModelPage(Models)
-            dropdown = gr.Dropdown(model_names, interactive=True, label="First solution")
+            dropdown = gr.Dropdown(model_names, interactive=True, label="AI service options")
             page1.render()
 
         with gr.Column():
             gr.Markdown(value=text2)
             page2 = models.ModelPage(Models)
-            dropdown2 = gr.Dropdown(model_names, interactive=True, label="Second solution")
+            dropdown2 = gr.Dropdown(model_names, interactive=True, label="AI service options")
             page2.render()
             
     dropdown.change(page1.make_model_visible, inputs=dropdown, outputs=page1.get_all_components())
     dropdown2.change(page2.make_model_visible, inputs=dropdown2, outputs=page2.get_all_components())
     
     gr.Markdown(value=text3)
-    compute_tco_btn = gr.Button("Compute TCOs", size="lg")
+    compute_tco_btn = gr.Button("Compute cost/token", size="lg")
     tco1 = gr.State()
     tco2 = gr.State()
     
     with gr.Row():  
         with gr.Column():
-            tco_output = gr.Text("Output 1: ", label="TCO for the first solution")
+            tco_output = gr.Text("Output 1: ", label="Cost/token for the first solution")
             with gr.Accordion("Open to see the formula", open=False):
                 tco_formula = gr.Markdown()
     
         with gr.Column():
-            tco_output2 = gr.Text("Output 2: ", label="TCO for the second solution")
+            tco_output2 = gr.Text("Output 2: ", label="Cost/token for the second solution")
             with gr.Accordion("Open to see the formula", open=False):
                 tco_formula2 = gr.Markdown()