Update app.py

app.py

@@ -46,8 +46,8 @@ def dictee_to_audio_segmented(dictee):
             audio_urls.append(audio_path)
     return audio_urls
 
-def generer_dictee():
-    prompt = f"Créer une dictée pour la classe {st.session_state.Classe} d'une longueur d'environ {st.session_state.Longueur} mots. Il est important de créer le texte uniquement de la dictée et de ne pas ajouter de consignes ou d'indications supplémentaires."
+def generer_dictee(classe, longueur):
+    prompt = f"Créer une dictée pour la classe {classe} d'une longueur d'environ {longueur} mots. Il est important de créer le texte uniquement de la dictée et de ne pas ajouter de consignes ou d'indications supplémentaires."
     generate_kwargs = {
         "temperature": 0.7,
         "max_new_tokens": 1000,
@@ -63,8 +63,8 @@ def generer_dictee():
     dictee = dictee.replace("</s>", "").strip()
     return dictee
 
-def correction_dictee():
-    prompt = f"Voici une dictée crée: {st.session_state.dictée} | Voici la dictée faite par l'utilisateur : {st.session_state.Dictee_User} - Corrige la dictée en donnant les explications, utilise les syntax du markdown pour une meilleur comprehesion de la correction. Il est important de comparer la dictée de l'utilisateur avec uniquement celle crée."
+def correction_dictee(dictee_generated, dictee_user):
+    prompt = f"Voici une dictée crée: {dictee_generated} | Voici la dictée faite par l'utilisateur : {dictee_user} - Corrige la dictée en donnant les explications, utilise les syntax du markdown pour une meilleur comprehesion de la correction. Il est important de comparer la dictée de l'utilisateur avec uniquement celle crée."
     generate_kwargs = {
         "temperature": 0.7,
         "max_new_tokens": 2000,  # Ajustez selon la longueur attendue de la correction
@@ -97,61 +97,37 @@ def segmenter_texte(texte):
     sentences = re.split(r'(?<=[.!?]) +', texte)
     return sentences
 
-# Stocker les variable de session pour qu'elles persistent
-if 'dictée' not in st.session_state:
-    st.session_state.dictée = None
-if 'dictéegenerated' not in st.session_state:
-    st.session_state.dictéegenerated = "Hobby"
-if 'Mode' not in st.session_state:
-    st.session_state.Mode = None
-if 'Classe' not in st.session_state:
-    st.session_state.Classe = None
-if 'Longueur' not in st.session_state:
-    st.session_state.Longueur = None
-if 'dictée' not in st.session_state:
-    st.session_state.dictée = None
-if 'dictee_user' not in st.session_state:
-    st.session_state.Dictee_User = None
-
 # Streamlit App Interface
 st.set_page_config(layout="wide")
 st.title('Générateur de Dictée')
 
-if st.session_state.dictéegenerated == "Hobby":
-    with st.expander("Paramètres de la dictée", expanded=True):
-        st.session_state.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.session_state.Classe = st.selectbox("Classe", ["CP", "CE1", "CE2", "CM1", "CM2", "6ème", "5ème", "4ème", "3ème", "Seconde", "Premiere", "Terminale"], index=2)
-        st.session_state.Longueur = st.slider("Longueur de la dictée (nombre de mots)", 50, 500, 200)
-else:
-    if st.button('Retour'):
-        st.session_state.dictéegenerated = "Hobby"
-        st.rerun()
-        
-    
+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)
+
 if st.button('Générer la Dictée'):
-    st.session_state.dictéegenerated = "Generated"
     with st.spinner("Génération de la dictée en cours..."):
-        dictee = generer_dictee()
-        if st.session_state.Mode == "S'entrainer: Vous aurez uniquement les audios suivi d'une correction par IA (Pour 1 seul personne)":
+        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:
-                st.session_state.Dictee_User = st.text_input("Écrivez la dictée ici:")
+                dictee_user = st.text_input("Écrivez la dictée ici:")
                 if st.button('Correction'):
-                    with st.spinner("Génération de la dictée en cours..."):
-                        correction = correction_dictee()
+                    with st.spinner("Génération de la correction en cours..."):
+                        correction = correction_dictee(dictee, dictee_user)
                         st.markdown("Voici la correction :")
                         st.markdown(correction)
-                    
-        elif st.session_state.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)
+
+        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)