import tensorflow as tf
import numpy as np
import cv2
import gradio as gr
import os

# Fonction pour chercher le modèle dans tout le système de fichiers
def find_model_file(filename="generator.h5"):
    for root, dirs, files in os.walk("/"):
        if filename in files:
            return os.path.join(root, filename)
    return None

# Fonction pour charger le modèle en toute sécurité
def load_model_safe(model_path):
    try:
        model = tf.keras.models.load_model(model_path, compile=False)
        return model
    except ValueError:
        return None

# Chercher le modèle 'generator.h5' dans le système
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)
else:
    generator = None

# Fonction pour générer une vidéo à partir du générateur
def generate_video():
    if generator is None:
        # Créer un fichier d'erreur générique sans message personnalisé
        error_file_path = "/content/error_file.txt"
        with open(error_file_path, "w") as f:
            pass  # Crée un fichier vide
        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
    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
    output_dir = "/content/generated_videos"
    os.makedirs(output_dir, exist_ok=True)
    filename = os.path.join(output_dir, "generated_video.mp4")
    
    # Enregistrer la vidéo avec OpenCV
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Codec vidéo
    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 generated_video:
        out.write(frame)
    out.release()

    # Retourner le chemin de la vidéo générée
    return filename

# Interface Gradio
def interface_function():
    # Appeler la fonction de génération et retourner la vidéo générée
    video_file = generate_video()
    return video_file

# Définir l'interface avec Gradio
gr.Interface(
    fn=interface_function,  # Fonction à exécuter
    inputs=[],              # Pas d'entrée utilisateur
    outputs=gr.File(label="Vidéo générée"),  # Utiliser gr.File pour le téléchargement du fichier
    title="Générateur de Vidéos avec IA",
    description="Cliquez sur le bouton ci-dessous pour générer une vidéo aléatoire avec l'IA."
).launch(server_name="0.0.0.0", server_port=7860)