Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 from transformers import pipeline
 from PIL import Image, ImageDraw
-import torch
 
 st.set_page_config(
     page_title="Fraktur Detektion",
@@ -11,86 +10,31 @@ st.set_page_config(
 
 st.markdown("""
 <style>
-    /* Reset et base */
     .stApp {
-        background-color: var(--background-color) !important;
         padding: 0 !important;
-        overflow: hidden !important;
         height: 100vh !important;
+        overflow: hidden !important;
     }
 
-    /* Variables de thème */
-    [data-theme="light"] {
-        --background-color: #ffffff;
-        --text-color: #1f2937;
-        --border-color: #e5e7eb;
-        --secondary-bg: #f3f4f6;
-    }
-
-    [data-theme="dark"] {
-        --background-color: #1f2937;
-        --text-color: #f3f4f6;
-        --border-color: #4b5563;
-        --secondary-bg: #374151;
-    }
-
-    /* Layout principal */
     .block-container {
-        padding: 0.5rem !important;
+        padding: 0.25rem !important;
         max-width: 100% !important;
     }
 
-    /* Contrôles et upload */
-    .uploadedFile {
-        border: 1px dashed var(--border-color);
-        border-radius: 0.375rem;
-        padding: 0.25rem;
-        background: var(--secondary-bg);
-    }
-
-    /* Ajustement des colonnes */
-    [data-testid="column"] {
-        padding: 0 0.5rem !important;
-    }
-
-    /* Images adaptatives */
     .stImage > img {
-        width: 100% !important;
+        width: 80% !important;
         height: auto !important;
-        max-height: 400px !important;
+        max-height: 200px !important;
         object-fit: contain !important;
     }
 
-    /* Résultats */
-    .result-box {
-        padding: 0.375rem;
-        border-radius: 0.375rem;
-        margin: 0.25rem 0;
-        background: var(--secondary-bg);
-        border: 1px solid var(--border-color);
-        color: var(--text-color);
-    }
-
-    /* Titres */
     h2, h3 {
-        margin: 0 !important;
-        padding: 0.5rem 0 !important;
-        font-size: 1rem !important;
-        color: var(--text-color) !important;
-    }
-
-    /* Nettoyage des éléments inutiles */
-    #MainMenu, footer, header, .viewerBadge_container__1QSob, .stDeployButton {
-        display: none !important;
+        font-size: 0.9rem !important;
     }
 
-    /* Ajustements espacement */
-    div[data-testid="stVerticalBlock"] {
-        gap: 0.5rem !important;
-    }
-
-    .element-container {
-        margin: 0.25rem 0 !important;
+    .result-box {
+        font-size: 0.8rem !important;
+        margin: 0.2rem 0 !important;
     }
 </style>
 """, unsafe_allow_html=True)
@@ -165,58 +109,37 @@ def main():
                 result_image = draw_boxes(result_image, filtered_preds)
                 st.image(result_image, use_container_width=True)
 
-                # Affichage des résultats supplémentaires uniquement si des fractures sont détectées
-                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:
-                        if pred['score'] >= conf_threshold:
-                            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:
-                        if pred['score'] >= conf_threshold:
-                            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("Kein Bruch erkannt.")
+            # 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)")
 
-    # Script pour la synchronisation du thème
-    st.markdown("""
-        <script>
-            function updateTheme(isDark) {
-                document.documentElement.setAttribute('data-theme', isDark ? 'dark' : 'light');
-            }
-
-            window.addEventListener('message', function(e) {
-                if (e.data.type === 'theme-change') {
-                    updateTheme(e.data.theme === 'dark');
-                }
-            });
-
-            updateTheme(window.matchMedia('(prefers-color-scheme: dark)').matches);
-        </script>
-    """, unsafe_allow_html=True)
-
 if __name__ == "__main__":
     main()