Changer affichage des groupes

app.py

@@ -20,6 +20,8 @@ if 'name' not in st.session_state.keys():
 @st.cache_resource
 def initialize(name):
 
+    if name == "dimanov_et_al": 
+        return None, None, None, None, None
     articles = pd.read_csv('extract_sciences_po.csv')
 
     with open(f"{SAVE_PATH}/{mapping[mapping_noms[name]]['save_path']}", 'r') as f : 
@@ -125,9 +127,13 @@ def display_article(article):
 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')
+st.selectbox("Choisir groupe", [mapping[k]['auteurs'] for k in mapping.keys()], key='name', index=1)
 
-if st.session_state['name']!="dimanov_et_al":
+if st.session_state['name']=="dimanov_et_al":
+
+    st.toast("Ce groupe n'a pas renseigné son prompt !")
+
+else :
     with st.expander(f"Prompt for model : {model}") : 
         st.markdown(prompt)
 

mapping_noms.txt

@@ -1,6 +1,6 @@
 {
-    "Inconnus 1": "sans_titre_1",
-    "Juliette Favarel-Denat, Clarisse Marion et Lucie Dradem": "favarel_et_al",
-    "Inconnus 2": "sans_titre_2",
-    "Dimanov, Girard, Barge, Cosic": "dimanov_et_al"
+    "Groupe 2": "sans_titre_1",
+    "Groupe 4": "favarel_et_al",
+    "Groupe 3": "sans_titre_2",
+    "Groupe 1": "dimanov_et_al"
 }

mapping_prompts.txt

@@ -1,30 +1,30 @@
 {
+    "dimanov_et_al": {
+        "auteurs": "Groupe 1",
+        "path_rapport": "rapports/POC - Dimanov, Girard, Barge, Cosic.pdf",
+        "save_path":"output_dimanov_et_al.txt",
+        "path_prompt": "prompts/prompt_dimanov_et_al.txt",
+        "client": "openai"
+    },
     "sans_titre_1": {
-        "auteurs": "Inconnus 1",
+        "auteurs": "Groupe 2",
         "path_rapport": "rapports/sans_titre_1.pdf",
         "path_prompt": "prompts/prompt_sans_titre_1.txt",
         "save_path":"output_sans_titre_1.txt",
         "client": "deepseek"
     },
-    "favarel_et_al": {
-        "auteurs": "Juliette Favarel-Denat, Clarisse Marion et Lucie Dradem",
-        "path_rapport": "rapports/POC_FAVAREL-DENAT_DRADEM_MARION.pdf",
-        "save_path":"output_favarel_et_al.txt",
-        "path_prompt": "prompts/prompt-favarel-et-al.txt",
-        "client": "deepseek"
-    },
     "sans_titre_2": {
-        "auteurs": "Inconnus 2",
+        "auteurs": "Groupe 3",
         "path_rapport": "rapports/sans_titre_2.pdf",
         "save_path":"output_sans_titre_2.txt",
         "path_prompt": "prompts/prompt_sans_titre_2.txt",
         "client": "openai"
     },
-    "dimanov_et_al": {
-        "auteurs": "Dimanov, Girard, Barge, Cosic",
-        "path_rapport": "rapports/POC - Dimanov, Girard, Barge, Cosic.pdf",
-        "save_path":"output_dimanov_et_al.txt",
-        "path_prompt": "prompts/prompt_dimanov_et_al.txt",
-        "client": "openai"
+    "favarel_et_al": {
+        "auteurs": "Groupe 4",
+        "path_rapport": "rapports/POC_FAVAREL-DENAT_DRADEM_MARION.pdf",
+        "save_path":"output_favarel_et_al.txt",
+        "path_prompt": "prompts/prompt-favarel-et-al.txt",
+        "client": "deepseek"
     }
 }