Update app.py

app.py

@@ -214,33 +214,35 @@ def main():
                 with col1:
                     st.write("#### 🤖 KI-Diagnose")
                     
-                    st.write("##### 🛡️ KnochenWächter")
+                    st.markdown("#### 🛡️ KnochenWächter")
+                    # Afficher tous les résultats de KnochenWächter
                     for pred in predictions_watcher:
-                        if pred['score'] >= conf_threshold:
-                            confidence_color = '#0066cc' if pred['score'] > 0.7 else '#ffa500'
-                            label_lower = pred['label'].lower()
-                            if 'fracture' in label_lower:
-                                has_fracture = True
-                                max_fracture_score = max(max_fracture_score, pred['score'])
-                            st.markdown(f"""
-                                <div class="result-box" style="color: #1a1a1a;">
-                                    <span style="color: {confidence_color}; font-weight: 500;">
-                                        {pred['score']:.1%}
-                                    </span> - {translate_label(pred['label'])}
-                                </div>
-                            """, unsafe_allow_html=True)
+                        confidence_color = '#0066cc' if pred['score'] > 0.7 else '#ffa500'
+                        label_lower = pred['label'].lower()
+                        # Mettre à jour max_fracture_score seulement pour les fractures
+                        if pred['score'] >= conf_threshold and 'fracture' in label_lower:
+                            has_fracture = True
+                            max_fracture_score = max(max_fracture_score, pred['score'])
+                        # Afficher tous les résultats
+                        st.markdown(f"""
+                            <div class="result-box" style="color: #1a1a1a;">
+                                <span style="color: {confidence_color}; font-weight: 500;">
+                                    {pred['score']:.1%}
+                                </span> - {translate_label(pred['label'])}
+                            </div>
+                        """, unsafe_allow_html=True)
                     
-                    st.write("#### 🎓 RöntgenMeister")
+                    st.markdown("#### 🎓 RöntgenMeister")
+                    # Afficher tous les résultats de RöntgenMeister
                     for pred in predictions_master:
-                        if pred['score'] >= conf_threshold:
-                            confidence_color = '#0066cc' if pred['score'] > 0.7 else '#ffa500'
-                            st.markdown(f"""
-                                <div class="result-box" style="color: #1a1a1a;">
-                                    <span style="color: {confidence_color}; font-weight: 500;">
-                                        {pred['score']:.1%}
-                                    </span> - {translate_label(pred['label'])}
-                                </div>
-                            """, unsafe_allow_html=True)
+                        confidence_color = '#0066cc' if pred['score'] > 0.7 else '#ffa500'
+                        st.markdown(f"""
+                            <div class="result-box" style="color: #1a1a1a;">
+                                <span style="color: {confidence_color}; font-weight: 500;">
+                                    {pred['score']:.1%}
+                                </span> - {translate_label(pred['label'])}
+                            </div>
+                        """, unsafe_allow_html=True)
                     
                     if max_fracture_score > 0:
                         st.write("#### 📊 Wahrscheinlichkeit")