From 939d46b127648edaf95503792e186818460f4e83 Mon Sep 17 00:00:00 2001
From: Steven <wusteven815@gmail.com>
Date: Thu, 22 Dec 2022 18:56:45 -0500
Subject: [PATCH] Added mp3 cover metadata

---
 main.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 0d161ad..2d0ecbd 100644
--- a/main.py
+++ b/main.py
@@ -48,7 +48,7 @@ def main(self):
 
     def save_file(self, initial_name):
 
-        fp = filedialog.asksaveasfilename(filetypes=[(self.file_type, f"*.{self.file_type}")],
+        fp = filedialog.asksaveasfilename(filetypes=[(self.file_type.upper(), f"*.{self.file_type}")],
                                           initialfile=initial_name)
 
         if fp == "":
@@ -72,10 +72,24 @@ def download_mp3(self):
         album = input(">>> Music Album: ")
         artist = input(">>> Music Artist: ")
 
+        print(">>> Cover Image: ", end="")
+        cover = filedialog.askopenfilename(filetypes=[("PNG", ".png"), ("JPG", ".jpg"), ("JPEG", ".jpeg")],
+                                           title="Open cover image")
+        if cover == "":
+            print("None selected")
+        else:
+            print(cover)
+
         metadata = eyed3_load(self.temp_file)
+        metadata.initTag(version=(2, 3, 0))
         metadata.tag.title = name
         metadata.tag.album = album
         metadata.tag.artist = artist
+
+        if cover != "":
+            with open(cover, "rb") as cover_file:
+                metadata.tag.images.set(3, cover_file.read(), "image/jpeg", u"Cover")
+
         metadata.tag.save()
 
         self.save_file(name)