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)
    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(
        {
            "Audit de contenus": [audit_page, compte_rendu, kg_page],
            "Equipe d'agents IA": [recommended_agents],
            "Chatbot": [chatbot],
            "Documentation": [documentation]
        }
    )

    pg.run()


if __name__ == "__main__":
    main()