Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import pandas as pd | |
| from datetime import datetime | |
| # Placeholder for boba data | |
| boba_data = pd.DataFrame(columns=["Date", "Shop", "Drink", "Toppings", "Size", "Sugar", "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') | |
| new_entry = pd.DataFrame([{ | |
| "Date": date, | |
| "Shop": shop, | |
| "Drink": drink, | |
| "Toppings": toppings, | |
| "Size": size, | |
| "Sugar": sugar, | |
| "Rating": rating, | |
| "Price": price | |
| }]) | |
| boba_data = pd.concat([boba_data, new_entry], ignore_index=True) | |
| return boba_data | |
| def get_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) | |
| 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() | |
| # 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 = [ | |
| gr.Checkbox(label="Use today's date?"), | |
| gr.Textbox(label="Date (YYYY-MM-DD)"), | |
| gr.Textbox(label="Shop"), | |
| gr.Textbox(label="Drink"), | |
| gr.Textbox(label="Toppings"), | |
| gr.Dropdown(["Medium", "Large"], label="Size"), | |
| gr.Slider(0, 200, label="Sugar (%)", step=25), | |
| gr.Slider(1, 5, label="Rating"), | |
| gr.Number(label="Price") | |
| ] | |
| iface = gr.Interface( | |
| fn=submit_boba_data, | |
| inputs=inputs, | |
| outputs=gr.Dataframe(), | |
| title="Boba Tracker Data Entry" | |
| ) | |
| leaderboard_iface = gr.Interface( | |
| fn=get_leaderboard, | |
| inputs=None, | |
| outputs=gr.Dataframe(), | |
| 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, | |
| inputs=None, | |
| outputs=[gr.Textbox(), gr.Image()], | |
| title="Boba Consumption Stats" | |
| ) | |
| # Combine all interfaces into one app | |
| gr.TabbedInterface([iface, leaderboard_iface, stats_iface], ["Data Entry", "Leaderboard", "Statistics"]).launch() | |