app.py

import os
import sys
import gradio as gr
import torch
from deoldify.visualize import get_video_colorizer

# Cloner le repo DeOldify et installer les dépendances si ce n'est pas déjà fait
if not os.path.exists('DeOldify'):
    os.system('git clone https://github.com/jantic/DeOldify.git')
    os.system('pip install -r DeOldify/requirements-colab.txt')

# Charger le modèle DeOldify pré-entraîné
video_colorizer = get_video_colorizer()

# Fonction pour coloriser une vidéo
def colorize_video(video_path):
    output_path = video_colorizer.colorize_from_file_name(video_path)
    return output_path

# Interface Gradio
def process_video(video):
    colorized_video = colorize_video(video.name)
    return colorized_video

# Interface Gradio pour uploader la vidéo
interface = gr.Interface(fn=process_video, 
                         inputs=gr.Video(label="Télécharger une vidéo en noir et blanc"), 
                         outputs=gr.Video(label="Vidéo colorisée"),
                         title="Colorisation de Vidéo avec DeOldify",
                         description="Téléchargez une vidéo en noir et blanc pour la coloriser.")

if __name__ == "__main__":
    interface.launch()