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", "Rating", "Price"])

def submit_boba_data(use_today, date, shop, drink, toppings, size, 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,
        "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'}).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.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(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()