File size: 2,244 Bytes
64f9ede |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import streamlit as st
import yaml
import os
from streamlit_chat import message
from model import selector
def display_messages():
for i, (msg, is_user) in enumerate(st.session_state["messages"]):
message(msg, is_user=is_user, key=str(i))
st.session_state["thinking_spinner"] = st.empty()
# Charger les données YAML
def load_yaml(file_path):
with open(file_path, 'r', encoding='utf-8') as file:
data = yaml.safe_load(file)
return data
def process_input():
if "user_input" in st.session_state and st.session_state["user_input"] and len(st.session_state["user_input"].strip()) > 0:
user_text = st.session_state["user_input"].strip()
prompt_sys = st.session_state.prompt_system if 'prompt_system' in st.session_state and st.session_state.prompt_system != '' else ""
with st.session_state["thinking_spinner"], st.spinner(f"Je réfléchis"):
agent_text = st.session_state["assistant"].ask(user_text, prompt_sys, st.session_state["messages"] if "messages" in st.session_state else [], variables=st.session_state["data_dict"])
st.session_state["messages"].append((user_text, True))
st.session_state["messages"].append((agent_text, False))
@st.dialog("Choisissez votre question prédéfinie")
def show_prompts():
file_path = os.path.join(os.path.dirname(__file__), '..', 'config.yaml')
yaml_data = load_yaml(file_path)
# Boucle à travers les éléments YAML
for categroy in yaml_data['prompts']: # Exemple avec la catégorie conversation
st.write(categroy.capitalize())
for item in yaml_data['prompts'][categroy]:
if st.button(item, key=f"button_{item}"):
st.session_state["user_input"] = item
st.rerun()
def page():
st.subheader("Posez vos questions")
if "user_input" in st.session_state:
process_input()
if "assistant" not in st.session_state:
st.text("Assistant non initialisé")
# Bouton pour ouvrir la modale
if st.button("Questions pré-définies"):
show_prompts()
selector.ModelSelector()
display_messages()
st.text_input("Message", key="user_input", on_change=process_input)
page() |