Update app.py

app.py

@@ -134,25 +134,26 @@ with st.expander("📝 Génération de la dictée", expanded=st.session_state.ex
         longueur = st.slider("Longueur de la dictée (nombre de mots)", 50, 500, 200)
         submitted = st.form_submit_button("🔮 Générer la Dictée", disabled=st.session_state.dicteecreation)
 
-if submitted or 'dictee' in st.session_state:
+if submitted or st.session_state.dictee != None:
     with st.spinner("🚀 Dictée en cours de création..."):
         if st.session_state.creationmodified == False:
             st.session_state.expandedmodified = True
             st.session_state.dicteecreation = True
 
-            if 'dictee' not in st.session_state:
+            if 'dictee' != None:
                 st.session_state.dictee = generer_dictee(classe, longueur)
-            dictee = st.session_state.dictee
             
             st.session_state.creationmodified = True
             st.rerun()
+
+    dictee = st.session_state.dictee
                 
     if mode.startswith("S'entrainer"):
-        if 'audio_urls' not in st.session_state:
+        if 'audio_urls' != None:
             with st.spinner("🔊 Préparation des audios..."):
                 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:
+        if 'concatenated_audio_path' != None:
             with st.spinner("🎵 Assemblage de l'audio complet..."):
                 st.session_state.concatenated_audio_path = concatenate_audio(audio_urls)
         concatenated_audio_path = st.session_state.concatenated_audio_path
@@ -185,7 +186,7 @@ if submitted or 'dictee' in st.session_state:
                     st.session_state.correction = correction_dictee(dictee, dictee_user)
                     st.rerun()
 
-        if 'correction' in st.session_state:
+        if st.session_state.correction != None:
             st.divider()
             st.markdown("### 🎉 Voici la correction (*Par IA*) :")
             st.markdown(st.session_state.correction)