Spaces:
Running
Running
| import streamlit as st | |
| import os | |
| import base64 | |
| from dotenv import load_dotenv | |
| from rag import Rag | |
| from util import getYamlConfig | |
| from utils.document import generate_pdf | |
| load_dotenv() | |
| GROUP_NAME = os.environ.get("APP_NAME") | |
| LOGO = "assets/logo.png" | |
| userId = None | |
| def init_app(): | |
| config = getYamlConfig() | |
| query_params = st.query_params | |
| if "user" in query_params: | |
| global userId | |
| userId = query_params["user"] | |
| if len(st.session_state) == 0: | |
| st.session_state["messages"] = [] | |
| st.session_state["assistant"] = Rag() | |
| # st.session_state["data_dict"] = config['variables'] | |
| st.session_state["files"] = [] | |
| st.session_state["prompt_system"] = config['prompt_system'] | |
| st.session_state["chapters"] = config['chapters'] | |
| for chapter in st.session_state["chapters"]: | |
| st.session_state[f"chapter_{chapter['num']}"] = { | |
| "title": chapter['name'], | |
| "num": chapter['num'], | |
| "messages": [], | |
| "prompt_system": chapter['prompt_system'], | |
| "prompts": chapter['prompts'], | |
| } | |
| def main(): | |
| init_app() | |
| st.set_page_config(page_title=GROUP_NAME) | |
| st.logo(LOGO) | |
| st.title(GROUP_NAME) | |
| # st.write(f"**User ID:** {userId}") | |
| documents = st.Page("pages/documents.py", url_path="documents", title="Vos documents", icon="📂") | |
| form = st.Page("pages/chapter_params.py", url_path="configurations", title="Paramètres", icon="📋") | |
| chapters_pages = [ | |
| st.Page(f"pages/chap_{chapter['num']}.py", url_path=chapter['key'], title=chapter['name'], icon="📄") | |
| for chapter in st.session_state["chapters"] | |
| ] | |
| pg = st.navigation( | |
| { | |
| "Documents": [ | |
| documents, | |
| ], | |
| "Configurations": [ | |
| form, | |
| ], | |
| "Dialogue": chapters_pages, | |
| } | |
| ) | |
| if st.sidebar.button("Générer le PDF"): | |
| chapters = [] | |
| for chapter in st.session_state["chapters"]: | |
| chapters.append(st.session_state[f"chapter_{chapter['num']}"]) | |
| pdf_buffer = generate_pdf(chapters) | |
| st.sidebar.download_button( | |
| label="📥 Télécharger le PDF", | |
| data=pdf_buffer, | |
| file_name="diagnostique_CEGARA.pdf", | |
| mime="application/pdf" | |
| ) | |
| st.sidebar.markdown("### Documentation") | |
| st.sidebar.markdown("*Documentation :* [*Lien*]({})".format("https://docs.google.com/presentation/d/1b-cnCKge1qRBS26U5FfMScQPco4gZTn7tN1idhccsso/edit?usp=sharing")) | |
| st.sidebar.markdown("*Evaluation App. IA :*") | |
| st.sidebar.markdown("---") | |
| st.sidebar.image("./assets/bziiit.png", width=100) | |
| st.sidebar.markdown(""" | |
| <ul style="padding-left:10px;"> | |
| <li style="margin:0;"><em>Contributeur AFNOR SPEC IA FRUGAL</em></li> | |
| <li style="margin:0;"><em>Certifié AFNOR COMPETENCES</em></li> | |
| <li style="margin:0;"><em>Membre HUB FRANCE IA</em></li> | |
| <li style="margin:0;"><em>Membre OSFARM</em></li> | |
| </ul> | |
| """, unsafe_allow_html=True) | |
| st.sidebar.markdown("") | |
| st.sidebar.markdown("<style> p { margin : 0px ; } </style>",unsafe_allow_html=True) | |
| st.sidebar.markdown("2025 : Open source en Licence MIT") | |
| st.sidebar.markdown("[email protected]") | |
| pg.run() | |
| if __name__ == "__main__": | |
| main() |