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()
    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 = [
    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"
)

stats_iface = gr.Interface(
    fn=get_statistics,
    inputs=None,
    outputs="text",
    title="Boba Consumption Stats"
)

# Combine all interfaces into one app
gr.TabbedInterface([iface, leaderboard_iface, stats_iface], ["Data Entry", "Leaderboard", "Statistics"]).launch()