update app.py

app.py

@@ -26,16 +26,18 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
     recommended_titles = [id_to_title[movie_id] for movie_id, _ in recommended_movies[:top]]
 
     # Contar cuántas películas de cada género hay en las recomendaciones
-   # genre_counts = df_filtered[df_filtered['id'].isin([movie_id for movie_id, _ in recommended_movies])][genres].sum()
-    
     recommended_movies_ids = [movie_id for movie_id, _ in recommended_movies[:top]]
     genre_counts = df_filtered[df_filtered['id'].isin(recommended_movies_ids)][genres].sum()
 
     # Limpiar la figura
     plt.clf()
 
-    # Crear el gráfico de barras
-    plt.bar(genres, genre_counts)
+    # Asignar colores específicos a cada género
+    genre_colors = {'Drama': 'blue', 'Comedy': 'orange', 'Horror': 'red', 'Romance': 'pink'}
+    colors = [genre_colors[genre] for genre in genres]
+
+    # Crear el gráfico de barras con los colores específicos
+    plt.bar(genres, genre_counts, color=colors)
     plt.xlabel('Género')
     plt.ylabel('Cantidad')
     plt.title('Cantidad de Películas por Género en las Recomendaciones')
@@ -53,6 +55,7 @@ def generar_recomendacion(svd_model, user_id, df, genres, top=5):
     # Devolver la lista de títulos y el gráfico como una imagen
     return ', '.join(recommended_titles), im
 
+
 # Leer los datos
 dfmerge = pd.read_csv('merged_data7.csv')