import gradio as gr
import pandas as pd

# Placeholder for boba data
boba_data = pd.DataFrame(columns=["Date", "Shop", "Drink", "Toppings", "Size", "Rating", "Price"])

def submit_boba_data(date, shop, drink, toppings, size, rating, price):
    global boba_data
    new_entry = {"Date": date, "Shop": shop, "Drink": drink, "Toppings": toppings, "Size": size, "Rating": rating, "Price": price}
    boba_data = boba_data.concat(new_entry, ignore_index=True)
    return boba_data

def get_leaderboard():
    leaderboard = boba_data.groupby('Shop').agg({'Rating': 'mean', 'Shop': 'count'}).sort_values(by='Rating', ascending=False)
    return 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}"

# Define Gradio inputs and interface
inputs = [
    gr.Textbox(label="Date"),
    gr.Textbox(label="Shop"),
    gr.Textbox(label="Drink"),
    gr.Textbox(label="Toppings"),
    gr.Dropdown(["Medium", "Large"], label="Size"),
    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()