Update app.py

app.py

@@ -25,8 +25,8 @@ def submit_boba_data(use_today, date, shop, drink, toppings, size, sugar, rating
     return boba_data
 
 def get_leaderboard():
-    leaderboard = boba_data.groupby('Shop').agg({'Rating': 'mean', 'Shop': 'count'}).sort_values(by='Rating', ascending=False)
-    return leaderboard
+    leaderboard = boba_data.groupby('Shop').agg({'Rating': 'mean', 'Shop': 'count'}).rename(columns={'Shop': 'Number of Entries'}).sort_values(by='Rating', ascending=False)
+    return leaderboard.reset_index()
 
 def get_statistics():
     total_drinks = len(boba_data)
@@ -35,17 +35,35 @@ def get_statistics():
         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}")
+        # Plotting
+        fig, axs = plt.subplots(1, 2, figsize=(12, 5))
+
+        # Plot 1: Sugar Distribution
+        sns.histplot(boba_data['Sugar'], bins=10, kde=True, ax=axs[0])
+        axs[0].set_title('Sugar Distribution')
+        axs[0].set_xlabel('Sugar (%)')
+        axs[0].set_ylabel('Frequency')
+
+        # Plot 2: Rating Distribution
+        sns.histplot(boba_data['Rating'], bins=5, kde=True, ax=axs[1])
+        axs[1].set_title('Rating Distribution')
+        axs[1].set_xlabel('Rating')
+        axs[1].set_ylabel('Frequency')
+
+        # Save plots to BytesIO and convert to image
+        img = BytesIO()
+        plt.savefig(img, format='png')
+        plt.close(fig)
+        img.seek(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}", img.getvalue())
+    else:
+        return "No data available", None
 
 # Define Gradio inputs and interface
 inputs = [
@@ -74,10 +92,17 @@ leaderboard_iface = gr.Interface(
     title="Boba Leaderboard"
 )
 
+def display_statistics():
+    stats_text, stats_img = get_statistics()
+    if stats_img:
+        return [stats_text, gr.Image(value=stats_img)]
+    else:
+        return stats_text
+
 stats_iface = gr.Interface(
-    fn=get_statistics,
+    fn=display_statistics,
     inputs=None,
-    outputs="text",
+    outputs=[gr.Textbox(), gr.Image()],
     title="Boba Consumption Stats"
 )