done

app.py

@@ -46,7 +46,7 @@ with col2:
             button = st.button("Prédire", use_container_width=True)
             if button:
                 predicted, confidence = predict(uploaded_file)
-                diagnosis_prompt = get_diagnosis(predicted, confidence)
+                diagnosis_prompt, label = get_diagnosis(predicted, confidence)
                 diagnosis = generate_text(diagnosis_prompt)
                 summary = summarize_text(diagnosis)
                 translation = translate_fr_to_es(summary)
@@ -59,6 +59,7 @@ if uploaded_file and translation:
             <div class="container card border-0 shadow-sm p-4 mt-2">
                 <div class="card-body">
                     <h2 class="gradient-text"> Resultat de la prédiction </h2>
+                    <h4 class="gradient-text text-underline"> Type de tumeur: {label} </h3>
                     <p class="lead"> {summary} </p>
                 </div>
             </div>
@@ -71,6 +72,7 @@ if uploaded_file and translation:
             """,
             unsafe_allow_html=True, 
         )
+        st.audio(output_path, format="audio/mp3")
 
     st.markdown(
         f"""
@@ -83,9 +85,7 @@ if uploaded_file and translation:
               """,
         unsafe_allow_html=True,
     )
-    _, col2, _ = st.columns([2, 3, 2])
-    with col2:
-        st.audio(output_path, format="audio/mp3")
+
 
 # Afficher le pied de page
 show_footer(linkedin_image_base64, github_image_base64, huggingface_image_base64)

functions.py

@@ -23,8 +23,8 @@ def summarize_text(text):
 def get_diagnosis(prediction, confidence):
     labels = ["glioma", "meningioma", "notumor", "pituitary"]
     result = labels[prediction]
-    prompt = f'Tu es un médecin qui s\'appelle Skym qui doit annoncer à un patient X le type de tumeur qu\'il a. Le type de tumeur prédit est "{result}" avec une confiance de {confidence}. Veuillez fournir un diagnostic concis, précis et des conseils.'
-    return prompt
+    prompt = f'Tu es un médecin (neurologue) qui s\'appelle Skym  qui doit annoncer à un patient X le type de tumeur qu\'il a. Le type de tumeur prédit est "{result}" avec une confiance de {confidence}. Veuillez fournir un diagnostic concis, précis et des conseils.'
+    return prompt, result
 
 
 def generate_text(input_text, max_retries=3):