Update app.py

app.py

@@ -103,6 +103,8 @@ st.title('🎓 Entrainement de Dictée')
 
 if 'expanded' not in st.session_state:
     st.session_state.expanded = True
+if 'dicteecreation' not in st.session_state:
+    st.session_state.dicteecreation = False #Inversed
 with st.expander("📝 Génération de la dictée", expanded=st.session_state.expanded):
     with st.form("dictation_form"):
         st.markdown("### 🚀 Choisissez votre mode de dictée")
@@ -111,33 +113,21 @@ with st.expander("📝 Génération de la dictée", expanded=st.session_state.ex
         classe = st.selectbox("Classe", ["CP", "CE1", "CE2", "CM1", "CM2", "6ème", "5ème", "4ème", "3ème", "Seconde", "Premiere", "Terminale"], index=2)
         st.markdown("### 📏 Définissez la longueur de la dictée")
         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")
+        submitted = st.form_submit_button("🔮 Générer la Dictée", disabled=st.session_state.dicteecreation)
 
 if submitted or 'dictee' in st.session_state:
-    # Delete all
-    if 'expandedmodified' in st.session_state:
-        del st.session_state['expandedmodified']
-    if 'dictee' in st.session_state:
-        del st.session_state['dictee']
-    if 'audio_urls' in st.session_state:
-        del st.session_state['audio_urls']
-    if 'concatenated_audio_path' in st.session_state:
-        del st.session_state['concatenated_audio_path']
-
-    # Creation
-    if 'dictee' not in st.session_state:
-        st.session_state.dictee = generer_dictee(classe, longueur)
-    if 'expandedmodified' not in st.session_state:
-        st.session_state.expandedmodified = False
-    dictee = st.session_state.dictee
-    st.session_state.expanded = False
-    st.divider()
     with st.spinner("🚀 Dictée en cours de création..."):
-        if not st.session_state.expandedmodified:
+        if st.session_state.creationmodified == False:
             st.session_state.expandedmodified = True
+            st.session_state.dicteecreation = True
+
+            if 'dictee' not in st.session_state:
+                st.session_state.dictee = generer_dictee(classe, longueur)
+            dictee = st.session_state.dictee
+            
+            st.session_state.creationmodified = True
             st.rerun()
-    del st.session_state['expandedmodified']
-    
+                
     if mode.startswith("S'entrainer"):
         if 'audio_urls' not in st.session_state:
             with st.spinner("🔊 Préparation des audios..."):
@@ -185,6 +175,8 @@ if submitted or 'dictee' in st.session_state:
                 del st.session_state['dictee']
                 del st.session_state['audio_urls']
                 del st.session_state['concatenated_audio_path']
+                st.session_state.dicteecreation = False
+                st.session_state.creationmodified = False
                 st.rerun()
 
     elif mode.startswith("Entrainer"):
@@ -193,4 +185,6 @@ if submitted or 'dictee' in st.session_state:
         if st.button("En faire une nouvelle"):
             del st.session_state['expandedmodified']
             del st.session_state['dictee']
+            st.session_state.dicteecreation = False
+            st.session_state.creationmodified = False
             st.rerun()