Update app.py

app.py

@@ -65,16 +65,22 @@ def recommend_song(song_name, artist_name, spotify_data_processed, top_n=4):
 
 
 
+
 def recommend_song_interface(song_name, artist_name):
     recommendations_df = recommend_song(song_name, artist_name, spotify_data_processed)
     
-    if isinstance(recommendations_df, pd.DataFrame):
-        # Convierte el DataFrame en una lista de listas y luego a un formato de texto plano para la salida
-        recommendations_list = recommendations_df.values.tolist()
-        return ["{} by {}".format(song, artist) for song, artist in recommendations_list]
+    # Verificar si el DataFrame está vacío o si las columnas necesarias están presentes
+    if isinstance(recommendations_df, pd.DataFrame) and not recommendations_df.empty and {'song', 'artist'}.issubset(recommendations_df.columns):
+        recommendations_list = recommendations_df[['song', 'artist']].values.tolist()
+        formatted_recommendations = ["{} by {}".format(song, artist) for song, artist in recommendations_list]
+        # Rellenar con cadenas vacías si hay menos de 4 recomendaciones
+        while len(formatted_recommendations) < 4:
+            formatted_recommendations.append("")
+        return formatted_recommendations[:4]
     else:
-        # Si no es un DataFrame, devolver el mensaje de error
-        return recommendations_df
+        # Devolver mensajes de error o cadenas vacías si no se cumplen las condiciones anteriores
+        return ["Error: No se pudo procesar las recomendaciones."] + [""] * 3
+
 
 # Crear la interfaz con Gradio
 iface = gr.Interface(
@@ -90,7 +96,11 @@ iface = gr.Interface(
     title="Recomendador de Canciones",
     description="Ingrese el título de una canción y el nombre del artista para obtener recomendaciones.",
     theme="dark",  # Comenta o elimina si el tema oscuro no está disponible
-
+    css="""
+        body {font-family: Arial, sans-serif;}
+        .input_text {background-color: #f0f0f0; border-radius: 5px;}
+        .output_text {border: 2px solid #f0f0f0; border-radius: 5px; padding: 10px;}
+        """
 )
 
 iface.launch()