Update app.py

app.py

@@ -35,35 +35,18 @@ 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
 
-        # 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')
+    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}")
 
-        # 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 = [
@@ -92,19 +75,12 @@ 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=display_statistics,
+    fn=get_statistics,
     inputs=None,
-    outputs=[gr.Textbox(), gr.Image()],
+    outputs="text",
     title="Boba Consumption Stats"
 )
 
 # Combine all interfaces into one app
-gr.TabbedInterface([iface, leaderboard_iface, stats_iface], ["Data Entry", "Leaderboard", "Statistics"]).launch()
+gr.TabbedInterface([iface, leaderboard_iface, stats_iface], ["Data Entry", "Leaderboard", "Statistics"]).launch()