diff --git a/wobuzz/ui/track_info.py b/wobuzz/ui/track_info.py index 6782a41..dc85b86 100644 --- a/wobuzz/ui/track_info.py +++ b/wobuzz/ui/track_info.py @@ -55,7 +55,6 @@ class TrackInfo(QToolBar): title = current_track.tags.title artist = current_track.tags.artist album = current_track.tags.album - cover_data = current_track.tags.images.any.data self.title.setText(title) @@ -71,6 +70,15 @@ class TrackInfo(QToolBar): else: self.album.setText("No Album") + cover = current_track.tags.images.any + + if cover is None: + self.track_cover.setPixmap(self.wobuzz_logo) + + return + + cover_data = cover.data + if isinstance(cover_data, bytes): cover_pixmap = QPixmap() cover_pixmap.loadFromData(cover_data)