Update app.py

app.py

@@ -9,6 +9,7 @@ st.set_page_config(
     initial_sidebar_state="collapsed"
 )
 
+# Configuration du style personnalisé
 st.markdown("""
 <style>
     .stApp {
@@ -33,13 +34,15 @@ st.markdown("""
     }
     
     .stButton > button {
+        width: 100% !important;
         background-color: var(--button-color) !important;
         color: white !important;
         border: none !important;
-        padding: 0.5rem 1rem !important;
+        padding: 0.75rem 1.5rem !important;
         border-radius: 0.375rem !important;
         font-weight: 500 !important;
-        transition: background-color 0.2s !important;
+        margin: 1rem 0 !important;
+        cursor: pointer !important;
     }
     
     .stButton > button:hover {
@@ -82,14 +85,14 @@ st.markdown("""
         padding: 0.25rem;
     }
     
-    .row-widget.stButton {
-        text-align: center;
-        margin: 1rem 0;
-    }
-    
     div[data-testid="stFileUploader"] {
         width: 100%;
     }
+
+    /* Cache le message d'upload par défaut */
+    .uploadedFile small {
+        display: none !important;
+    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -125,57 +128,62 @@ def draw_boxes(image, predictions):
     return image
 
 def main():
-    models = load_models()
-    
-    st.markdown("### 📤 Röntgenbilder Upload")
-    
-    # File uploader avec label caché
-    uploaded_files = st.file_uploader(
-        "Wählen Sie Röntgenbilder aus",
-        type=['png', 'jpg', 'jpeg'],
-        accept_multiple_files=True,
-        label_visibility="collapsed"
-    )
-    
-    if uploaded_files:
-        # Bouton d'analyse
-        if st.button("Bilder analysieren", key="analyze_button"):
-            col1, col2 = st.columns([1, 1])
+    # Initialisation de la session state si nécessaire
+    if 'models_loaded' not in st.session_state:
+        st.session_state.models_loaded = False
+        st.session_state.models = load_models()
+        st.session_state.models_loaded = True
+
+    # Créer deux colonnes pour l'upload et le bouton
+    col_upload, col_button = st.columns([3, 1])
+
+    with col_upload:
+        uploaded_files = st.file_uploader(
+            "Röntgenbilder hochladen",
+            type=['png', 'jpg', 'jpeg'],
+            accept_multiple_files=True,
+            label_visibility="collapsed"
+        )
+
+    # Bouton d'analyse dans la deuxième colonne
+    with col_button:
+        analyze_clicked = st.button("📋 Analysieren", key="analyze_btn", disabled=not uploaded_files)
+
+    # Analyse des images si le bouton est cliqué
+    if uploaded_files and analyze_clicked:
+        col1, col2 = st.columns([1, 1])
+        
+        for idx, uploaded_file in enumerate(uploaded_files):
+            image = Image.open(uploaded_file)
+            
+            # Analyse avec KnochenAuge (localisierung)
+            predictions = st.session_state.models["KnochenAuge"](image)
+            fractures_found = any(p['label'].lower() == 'fracture' and p['score'] > 0.6 for p in predictions)
             
-            for idx, uploaded_file in enumerate(uploaded_files):
-                image = Image.open(uploaded_file)
-                
-                # Analyse avec KnochenAuge (localisierung)
-                predictions = models["KnochenAuge"](image)
-                fractures_found = any(p['label'].lower() == 'fracture' and p['score'] > 0.6 for p in predictions)
-                
-                # Afficher uniquement si des fractures sont détectées
-                if fractures_found:
-                    with col1 if idx % 2 == 0 else col2:
-                        result_image = image.copy()
-                        result_image = draw_boxes(result_image, predictions)
-                        st.image(result_image, caption=f"Bild {idx + 1}", use_column_width=True)
-                        
-                        # Analyse KnochenWächter et RöntgenMeister
-                        pred_wachter = models["KnochenWächter"](image)[0]
-                        pred_meister = models["RöntgenMeister"](image)[0]
-                        
-                        if pred_wachter['score'] > 0.6 or pred_meister['score'] > 0.6:
-                            st.markdown(f"""
-                                <div class='result-box'>
-                                    <span style='color: #2563eb'>KnochenWächter:</span> {pred_wachter['score']:.1%}<br>
-                                    <span style='color: #2563eb'>RöntgenMeister:</span> {pred_meister['score']:.1%}
-                                </div>
-                            """, unsafe_allow_html=True)
+            # Afficher uniquement si des fractures sont détectées
+            if fractures_found:
+                with col1 if idx % 2 == 0 else col2:
+                    result_image = image.copy()
+                    result_image = draw_boxes(result_image, predictions)
+                    st.image(result_image, caption=f"Bild {idx + 1}", use_column_width=True)
+                    
+                    # Analyse KnochenWächter et RöntgenMeister
+                    pred_wachter = st.session_state.models["KnochenWächter"](image)[0]
+                    pred_meister = st.session_state.models["RöntgenMeister"](image)[0]
+                    
+                    if pred_wachter['score'] > 0.6 or pred_meister['score'] > 0.6:
+                        st.markdown(f"""
+                            <div class='result-box'>
+                                <span style='color: #2563eb'>KnochenWächter:</span> {pred_wachter['score']:.1%}<br>
+                                <span style='color: #2563eb'>RöntgenMeister:</span> {pred_meister['score']:.1%}
+                            </div>
+                        """, unsafe_allow_html=True)
 
-    # Script pour la synchronisation du thème
+# Script pour la synchronisation du thème
     st.markdown("""
         <script>
-            // Fonction pour mettre à jour le thème
             function updateTheme(isDark) {
                 document.documentElement.setAttribute('data-theme', isDark ? 'dark' : 'light');
-                
-                // Mise à jour des styles en fonction du thème
                 const root = document.documentElement;
                 if (isDark) {
                     root.style.setProperty('--background-color', '#1f2937');
@@ -188,14 +196,12 @@ def main():
                 }
             }
             
-            // Écouter les messages du parent
             window.addEventListener('message', function(e) {
                 if (e.data.type === 'theme-change') {
                     updateTheme(e.data.theme === 'dark');
                 }
             });
             
-            // Thème initial basé sur les préférences système
             updateTheme(window.matchMedia('(prefers-color-scheme: dark)').matches);
         </script>
     """, unsafe_allow_html=True)