update app.py

app.py

@@ -1,11 +1,11 @@
 
+
 import matplotlib.pyplot as plt
 import io
 from PIL import Image
 import pickle
 import pandas as pd
 import gradio as gr
-from IPython.display import Video
 
 def generar_recomendacion(svd_model, user_id, df, genres, top=5):
     # Filtrar las películas que correspondan al usuario y a los géneros de interés
@@ -84,31 +84,24 @@ def wrap_generar_recomendacion(user_id, drama, comedy, horror, romance, top=5):
     # Llamar a la función de recomendación y devolver los resultados como una cadena y una imagen
     return generar_recomendacion(svd_model, user_id, dfmerge, genres, int(top))
 
-    # Ruta del video descargado
-video_path = "preview.mp4"
-
-# Genera una vista previa del video
-video_preview = Video(video_path, width=250)
-
-# Modificar la interfaz de Gradio para mostrar el video en el título
-
+# Modificar la interfaz de Gradio para mostrar una imagen también
 demo = gr.Interface(
     fn=wrap_generar_recomendacion,
     inputs=[gr.inputs.Number(label="User ID"), gr.inputs.Checkbox(label="Drama"), gr.inputs.Checkbox(label="Comedy"), gr.inputs.Checkbox(label="Horror"), gr.inputs.Checkbox(label="Romance"), gr.inputs.Number(label="Top")], 
     outputs=[gr.outputs.Textbox(), gr.outputs.Image(type='pil')],
     title = '<h1 style="text-align: center; color: #FF6347;">STREAMREC</h1>',
-    description = f"""
-    <p>
-    <center>
-    <font size="4" face="Arial" color="white">
-    Sistema de Recomendaciones Personalizadas de Películas y Series
-    </font>
-    <p><b style="color: #DC143C;">Advertencia: Ingresa el ID del usuario (user_id), selecciona los géneros de interés y la cantidad de recomendaciones que te gustaría generar.
-    Te mostraremos algunas películas que pueden gustarte.</b></p>
-    {video_preview}
-    </center>
-    </p>
-    """,
+    description = """
+<p>
+<center>
+<font size="4" face="Arial" color="white">
+Sistema de Recomendaciones Personalizadas de Películas y Series
+</font>
+<p><b style="color: #DC143C;">Advertencia: Ingresa el ID del usuario (user_id), selecciona los géneros de interés y la cantidad de recomendaciones que te gustaría generar.
+Te mostraremos algunas películas que pueden gustarte.</b></p>
+<img src="https://www.shutterstock.com/video/clip-13547351-02438-set-movie-icons-boy-enjoying-drink" alt="logo" width="250"/>
+</center>
+</p>
+""",
     allow_flagging='auto',
     theme="huggingface",  # establece un tema predefinido
     favicon="https://iconos8.es/icon/OrZ75sWwdNU2/comedia",  # establece tu favicon personalizado