Refactor chatbot.py to use getYamlConfig for loading prompts and streamline input processing

pages/chatbot.py

@@ -1,8 +1,8 @@
 import streamlit as st
-import yaml
-import os
 from streamlit_chat import message
 from model import selector
+from util import getYamlConfig
+
 
 def display_messages():
     for i, (msg, is_user) in enumerate(st.session_state["messages"]):
@@ -10,13 +10,6 @@ def display_messages():
     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()
@@ -24,24 +17,26 @@ def process_input():
         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"])
-
+            agent_text = st.session_state["assistant"].ask(user_text, prompt_system=prompt_sys, messages=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.session_state["user_input"] = ""
+
 
-@st.dialog("Choisissez votre prompt")
 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}"):
+    yaml_data = getYamlConfig()["prompts"]
+    
+    expander = st.expander("Prompts pré-définis")
+    
+    for categroy in yaml_data:
+        expander.write(categroy.capitalize())
+
+        for item in yaml_data[categroy]:
+            if expander.button(item, key=f"button_{item}"):
                 st.session_state["user_input"] = item
-                st.rerun()
+                process_input()
+
 
 def page():
     st.subheader("Posez vos questions")
@@ -51,16 +46,18 @@ def page():
 
     if "assistant" not in st.session_state:
         st.text("Assistant non initialisé")
-    
-    # Bouton pour ouvrir la modale
-    if st.button("Prompts pré-définis"):
-        show_prompts()
 
+    # Collpase for default prompts
+    show_prompts()
 
+    # Models selector
     selector.ModelSelector()
-
+    
+    # Displaying messages
     display_messages()
 
+    # Input user query
     st.text_input("Message", key="user_input", on_change=process_input)
 
+
 page()