chore: Refactor get_prompts_list function to improve readability and user experience

prompt.py

@@ -16,7 +16,13 @@ def get_prompts_list():
         # Check if 'name', 'context', and 'text' are in df columns
         if 'name' in df.columns and 'context' in df.columns and 'text' in df.columns:
             # Filter prompts with id_context as 'rse' or 'transition'
-            df = df[df['id_context'].isin(['pour-accelerer-la-demarche-rse', 'pour-accelerer-la-demarche-transition-ecologique','ressources-humaines'])]
+            df = df[df['id_context'].isin([
+                'identifier-parties-prenantes-organisation',
+                'animer-parties-prenantes-organisation',
+                'pour-accelerer-la-demarche-rse',
+                'pour-accelerer-la-demarche-transition-ecologique',
+                'ressources-humaines'
+            ])]
 
             # Extract 'name' from 'context' dictionary
             df['context'] = df['context'].apply(lambda x: x.get('name') if isinstance(x, dict) else x)
@@ -27,36 +33,35 @@ def get_prompts_list():
             # Group by 'context'
             grouped = df.groupby('context')
 
-            num = 1
 
             for name, group in grouped:
-                st.subheader(name)  # Display the context name as a subheader
-                for i, row in group.iterrows():
-                    col1, col2, col3, col4 = st.columns((1, 2, 2, 2))
-                    col1.write(num)  # index
-                    col2.write(row['name'])  # name
-                    col3.write(row['text'])  # text
-                    num += 1
-                    
-                    button_phold = col4.empty()  # create a placeholder
-                    but1, but2 = button_phold.columns(2)
-                    
-                    do_action = but1.button('Voir plus', key=f"v{i}")
-                    execute = but2.button('Executer', key=f"e{i}")
-                    if execute:
-                        st.session_state.chat_history.append(HumanMessage(content=prompts[i]['text']))
-                        return 1
-                    if do_action:
-                        prompt_html = prompts[i]['text'].replace('\n', '<br>')
-                        prompt_metadata = extract_metadata(prompts[i])
+                num = 1
+                
+                with st.expander(name):
+                    for i, row in group.iterrows():
+                        col1,  col3, col4 = st.columns((0.4,  4, 2))
+                        col1.write(num)  # index
+                        # col2.write(row['name'])  # name
+                        col3.write(row['text'])  # text
+                        num += 1
+                        
+                        button_phold = col4.empty()  # create a placeholder
+                        but1, but2 = button_phold.columns(2)
                         
-                        for text in prompt_metadata:
-                            prompt_html = prompt_html.replace(f"{text}", f"<span style='font-weight:bold'>{text}</span>")
-                                
-                        st.html(prompt_html)
-                        i
-                            # Display the full text
-                          # remove button
+                        do_action = but1.button('Voir plus', key=f"v{i}")
+                        execute = but2.button('Executer', key=f"e{i}")
+                        if execute:
+                            st.session_state.chat_history.append(HumanMessage(content=prompts[i]['text']))
+                            return 1
+                        if do_action:
+                            prompt_html = prompts[i]['text'].replace('\n', '<br>')
+                            prompt_metadata = extract_metadata(prompts[i])
+                            
+                            for text in prompt_metadata:
+                                prompt_html = prompt_html.replace(f"{text}", f"<span style='font-weight:bold'>{text}</span>")
+                                    
+                            st.html(prompt_html)
+
         else:
             st.write("Data does not contain 'name', 'context', and 'text' fields.")
     else: