import streamlit as st
import dotenv
import os

def main():
    
    dotenv.load_dotenv(dotenv_path=os.path.join('.streamlit', '.env'))

    st.set_page_config(page_title="RAG Agent", page_icon="🤖", layout="wide")

    # audit_page = st.Page("audit_page/audit.py", title="Audit", icon="📋", default=True)
    dialog_page = st.Page("audit_page/dialogue_doc.py", title="Dialoguer avec le document", icon="💬")
    kg_page = st.Page("audit_page/knowledge_graph.py", title="Graphe de connaissance", icon="🧠")
    # agents_page = st.Page("agents_page/catalogue.py", title="Catalogue des agents", icon="📇")
    # compte_rendu = st.Page("audit_page/compte_rendu.py", title="Compte rendu", icon="📝")
    # recommended_agents = st.Page("agents_page/recommended_agent.py", title="Agents recommandés", icon="⭐")
    # chatbot = st.Page("chatbot_page/chatbot.py", title="Chatbot", icon="💬")
    # documentation = st.Page("doc_page/documentation.py", title="Documentation", icon="📚")

    pg = st.navigation(
        {
            "Graphe de connaissance": [kg_page],
        }
    )

    pg.run()


if __name__ == "__main__":
    main()