import streamlit as st
from datasets import load_dataset

st.title("Code Arena")

with st.spinner("Loading data...", show_time=True):
    ds = load_dataset("Elfsong/leetcode_data")

tab_problem, tab_model = st.tabs(["Problems", "Models"])

with tab_problem:
    for problem in range(10):
        with st.container(border=True):
            id_col, title_col, difficulty_col, dp_col = st.columns(4)
            id_col.write(problem)
            title_col.write(problem)
            difficulty_col.write(problem)
            dp_col.write(problem)

with tab_model:
    st.header("Models")