Update app.py

app.py

@@ -64,10 +64,28 @@ def generer_dictee(classe, longueur):
     dictee = dictee.replace("</s>", "").strip()
     return dictee
 
+def correction_dictee(dictee, dictee_utilisateur):
+    prompt = f"Voici une dictée crée: {dictee} | Voici la dictée faite par l'utilisateur : {dictee_utilisateur} - Corrige la dictée en donnant les explications, utilise les syntax du markdown pour une meilleur comprehesion de la correction."
+    generate_kwargs = {
+        "temperature": 0.7,
+        "max_new_tokens": 2000,  # Ajustez selon la longueur attendue de la correction
+        "top_p": 0.95,
+        "repetition_penalty": 1.2,
+        "do_sample": True,
+    }
+    formatted_prompt = f"<s>[INST] {prompt} [/INST]"
+    stream = client_hf.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
+    texte_ameliore = ""
+    for response in stream:
+        texte_ameliore += response.token.text
+    texte_ameliore = texte_ameliore.replace("</s>", "").strip()
+    return correction
+
+
 def replace_punctuation(text):
     replacements = {
-        ".": " point,",
-        ",": " virgule.",
+        ".": " point.",
+        ",": " virgule,",
         ";": " point-virgule;",
         ":": " deux-points:",
         "!": " point d'exclamation!",
@@ -109,11 +127,11 @@ if st.button('Générer la Dictée'):
                         st.audio(url, format='audio/wav')
             
             with col2:
-                user_input = st.text_area("Écrivez la dictée ici:", height=300)
+                dictee_utilisateur = st.text_area("Écrivez la dictée ici:", height=300)
                 if st.button('Correction'):
                     st.write("Dictée originale:")
-                    st.text(dictee)
-                    # Add more detailed comparison/correction logic here as needed
+                    correction = correction_dictee(dictee, dictee_utilisateur)
+                    st.text_area("Voici la correction :", correction, height=500)
                     
         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)