Update app.py

app.py

@@ -7,7 +7,7 @@ import matplotlib.pyplot as plt
 text = "<h1 style='text-align: center; color: #333333; font-size: 40px;'>TCO Comparison Calculator"
 text2 = "Please note that the cost/request only defines the infrastructure cost for deployment. The labor cost must be added for the whole AI model service deployment TCO."
 description=f"""
-<p>In this demo application, we help you compare different AI model services, such as Open source or SaaS solutions, based on the Total Cost of Ownership for their deployment. 😊</p> 
+<p>In this demo application, we help you compare different AI model services, such as Open source or SaaS solutions, based on the Total Cost of Ownership (TOC) for their deployment. 😊</p> 
 <p>Please note that we focus on getting the service up and running, but not the maintenance that follows.🚀</p>
 <p>If you want to <strong>contribute to the calculator</strong> by adding your own AI service option, follow this <a href="https://huggingface.co/spaces/mithril-security/TCO_calculator/blob/main/How_to_contribute.md">tutorial</a> 👈. </p>
 """
@@ -77,23 +77,23 @@ def create_table(tco1, tco2, labor_cost1, labor_cost2, dropdown, dropdown2, late
 def compute_cost_per_request(*args):
     dropdown_id = args[-2]
     dropdown_id2 = args[-1]
-    if dropdown_id!=None and dropdown_id2!=None:
+    print("Dropdown 1 value:", dropdown_id)
+    print("Dropdown 2 value:", dropdown_id2)
+    if dropdown_id!="" and dropdown_id2!="":
         # Separate the arguments for page1 and page2
-        args_list = list(args)
-        args_page1 = args_list[:len(page1.get_all_components_for_cost_computing())] + [dropdown_id, input_tokens, output_tokens]
-        args_page2 = args_list[len(page1.get_all_components_for_cost_computing()):] + [dropdown_id2, input_tokens, output_tokens]
+        args_page1 = list(args) + [dropdown_id, input_tokens, output_tokens]
+        args_page2 = list(args) + [dropdown_id2, input_tokens, output_tokens]
+        print("Args for Page 1:", args_page1)
+        print("Args for Page 2:", args_page2)
         # Compute and compare using both pages
         result_page1 = page1.compute_cost_per_token(*args_page1)
         result_page2 = page2.compute_cost_per_token(*args_page2)
         # Unpack the results from the functions
         tco1, latency, labor_cost1 = result_page1
-        tco2, latency2, labor_cost2 = result_page2
-        
+        tco2, latency2, labor_cost2 = result_page2   
         return tco1, latency, labor_cost1, tco2, latency2, labor_cost2
     else: 
         raise gr.Error("Please select two AI service options.")
-    
-    
 
 def update_plot(tco1, tco2, dropdown, dropdown2, labour_cost1, labour_cost2):