chore: Update prompt execution logic and metadata handling

RAG_PDF_WEB.py

@@ -12,6 +12,8 @@ from langchain_core.runnables import RunnablePassthrough
 from langchain_community.document_loaders import WebBaseLoader
 import os
 
+from session import set_rag
+
 def get_docs_from_website(urls):
     loader = WebBaseLoader(urls, header_template={
       'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.0.0 Safari/537.36',
@@ -214,7 +216,4 @@ def rag_pdf_web():
                     st.markdown(response.content)
 
     if vectorstore and chain:
-        st.success("Le vectorstore et le RAG chain ont été créés avec succès!")
-        return vectorstore, chain
-    else:
-        return None, None
+        set_rag(vectorstore, chain)

app.py

@@ -85,10 +85,7 @@ def main():
                 display_materiality_matrix(selected_company, data, bziiit_data)
         
         elif ia_mode == "Audit flash RSE de vos contenus":
-            vectorstore, chain = rag_pdf_web()
-            if vectorstore and chain:
-                st.session_state['vectorstore'] = vectorstore
-                st.session_state['chain'] = chain
+            rag_pdf_web()
         
         elif ia_mode == "Prompt RSE disponibles":
             get_prompts_list()

prompt.py

@@ -51,7 +51,7 @@ def get_prompts_list():
 def prompt_execution():
     prompts = get_prompts()
 
-    selected_prompt = st.selectbox("Choisissez un prompt", prompts, format_func=lambda prompt: prompt['name'])
+    selected_prompt = st.selectbox("Choisissez un prompt", prompts, index=None, format_func=lambda prompt: prompt['name'])
     if selected_prompt:
         return selected_prompt
     
@@ -68,14 +68,14 @@ def execute_prompt(prompt):
     prompt['html'] = prompt['text'].replace('\n', '<br>')
     
     if prompt_metadata:
-        st.info("Métadonnées à compléter : ")
+        st.info("Données à compléter")
 
         # Demander à l'utilisateur de fournir des valeurs pour chaque métadonnée extraite
         user_inputs = {}
         for text in prompt_metadata:
             prompt['html'] = prompt['html'].replace(f"{text}", f"<span style='font-weight:bold'>{text}</span>")
                 
-            user_input = st.text_input(f"Enter value for {text}")
+            user_input = st.text_input(f"{text}")
             user_inputs[text] = user_input  # Stocker la valeur de l'entrée utilisateur
 
         # Remplacer les valeurs par le texte correspondant dans prompt['text']
@@ -85,16 +85,20 @@ def execute_prompt(prompt):
                 prompt['metadata'] = prompt['metadata'].replace(f"{key}", f"{value}")
 
     # Afficher les informations du prompt
+    if prompt_metadata:
+        st.markdown("---")
+
     st.html(prompt.get('html', 'No Text Provided'))
 
     if vectorstore and chain:
-        st.success("Vectorestore et chain trouvés")
         
         if st.button("Exécuter le prompt"):
             with st.spinner("Processing..."):
                 ambition = chain.invoke(prompt['metadata'])
                 st.markdown("### Réponse :")
                 st.markdown(ambition.content)
+    else:
+        st.error("RAG non configuré. Veuillez configurer votre RAG pour exécuter le prompt.")
 
 # Extraire le texte entre crochets dans le prompt
 def extract_metadata(prompt):

session.py

@@ -7,4 +7,8 @@ def get_rag():
         
         return vectorstore, chain
     else:
-        return None, None
+        return None, None
+    
+def set_rag(vectorstore, chain):
+    st.session_state['vectorstore'] = vectorstore
+    st.session_state['chain'] = chain