Update app.py

app.py

@@ -105,27 +105,47 @@ if 'dictée' not in st.session_state:
 st.set_page_config(layout="wide")
 st.title('Générateur de Dictée')
 
-with st.expander("Paramètres de la dictée", expanded=True):
-    mode = st.radio("Mode:", ["S'entrainer: Vous aurez uniquement les audios suivi d'une correction par IA (Pour 1 seul personne)", "Entrainer:  Vous aurez uniquement le texte de la dictée pour entrainer quelqu'un d'autre (Pour 2 ou + personnes)"])
+# Ajouter un container pour le paramètre de mode
+with st.container():
+    mode = st.radio("Mode:", ["S'entrainer", "Entrainer"], key="mode")
+
+# Ajouter un container pour les paramètres de la dictée
+params_placeholder = st.empty()
+
+if mode == "S'entrainer":
+    st.subheader("Paramètres de la dictée pour s'entrainer")
+else:
+    st.subheader("Paramètres de la dictée pour entrainer quelqu'un d'autre")
+
+# Utiliser des colonnes pour organiser les éléments
+col1, col2 = st.columns(2)
+
+with col1:
     classe = st.selectbox("Classe", ["CP", "CE1", "CE2", "CM1", "CM2", "6ème", "5ème", "4ème", "3ème", "Seconde", "Premiere", "Terminale"], index=2)
+with col2:
     longueur = st.slider("Longueur de la dictée (nombre de mots)", 50, 500, 200)
-    bouton_generer = st.button('Générer la Dictée')
+
+bouton_generer = st.button('Générer la Dictée')
 
 if bouton_generer:
+    # Masquer les paramètres
+    params_placeholder.empty()
+
     with st.spinner("Génération de la dictée en cours..."):
         dictee = generer_dictee(classe, longueur)
         st.session_state.dictée = dictee
 
 if st.session_state.dictée:
     # Afficher la dictée ou les audios en fonction du mode
-    if mode == "S'entrainer: Vous aurez uniquement les audios suivi d'une correction par IA (Pour 1 seul personne)":
+    if mode == "S'entrainer":
         audio_urls = dictee_to_audio_segmented(st.session_state.dictée)
         concatenated_audio_path = concatenate_audio(audio_urls)
 
         col1, col2 = st.columns(2)
 
         with col1:
-            st.audio(concatenated_audio_path, format='audio/wav', start_time=0)
+            with st.expander("Écouter la dictée complète"):
+                st.audio(concatenated_audio_path, format='audio/wav', start_time=0)
             with st.expander("Phrases de la Dictée"):
                 for idx, url in enumerate(audio_urls, start=1):
                     st.markdown(f"**Phrase {idx}:**")
@@ -138,9 +158,16 @@ if st.session_state.dictée:
                     correction = correction_dictee(dictee_utilisateur)
                     st.markdown("Voici la correction :", correction, height=500)
 
-    elif mode == "Entrainer:  Vous aurez uniquement le texte de la dictée pour entrainer quelqu'un d'autre (Pour 2 ou + personnes)":
+    elif mode == "Entrainer":
         st.text_area("Voici votre dictée :", st.session_state.dictée, height=300)
 
     # Ajouter un bouton "Retour" pour réinitialiser la session
-    if st.button('Retour'):
-        st.session_state.dictée = None
+    col1, col2 = st.columns(2)
+    with col2:
+        if st.button('Retour'):
+            st.session_state.dictée = None
+            # Réafficher les paramètres
+            params_placeholder.subheader("Paramètres de la dictée")
+            params_placeholder.selectbox("Classe", ["CP", "CE1", "CE2", "CM1", "CM2", "6ème", "5ème", "4ème", "3ème", "Seconde", "Premiere", "Terminale"], index=2)
+            params_placeholder.slider("Longueur de la dictée (nombre de mots)", 50, 500, 200)
+            params_placeholder.button('Générer la Dictée')