Update app.py

app.py

@@ -212,44 +212,32 @@ if stroke_file and st.button("Analyser l'IRM Stroke", key="stroke_button"):
     else:
         st.image(result_path_stroke, caption="Stroke annotée", use_container_width=True)
 
-# ---------------- Upload IRM 3D (NIfTI) ----------------
-st.header("🧠 Analyse IRM 3D (NIfTI)")
 
-irm3d_files = st.file_uploader(
-    "Uploader les 4 fichiers IRM (FLAIR, T1, T1CE, T2)",
-    type=["nii", "nii.gz"],
-    accept_multiple_files=True,
-    key="irm3d_uploader"
-)
-
-if irm3d_files and st.button("Analyser l'IRM 3D", key="irm3d_button"):
+# Upload IRM 3D (cancer)
+st.header("🧠 Détection Tumeur (IRM 3D)")
+irm3d_files = st.file_uploader("Uploader 4 séquences (FLAIR, T1, T1CE, T2)", type=["nii", "nii.gz"], accept_multiple_files=True)
+if irm3d_files and st.button("Analyser IRM 3D"):
     if len(irm3d_files) != 4:
         st.error("⚠️ Merci d’uploader exactement 4 fichiers IRM (FLAIR, T1, T1CE, T2)")
     else:
         tmp_paths = []
         for f in irm3d_files:
-            tmp_path = os.path.join(SAVE_DIR, f.name)
-            with open(tmp_path, "wb") as out:
+            path = os.path.join(SAVE_DIR, f.name)
+            with open(path, "wb") as out:
                 out.write(f.read())
-            tmp_paths.append(tmp_path)
-
+            tmp_paths.append(path)
         seg, report_text, (nii_path, report_path, mask_path) = irm_cancer_module.run(tmp_paths)
-
-        # Affichage rapport
-        st.subheader("📄 Rapport automatique")
+        st.subheader("📝 Rapport automatique")
         st.text(report_text)
-
-        # Affichage masque segmenté
         if mask_path and os.path.exists(mask_path):
-            st.image(mask_path, caption="Masque segmenté (vue 2D)", use_container_width=True)
+            st.image(mask_path, caption="Segmentation annotée", use_container_width=True)
 
-# ---------------- Disclaimer ----------------
+# Disclaimer
 st.markdown(f"""
 ---
 👨‍💻 **Badsi Djilali** — Ingénieur Deep Learning  
 🚀 Créateur de **Stroke_IA_Detection**  
-🧠 (Détection d'asymétrie faciale & AVC par IA)
 
-⚠️ **Disclaimer :** Stroke-IA est une démo technique, pas un avis médical.  
+⚠️ Démo technique, pas un avis médical.  
 © {datetime.now().year} — Badsi Djilali.
 """)