Update app.py

app.py

@@ -7,7 +7,7 @@ import seaborn as sns
 from sklearn.metrics import confusion_matrix, classification_report
 from utils.preprocess import load_and_preprocess_dataset
 from model.cnn_model import CNNModel
-from model.sca import CustomSCA
+from model.gwo import GWO  
 
 # Configuration de la page
 st.set_page_config(
@@ -190,17 +190,16 @@ with tab2:
 
         if 'X_train' in st.session_state:
             st.subheader("Optimisation des Hyperparamètres")
-
-            if st.button("⚡ Optimiser avec SCA", type="primary"):
-                with st.spinner("Optimisation en cours (peut prendre plusieurs minutes)..."):
+            if st.button("⚡ Optimiser avec GWO", type="primary"):  # Changer le texte du bouton
+                with st.spinner("Optimisation en cours (GWO)..."):
                     try:
-                        sca = CustomSCA(
+                        optimizer = GWO(
                             obj_func=objective_function,
                             bounds=bounds,
-                            population_size=8,
-                            max_iter=15
+                            population_size=10,  # Nombre de loups
+                            max_iter=15          # Itérations
                         )
-                        result = sca.optimize()
+                        result = optimizer.optimize()
 
                         st.session_state['best_params'] = {
                             'filter1': int(result['best_solution'][0]),
@@ -210,18 +209,18 @@ with tab2:
                             'dropout': result['best_solution'][4]
                         }
 
-                        st.success("Optimisation terminée!")
+                        st.success("Optimisation GWO terminée !")
                         st.json(st.session_state['best_params'])
-
-                        # Visualisation de la convergence
-                        st.subheader("Courbe de Convergence")
+        
+                        # Affichage de la courbe de convergence
+                        st.subheader("Courbe de Convergence (GWO)")
                         st.line_chart({
                             'Erreur Validation': result['convergence_curve'],
                             'Précision': [1 - x for x in result['convergence_curve']]
                         })
 
                     except Exception as e:
-                        st.error(f"Erreur lors de l'optimisation : {str(e)}")
+                        st.error(f"Erreur lors de l'optimisation GWO : {str(e)}")
 
 # Onglet 3: Résultats et export
 with tab3: