import gradio as gr
import pandas as pd
import numpy as np

# UI Root
with gr.Blocks() as demo:
    gr.Markdown("## CompUGE-Bench: Comparative Understanding and Generation Evaluation Benchmarks")

    # Main Tabs
    with gr.Tab("Leaderboards"):
        gr.Markdown("### Leaderboards")

        # CQI Tab
        with gr.Tab("Question Identification"):
            gr.Markdown("### Comparative Question Identification Leaderboard")
            # read dataframe from CQI_Learboard.csv
            # TODO: replace the following line with the actual leaderboard file
            CQI_Leaderboard = pd.read_csv("CQI_Leaderboard.csv")
            cqi_leaderboard = gr.components.Dataframe(CQI_Leaderboard)

        # OAI Tab
        with gr.Tab("Object and Aspect Identification"):
            gr.Markdown("### Object & Aspect Identification Leaderboard")
            gr.Markdown("The OAI leaderboard will be opened soon!")

        # SC Tab
        with gr.Tab("Stance Classification"):
            gr.Markdown("### Stance Clasification Leaderboard")
            gr.Markdown("The SC leaderboard will be opened soon!")

        # SG Tab
        with gr.Tab("Summary Generation"):
            gr.Markdown("### Summary Generation Leaderboard")
            gr.Markdown("The Summary Generation leaderboard will be opened soon!")

    # Model Submissions Tab
    with gr.Tab("Model Submissions"):
        gr.Markdown("### Submission")
        gr.Markdown("The submission will be opened soon!")

    # About Tab
    with gr.Tab("About"):
        gr.Markdown("### About")
        gr.Markdown("CompUGE-Bench is a benchmark for comparative understanding and generation evaluation.")

    # Contact Tab
    with gr.Tab("Contact"):
        gr.Markdown("### Contact")
        gr.Markdown("For any questions, please contact us at ahmad.shallouf@uni-hamburg.de")
# Launch public demo
demo.launch(share=True)