Update app.py

app.py

@@ -98,41 +98,88 @@ def segmenter_texte(texte):
     return sentences
 
 # Streamlit App Interface
-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.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:
         st.session_state.dictee = generer_dictee(classe, longueur)
-        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()
-
+    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:
+                    st.session_state.correction = correction_dictee(dictee, dictee_user)
+                    st.rerun()
+
+        if 'correction' in st.session_state:
+            st.divider()
+            st.markdown("### 🎉 Voici la correction (*Par IA*) :")
+            st.markdown(st.session_state.correction)
+            if st.button("En faire une nouvelle"):
+                del st.session_state['expandedmodified']
+                del st.session_state['dictee']
+                del st.session_state['audio_urls']
+                del st.session_state['concatenated_audio_path']
+                st.rerun()
+
+    elif mode.startswith("Entrainer"):
+        st.markdown("### 📚 Voici la dictée :")
+        st.markdown(dictee)
+        if st.button("En faire une nouvelle"):
+            del st.session_state['expandedmodified']
+            del st.session_state['dictee']
+            st.rerun()