Update app.py

app.py

@@ -101,45 +101,40 @@ def segmenter_texte(texte):
 st.set_page_config(layout="wide")
 st.title('Générateur de Dictée')
 
-# Create a form for the user to fill out
 with st.form("dictation_form"):
-    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)"])
+    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)"])
     classe = st.selectbox("Classe", ["CP", "CE1", "CE2", "CM1", "CM2", "6ème", "5ème", "4ème", "3ème", "Seconde", "Premiere", "Terminale"], index=2)
     longueur = st.slider("Longueur de la dictée (nombre de mots)", 50, 500, 200)
-
-    # Submit button for the form
     submitted = st.form_submit_button("Générer la Dictée")
 
-if submitted:
-    with st.spinner("Génération de la dictée en cours..."):
-        dictee = generer_dictee(classe, longueur)
-        if mode == "S'entrainer: Vous aurez uniquement les audios suivi d'une correction par IA (Pour 1 seul personne)":
-            audio_urls = dictee_to_audio_segmented(dictee)
-            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("Phrases de la Dictée"):
-                    for idx, url in enumerate(audio_urls, start=1):
-                        st.markdown(f"**Phrase {idx}:**")
-                        st.audio(url, format='audio/wav')
-
-            with col2:
-                # Create a new form for the user's dictation
-                with st.form("user_dictation_form"):
-                    dictee_user = st.text_input("Écrivez la dictée ici:")
-                    submitted_user_dictation = st.form_submit_button("Correction")
-
-                if submitted_user_dictation:
-                    st.session_state.dictee_user = dictee_user
-
-                if 'dictee_user' in st.session_state:
-                    with st.spinner("Génération de la correction en cours..."):
-                        correction = correction_dictee(dictee, st.session_state.dictee_user)
-                        st.markdown("Voici la correction :")
-                        st.markdown(correction)
-
-        elif mode == "Entrainer:  Vous aurez uniquement le texte de la dictée pour entrainer quelqu'un d'autre (Pour 2 ou + personnes)":
-            st.text_area("Voici votre dictée :", dictee, height=300)
+if submitted or 'dictee' in st.session_state:
+    if 'dictee' not in st.session_state:
+        st.session_state.dictee = generer_dictee(classe, longueur)
+    dictee = st.session_state.dictee
+    if mode.startswith("S'entrainer"):
+        if 'audio_urls' not in st.session_state:
+            st.session_state.audio_urls = dictee_to_audio_segmented(dictee)
+        audio_urls = st.session_state.audio_urls
+        if 'concatenated_audio_path' not in st.session_state:
+            st.session_state.concatenated_audio_path = concatenate_audio(audio_urls)
+        concatenated_audio_path = st.session_state.concatenated_audio_path
+        
+        col1, col2 = st.columns(2)
+        with col1:
+            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}:**")
+                    st.audio(url, format='audio/wav')
+        
+        with col2:
+            dictee_user = st.text_input("Écrivez la dictée ici:", key="dictee_user")
+            if st.button("Correction", key="submit_correction"):
+                st.session_state.correction = correction_dictee(dictee, dictee_user)
+        
+        if 'correction' in st.session_state:
+            st.markdown("Voici la correction :")
+            st.markdown(st.session_state.correction)
+
+    elif mode.startswith("Entrainer"):
+        st.text_area("Voici votre dictée :", dictee, height=300)