Update app.py

app.py

@@ -3,9 +3,9 @@ import pandas as pd
 from datetime import datetime
 
 # Placeholder for boba data
-boba_data = pd.DataFrame(columns=["Date", "Shop", "Drink", "Toppings", "Size", "Rating", "Price"])
+boba_data = pd.DataFrame(columns=["Date", "Shop", "Drink", "Toppings", "Size", "Sugar", "Rating", "Price"])
 
-def submit_boba_data(use_today, date, shop, drink, toppings, size, rating, price):
+def submit_boba_data(use_today, date, shop, drink, toppings, size, sugar, rating, price):
     global boba_data
     if use_today:
         date = datetime.today().strftime('%Y-%m-%d')
@@ -16,6 +16,7 @@ def submit_boba_data(use_today, date, shop, drink, toppings, size, rating, price
         "Drink": drink,
         "Toppings": toppings,
         "Size": size,
+        "Sugar": sugar,
         "Rating": rating,
         "Price": price
     }])
@@ -29,10 +30,22 @@ def get_leaderboard():
 
 def get_statistics():
     total_drinks = len(boba_data)
-    favorite_drink = boba_data['Drink'].mode()[0] if total_drinks > 0 else "No data"
-    total_spent = boba_data['Price'].sum()
-    return f"Total drinks: {total_drinks}\nFavorite drink: {favorite_drink}\nTotal spent: ${total_spent}"
+    if total_drinks > 0:
+        favorite_drink = boba_data['Drink'].mode()[0]
+        favorite_topping = boba_data['Toppings'].mode()[0]
+        average_sugar = boba_data['Sugar'].mean()
+        total_spent = boba_data['Price'].sum()
+    else:
+        favorite_drink = "No data"
+        favorite_topping = "No data"
+        average_sugar = "No data"
+        total_spent = 0
 
+    return (f"Total drinks: {total_drinks}\n"
+            f"Favorite drink: {favorite_drink}\n"
+            f"Favorite topping: {favorite_topping}\n"
+            f"Average sugar level: {average_sugar:.2f}%\n"
+            f"Total spent: ${total_spent:.2f}")
 
 # Define Gradio inputs and interface
 inputs = [
@@ -41,7 +54,8 @@ inputs = [
     gr.Textbox(label="Shop"),
     gr.Textbox(label="Drink"),
     gr.Textbox(label="Toppings"),
-    gr.Dropdown(["Medium", "Large"], label="Size"),
+    gr.Dropdown(["Small", "Medium", "Large"], label="Size"),
+    gr.Slider(0, 100, label="Sugar (%)", step=5),
     gr.Slider(1, 5, label="Rating"),
     gr.Number(label="Price")
 ]