Update app.py

app.py

@@ -98,76 +98,41 @@ def segmenter_texte(texte):
     return sentences
 
 # Streamlit App Interface
-st.set_page_config(layout="wide")
-st.title('🎓 Entrainement de Dictée')
-
-if 'expanded' not in st.session_state:
-    st.session_state.expanded = True
-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")
-        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)"])
-        st.markdown("### 🎒 Sélectionnez la classe")
-        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")
-
-if submitted or 'dictee' in st.session_state:
-    if 'dictee' not in st.session_state:
+def main():
+    if 'params' not in st.session_state:
+        st.session_state.params = {'classe': 'CM2', 'longueur': 200, 'mode': 'S\'entrainer'}
+
+    classe = st.selectbox("Classe", ["CP", "CE1", "CE2", "CM1", "CM2", "6ème", "5ème", "4ème", "3ème", "Seconde", "Premiere", "Terminale"], index=2, key="classe")
+    longueur = st.slider("Longueur de la dictée (nombre de mots)", 50, 500, 200, key="longueur")
+    mode = st.radio("Mode:", ["S'entrainer", "Entrainer"], key="mode")
+
+    st.session_state.params = {'classe': classe, 'longueur': longueur, 'mode': mode}
+
+    if st.button("Générer la dictée"):
         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:
-            st.session_state.expandedmodified = 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..."):
-                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:
-            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
-        
-        col1, col2 = st.columns(2)
-        with col1:
-            st.markdown("## 📖 Dictée en entier")
-            st.audio(concatenated_audio_path, format='audio/wav', start_time=0)
-            st.divider()
-            st.markdown("## 📖 Phrases de la Dictée")
-            with st.expander("Cliquez ici pour ouvrir"):
-                cols_per_row = 2
-                rows = (len(audio_urls) + cols_per_row - 1) // cols_per_row  # Arrondir au nombre supérieur
-                for i in range(rows):
-                    cols = st.columns(cols_per_row)
-                    for j in range(cols_per_row):
-                        idx = i * cols_per_row + j
-                        if idx < len(audio_urls):
-                            with cols[j]:
-                                st.markdown(f"**Phrase {idx + 1}:**")
-                                st.audio(audio_urls[idx], format='audio/wav')
-        
-        with col2:
-            st.markdown("## ✍️ Votre Dictée")
-            with st.form("dictee_form"):
-                dictee_user = st.text_area("Écrivez la dictée ici:", key="dictee_user", height=350)
-                correct = st.form_submit_button("📝 Correction")
-        
-        if correct or 'correction' in st.session_state:
-            st.session_state.correction = correction_dictee(dictee, dictee_user)
-            st.divider()
-            st.markdown("### 🎉 Voici la correction (*Par IA*) :")
-            st.markdown(st.session_state.correction)
-
-    elif mode.startswith("Entrainer"):
-        st.markdown("### 📚 Voici la dictée :")
-        st.markdown(dictee)
+        st.session_state.dictee_user = ""
+        st.session_state.correction = ""
+
+    if 'dictee' in st.session_state:
+        if mode == "S'entrainer":
+            # Afficher la dictée et le formulaire pour saisir la dictée de l'utilisateur
+            st.markdown("## 📖 Dictée")
+            st.markdown(st.session_state.dictee)
+
+            dictee_user = st.text_area("Écrivez la dictée ici:", key="dictee_user", height=350)
+
+            if st.button("Correction"):
+                st.session_state.correction = correction_dictee(st.session_state.dictee, dictee_user)
+
+            if 'correction' in st.session_state:
+                st.markdown("## 🎉 Correction")
+                st.markdown(st.session_state.correction)
+
+        elif mode == "Entrainer":
+            # Afficher uniquement la dictée
+            st.markdown("## 📖 Dictée")
+            st.markdown(st.session_state.dictee)
+
+if __name__ == "__main__":
+    main()