Update Recomendation.py

Recomendation.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(