Update app.py

app.py

@@ -63,8 +63,8 @@ def generer_dictee(classe, longueur):
     dictee = dictee.replace("</s>", "").strip()
     return dictee
 
-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."
+def correction_dictee(dictee, dictee_user):
+    prompt = f"Voici une dictée crée: {dictee} | 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
@@ -149,12 +149,14 @@ if submitted or 'dictee' in st.session_state:
                     st.audio(url, format='audio/wav')
         
         with col2:
+            st.markdown("## ✍️ Votre Dictée")
             dictee_user = st.text_area("Écrivez la dictée ici:", key="dictee_user")
-            if st.button("Correction", key="submit_correction"):
-                st.session_state.correction = correction_dictee(dictee, dictee_user)
+            if st.button("📝 Correction", key="submit_correction"):
+                with st.spinner("🤖 Dictée en cours de correction..."):
+                    st.session_state.correction = correction_dictee(dictee, dictee_user)
         
-        if 'correction' in st.session_state:
-            st.markdown("### Voici la correction (*Par IA*) :")
+        if st.session_state.correction:
+            st.markdown("### 🎉 Voici la correction (*Par IA*) :")
             st.markdown(st.session_state.correction)
 
     elif mode.startswith("Entrainer"):