Display model

app.py

@@ -77,10 +77,11 @@ def initialize(name):
 ))
     
     path_prompt = mapping[mapping_noms[name]]['path_prompt']
+    model = mapping[mapping_noms[name]]['client']
     with open(path_prompt, 'r') as f : 
         prompt = f.read()
 
-    return fig, display_principale, articles, prompt
+    return fig, display_principale, articles, prompt, model
 
 
 def display_article(article):
@@ -121,13 +122,13 @@ def display_article(article):
     #     return 
         
 
-fig, display_principale, articles, prompt = initialize(st.session_state['name'])
+fig, display_principale, articles, prompt, model = initialize(st.session_state['name'])
 # col1, col2, col3 = st.columns([0.5, 0.2, 0.3])
 
 st.selectbox("Choisir groupe", [mapping[k]['auteurs'] for k in mapping.keys()], key='name')
 
 if st.session_state['name']!=dimanov_et_al:
-    with st.expander("prompt") : 
+    with st.expander(f"Prompt for model : {model}") : 
         st.markdown(prompt)
 
     st.subheader('Répartition des articles par catégorie')