import random 
import pandas as pd
import streamlit as st
from datasets import load_dataset

st.title("Code Arena")

with st.spinner("Loading data...", show_time=True):
    problem_dict = dict()
    ds = load_dataset("Elfsong/leetcode_data", split='train')
    for problem in ds:
        problem_id = problem["problem_id"]
        problem_dict[problem_id] = problem


if "problem" in st.query_params:
    problem_id = int(st.query_params["problem"])
    st.write(f"Problem Description - [{problem_id}]")
    st.write(problem_dict[problem_id])
    
else:
    with st.spinner("Loading Framework...", show_time=True):
        df = pd.DataFrame(
            {
                "problem_id": [int(problem['problem_id']) for problem in ds],
                "problem_link": ["https://huggingface.co/spaces/Elfsong/CodeArena/?problem=" + str(problem['problem_id']) for problem in ds],
                "dynamic_point": [[random.randint(0, 5000) for _ in range(30)] for problem in ds],
            }
        )
    
    tab_problem, tab_submission, tab_model = st.tabs(["Problems", "Submissions", "Models"])

    with tab_problem:
        st.dataframe(
            df,
            column_config={
                "problem_id": st.column_config.NumberColumn("Problem ID"),
                "dynamic_point": st.column_config.LineChartColumn("Dynamic Point", y_min=0, y_max=5000),
                "problem_link": st.column_config.LinkColumn("Link", display_text="Open"),
            },
            column_order=("problem_id", "dynamic_point", "problem_link"),
            hide_index=True,
        )

    with tab_submission:
        st.header("Submissions")

    with tab_model:
        st.header("Models")