Update app.py

app.py

@@ -22,9 +22,9 @@ st.markdown("""
     }
 
     .stImage > img {
-        width: 80% !important;
+        width: 70% !important;
         height: auto !important;
-        max-height: 200px !important;
+        max-height: 150px !important;
         object-fit: contain !important;
     }
 
@@ -70,6 +70,15 @@ def draw_boxes(image, predictions):
             width=2
         )
 
+        # Ajouter des points de "chaleur" aux fractures détectées
+        center_x = (box['xmin'] + box['xmax']) / 2
+        center_y = (box['ymin'] + box['ymax']) / 2
+        radius = 5
+        draw.ellipse(
+            [(center_x - radius, center_y - radius), (center_x + radius, center_y + radius)],
+            fill=color
+        )
+
         # Label plus compact
         text_bbox = draw.textbbox((box['xmin'], box['ymin']-15), label)
         draw.rectangle(text_bbox, fill=color)
@@ -97,47 +106,43 @@ def main():
         if uploaded_file:
             image = Image.open(uploaded_file)
 
-            st.markdown("### 🔍 Meinung der KI-Experten")
+            # Toujours afficher les résultats des autres modèles
+            st.markdown("### 🎯 KI-Analyse")
+
+            st.markdown("**🛡️ Der KnochenWächter**")
+            predictions_wachter = models["KnochenWächter"](image)
+            for pred in predictions_wachter:
+                score_color = "#22c55e" if pred['score'] > 0.7 else "#eab308"
+                st.markdown(f"""
+                    <div class='result-box'>
+                        <span style='color: {score_color}; font-weight: 500;'>
+                            {pred['score']:.1%}
+                        </span> - {translate_label(pred['label'])}
+                    </div>
+                """, unsafe_allow_html=True)
+
+            st.markdown("**🎓 Der RöntgenMeister**")
+            predictions_meister = models["RöntgenMeister"](image)
+            for pred in predictions_meister:
+                score_color = "#22c55e" if pred['score'] > 0.7 else "#eab308"
+                st.markdown(f"""
+                    <div class='result-box'>
+                        <span style='color: {score_color}; font-weight: 500;'>
+                            {pred['score']:.1%}
+                        </span> - {translate_label(pred['label'])}
+                    </div>
+                """, unsafe_allow_html=True)
 
             # Analyse avec KnochenAuge (localisation)
-            st.markdown("#### 👁️ Das KnochenAuge - Lokalisation")
-            predictions = models["KnochenAuge"](image)
-            filtered_preds = [p for p in predictions if p['score'] >= conf_threshold]
+            predictions_auge = models["KnochenAuge"](image)
+            filtered_preds = [p for p in predictions_auge if p['score'] >= conf_threshold]
 
             if filtered_preds:
+                st.markdown("#### 👁️ Das KnochenAuge - Lokalisation")
                 result_image = image.copy()
                 result_image = draw_boxes(result_image, filtered_preds)
                 st.image(result_image, use_container_width=True)
 
-            # Toujours afficher les résultats des autres modèles
-            st.markdown("#### 🎯 KI-Analyse")
-            col_left, col_right = st.columns(2)
-
-            with col_left:
-                st.markdown("**🛡️ Der KnochenWächter**")
-                predictions = models["KnochenWächter"](image)
-                for pred in predictions:
-                    score_color = "#22c55e" if pred['score'] > 0.7 else "#eab308"
-                    st.markdown(f"""
-                        <div class='result-box'>
-                            <span style='color: {score_color}; font-weight: 500;'>
-                                {pred['score']:.1%}
-                            </span> - {translate_label(pred['label'])}
-                        </div>
-                    """, unsafe_allow_html=True)
-
-            with col_right:
-                st.markdown("**🎓 Der RöntgenMeister**")
-                predictions = models["RöntgenMeister"](image)
-                for pred in predictions:
-                    score_color = "#22c55e" if pred['score'] > 0.7 else "#eab308"
-                    st.markdown(f"""
-                        <div class='result-box'>
-                            <span style='color: {score_color}; font-weight: 500;'>
-                                {pred['score']:.1%}
-                            </span> - {translate_label(pred['label'])}
-                        </div>
-                    """, unsafe_allow_html=True)
         else:
             st.info("Bitte laden Sie ein Röntgenbild hoch (JPEG, PNG)")