import os

import streamlit as st
import streamlit.components.v1 as components

from pages.nav import Navbar
from pages.utils import start_server


certificate = os.getenv('PRIVATE_KEY')
dns_name = os.getenv('SERVER_DNS_NAME')


def main():

    ## Set Streamlit configuration
    st.set_page_config(page_title='Merlion-AI-Audio', page_icon = "🔥", layout='wide')

    Navbar()

    ## Set up home page Title
    col1, col2 = st.columns([1, 4])
    custom_html = """
    <div class="banner">
        <img src="https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcRhB2e_AhOe11wKxnnwOmOVg9E7J1MBgiTeYzzFAESwcCP5IbBAc2X8BwGChMfJzwqtVg&usqp=CAU" alt="Banner Image">
    </div>
    <style>
        .banner {
            width: 100%;
            height: 200px;
            overflow: visible;
        }
        .banner img {
            width: 100%;
            object-fit: cover;
        }
    </style>
    """
    with col1:
        components.html(custom_html)
    with col2: 
        st.write("# Welcome to Merlion AI - AudioLLMs 🤖")

    ## Set up home page other information
    st.markdown('')


if __name__ == '__main__':
    start_server(certificate, dns_name)
    main()