import streamlit as st

def main():
    st.set_page_config(page_title="World Model Challenge", layout="wide")
    
    st.title("World Model Challenge")
    st.markdown("### Welcome")
    st.write(
        "Welcome to the World Model Challenge server. This platform hosts three interrelated challenges "
        "designed to advance research in world models for robotics: Compression, Sampling, and Evaluation."
    )
    
    st.markdown("### Motivation")
    st.write(
        "Real-world robotics faces a fundamental challenge: environments are dynamic and change over time, "
        "making consistent evaluation of robot performance difficult. World models offer a solution by "
        "learning to simulate complex real-world interactions from raw sensor data. We believe these learned simulators will enable"
        "robust evaluation and iterative improvement of robot policies without the constraints of a physical testbed."
    )
    
    st.markdown("### The Challenges")
    
    st.markdown("#### Compression Challenge")
    st.write(
        "The Compression Challenge focuses on minimizing training loss over a diverse robot dataset. "
        "By effectively compressing and understanding the data, the challenge aims to measure how well a model "
        "can capture the complexities of real-world robot interactions. A lower loss indicates a better grasp "
        "of the underlying data."
    )
    
    st.markdown("#### Sampling Challenge")
    st.write(
        "Sampling focuses on generating realistic future outcomes in video sequences by predicting the next" 
        "frame given a sequence of prior frames. The goal is to produce coherent and plausible continuations"
        "of the video, accurately reflecting the dynamics of the scene. "
    )
    
    st.markdown("#### Evaluation Challenge")
    st.write(
        "The Evaluation Challenge tackles the ultimate question: Can you predict a robot's performance in the real world "
        "without physically deploying it? In this challenge, participants are tasked with ranking a set of robot policies based solely on simulation data."
        "These rankings will be compared to the real-world performance of the policies to determine a winner."
)

    
if __name__ == '__main__':
    main()