|
import streamlit as st |
|
import dotenv |
|
import os |
|
|
|
from rag import Rag |
|
from vectore_store.PineconeConnector import PineconeConnector |
|
from vectore_store.VectoreStoreManager import VectoreStoreManager |
|
|
|
GROUP_NAME = "Groupe 1" |
|
|
|
def main(): |
|
|
|
if len(st.session_state) == 0: |
|
|
|
pinecone_connector = PineconeConnector() |
|
vs_manager = VectoreStoreManager(pinecone_connector) |
|
|
|
st.session_state["messages"] = [] |
|
st.session_state["assistant"] = Rag(vectore_store=vs_manager) |
|
|
|
st.set_page_config(page_title=GROUP_NAME) |
|
|
|
st.title(GROUP_NAME) |
|
|
|
prompt_system = st.Page("pages/prompt_system.py", title="Prompt système", icon="📋", default=True) |
|
saved_documents = st.Page("pages/persistent_documents.py", title="Documents Communs", icon="📋") |
|
documents = st.Page("pages/documents.py", title="Documents", icon="📋") |
|
form = st.Page("pages/form.py", title="Formulaire", icon="📋") |
|
chatbot = st.Page("pages/chatbot.py", title="Chatbot", icon="📋") |
|
|
|
pg = st.navigation( |
|
[ |
|
saved_documents, |
|
prompt_system, |
|
documents, |
|
form, |
|
chatbot |
|
] |
|
) |
|
|
|
pg.run() |
|
|
|
|
|
if __name__ == "__main__": |
|
main() |
