Update app.py

app.py

@@ -1,33 +1,40 @@
 import os
-import sys
+import subprocess
 import gradio as gr
-import torch
-from deoldify.visualize import get_video_colorizer
+from deoldify.visualize import get_image_colorizer, 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')
+# Vérifie si le dépôt DeOldify existe, sinon le clone
+if not os.path.exists("DeOldify"):
+    subprocess.run(["git", "clone", "https://github.com/jantic/DeOldify.git"])
+    
+# Change le répertoire de travail
+os.chdir("DeOldify")
 
-# Charger le modèle DeOldify pré-entraîné
+# Crée un fichier requirements.txt
+with open("requirements.txt", "w") as f:
+    f.write("\ndeoldify\ngradio\ntorch\nopencv-python\n")
+
+# Installe les dépendances
+subprocess.run(["pip", "install", "-r", "requirements.txt"])
+
+# Initialiser les colorisateurs
+image_colorizer = get_image_colorizer()
 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
+def colorize_image(input_image):
+    # Processer l'image d'entrée et retourner la version colorisée
+    colorized_image = image_colorizer.get_transformed_image(input_image)
+    return colorized_image
 
-# Interface Gradio
-def process_video(video):
-    colorized_video = colorize_video(video.name)
+def colorize_video(input_video):
+    # Processer la vidéo d'entrée et retourner la version colorisée
+    colorized_video = video_colorizer.get_transformed_video(input_video)
     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.")
+# Créer l'interface Gradio
+image_interface = gr.Interface(fn=colorize_image, inputs="image", outputs="image", title="Colorisation d'Image")
+video_interface = gr.Interface(fn=colorize_video, inputs="video", outputs="video", title="Colorisation de Vidéo")
 
-if __name__ == "__main__":
-    interface.launch()
+# Lancer les interfaces
+image_interface.launch(share=True)
+video_interface.launch(share=True)