Update app.py

app.py

@@ -10,7 +10,6 @@ def find_model_file(filename="generator.h5"):
     for root, dirs, files in os.walk("/"):
         if filename in files:
             model_path = os.path.join(root, filename)
-            print(f"Modèle trouvé à : {model_path}")
             return model_path
     return None
 
@@ -20,8 +19,7 @@ def load_model_safe(model_path):
         # Charger le modèle sans compilation pour éviter des erreurs liées au batch_shape
         model = tf.keras.models.load_model(model_path, compile=False)
         return model
-    except ValueError as e:
-        print(f"Erreur lors du chargement du modèle: {e}")
+    except ValueError:
         return None
 
 # Chercher le modèle 'generator.h5' dans le système
@@ -30,28 +28,27 @@ model_path = find_model_file("generator.h5")
 # Vérifier si le modèle est trouvé et charger
 if model_path is not None:
     generator = load_model_safe(model_path)
-    print(f"Modèle chargé depuis : {model_path}")
 else:
-    print("Le modèle 'generator.h5' n'a pas été trouvé sur le serveur.")
     generator = None
 
 # Fonction pour générer une vidéo à partir du générateur
 def generate_video():
     if generator is None:
-        return "Le modèle n'a pas pu être chargé."
+        # Retourner un fichier d'erreur générique si le modèle n'est pas trouvé
+        error_file_path = "/content/error_file.txt"
+        
+        with open(error_file_path, "w") as f:
+            f.write("Le modèle n'a pas pu être chargé.")
+        
+        # Retourner le fichier d'erreur
+        return error_file_path
 
     # Générer un bruit aléatoire (entrée pour le générateur)
     noise = np.random.normal(0, 1, (1, 16, 64, 64, 3))  # Exemple de bruit pour 16 frames de 64x64x3
     generated_video = generator.predict(noise)  # Générer la vidéo
 
     # Vérifier la forme des données générées
-    print(f"Shape of generated video: {generated_video.shape}")
-    
-    # Normaliser les données générées
-    video = (generated_video[0] * 255).astype(np.uint8)  # Convertir en entier 8 bits
-
-    # Vérifier si la vidéo générée a la bonne forme
-    if len(video.shape) != 4 or video.shape[0] != 16:
+    if len(generated_video.shape) != 4 or generated_video.shape[0] != 16:
         return "Erreur dans les dimensions de la vidéo générée."
 
     # Créer le répertoire pour la vidéo
@@ -61,11 +58,11 @@ def generate_video():
     
     # Enregistrer la vidéo avec OpenCV
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Codec vidéo
-    height, width, _ = video[0].shape
+    height, width, _ = generated_video[0].shape
     out = cv2.VideoWriter(filename, fourcc, 15, (width, height))  # 15 FPS
 
     # Ajouter les frames dans le fichier vidéo
-    for frame in video:
+    for frame in generated_video:
         out.write(frame)
     out.release()