Finished apply to boss or level music

README.md

@@ -23,13 +23,13 @@ To use it just launch the executable file (if available for your os) or type "py
 If you have any issues make a gitlab issue. There may be bugs.
 
 ## Roadmap
+* modify restore_ogmusic so it is able to restore only the last change. Maybe config files will be needed for this.
+* make it so that if "you press apply music and then replace song" it checks for that and doesn't allow the "replaced song" to be reverted back to original.
 * "now playing" text
 * ability to preserve changes through Mega Man Maker updates, by using a config file and/or a downloaded zip of the game.
 * stop using os.path in favour of pathlib
 * switch to using a config file (or database) for keeping track of which tracks have been changed, that way only having one original_music folder and being able to *show* the user which song you used to replace which file.
 * add selective mute, so that you mute every music except the boss music, or every song except menu music or both.
-* figure out the best way to add a setting for changing each of the level editor songs, menu music. Maybe an extra popup window or a check box for "replace every one" "menu" and "levelBuild"
-* add a way to change boss music separate from level music and menu music
 * add a way to show the available nsf files in the selected folder, like a playlist.
 * Instead of duplicating a single file a lot of times, maybe I'll implement symlink support.
 

mmm-nsf-changer.py

@@ -20,6 +20,27 @@ def safelist():
     safe_files = ['MetalMan - (GB II).nsf', 'IceMan - (GB I).nsf', 'CutMan - (GB I).nsf', 'Neptune - (GB V).nsf', 'GBVWily2.nsf', 'WoodMan - (GB II).nsf', 'Saturn - (GB V).nsf', 'Jupiter - (GB V).nsf', 'Terra - (GB V).nsf', 'Venus - (GB V).nsf', 'GBIIIWily2.nsf', 'AirMan - (GB II).nsf', 'Uranus - (GB V).nsf', 'Pluto - (GB V).nsf', 'GBVWily1.nsf', 'Mercury - (GB V).nsf', 'GBIIWily2.nsf', 'MagnetMan - (GB II).nsf', 'Mars - (GB V).nsf', 'GBIIIWily1.nsf', 'ElecMan - (GB I).nsf', 'GBIWily2.nsf', 'TopMan - (GB II).nsf', 'CrashMan - (GB II).nsf', 'GBIWily1.nsf', 'Punk - (GB III).nsf', 'HardMan - (GB II).nsf', 'NeedleMan - (GB II).nsf', 'FireMan - (GB I).nsf', 'GBIVWily1.nsf', 'GBIIWily1.nsf', 'FreezeMan.nsf', 'JunkMan.nsf', 'SlashMan.nsf', 'CloudMan.nsf', 'SpringMan.nsf', 'Wily4.nsf', 'RobotMuseum.nsf', 'Wily1.nsf', 'Intro.nsf', 'Wily3.nsf', 'ShadeMan.nsf', 'ShadeManAlt.nsf', 'TurboMan.nsf', 'BurstMan.nsf', 'Wily2.nsf', 'WilyTower2.nsf', 'HyperStormH.nsf', 'WilyTower1.nsf', 'WilyTower3.nsf', 'MegaWaterS.nsf', 'WilyTower4.nsf', 'BusterRodG.nsf', 'GravityMan.nsf', 'WaveMan.nsf', 'StoneMan.nsf', 'MM5WilyCirno.nsf', 'MM5Wily.nsf', 'StarMan.nsf', 'NapalmMan.nsf', 'CrystalMan.nsf', 'ProtoCastle.nsf', 'ChargeMan.nsf', 'GyroMan.nsf', 'FlameMan.nsf', 'MrX.nsf', 'YamatoMan.nsf', 'CentaurMan.nsf', 'WindMan.nsf', 'PlantMan.nsf', 'MM6Wily.nsf', 'KnightMan.nsf', 'BlizzardMan.nsf', 'TomahawkMan.nsf', 'MM2Wily2.nsf', 'Crashman.nsf', 'MM2Wily1.nsf', 'MM2Wily6.nsf', 'Metalman.nsf', 'Flashman.nsf', 'Bubbleman.nsf', 'Woodman.nsf', 'Heatman.nsf', 'Quickman.nsf', 'Airman.nsf', 'GameOver.nsf', 'WilyIntro3.nsf', 'StageStart.nsf', 'WilyIntro2.nsf', 'WilyIntro1.nsf', 'Victory.nsf', 'ProtoWhistle.nsf', 'Bombman.nsf', 'MM1Wily1.nsf', 'MM1Wily2.nsf', 'Fireman.nsf', 'Iceman.nsf', 'Cutman.nsf', 'Elecman.nsf', 'PUIntroStage.nsf', 'Timeman.nsf', 'Oilman.nsf', 'Gutsman.nsf', 'FrostMan.nsf', 'GrenadeMan.nsf', 'AquaMan.nsf', 'AstroMan.nsf', 'Wily4.nsf', 'TenguMan(SS).nsf', 'Wily1.nsf', 'IntroStage(Underwater).nsf', 'SwordMan.nsf', 'Wily3.nsf', 'SearchMan.nsf', 'IntroStage.nsf', 'ClownMan.nsf', 'Wily2.nsf', 'TenguMan(PSX).nsf', 'Megaman4.nsf', 'ToadMan.nsf', 'MM4Wily2.nsf', 'RingMan.nsf', 'MM4Wily1.nsf', 'DrillMan.nsf', 'BrightMan.nsf', 'DiveMan.nsf', 'SkullMan.nsf', 'DustMan.nsf', 'Cossack1.nsf', 'PharaohMan.nsf', 'Cossack2.nsf', 'MagnetMan.nsf', 'MM3Wily2.nsf', 'MM3Wily3.nsf', 'SnakeMan.nsf', 'Megaman3.nsf', 'MM3Wily1.nsf', 'GeminiMan.nsf', 'ShadowMan.nsf', 'HardMan.nsf', 'TopMan.nsf', 'NeedleMan.nsf', 'SparkMan.nsf', 'KazeYoTsutaete.nsf', 'MMB_-_King_Nix.nsf', '30XXTitleLooped.nsf', 'FakeMan.nsf', 'PlugMan.nsf', 'GalaxyMan.nsf', 'EndlessAttack.nsf', 'SplashWoman.nsf', 'Wily4.nsf', 'Wily1.nsf', 'MagmaMan.nsf', 'JewelMan.nsf', 'ConcreteMan.nsf', 'Wily3.nsf', 'TornadoMan.nsf', 'Wily2.nsf', 'HornetMan.nsf', 'StrikeMan.nsf', 'Challenges.nsf', 'Wily1Outside.nsf', 'SolarMan.nsf', 'EndlessAttack.nsf', 'Wily4.nsf', 'NitroMan.nsf', 'Wily1Inside.nsf', 'CommandoMan.nsf', 'Punk.nsf', 'Wily5.nsf', 'Wily3.nsf', 'Enker.nsf', 'Ballade.nsf', 'ChillMan.nsf', 'BladeMan.nsf', 'PumpMan.nsf', 'Wily2.nsf', 'SheepMan.nsf', 'MM4.nsf', 'MM&B.nsf', 'MM8Wily.nsf', 'Sunstar - (GB V).nsf', 'MM7WilyCapsule.nsf', 'GBIVWilyBoss.nsf', 'MM10WilyCapsule.nsf', 'MMPUWilyBoss.nsf', 'MM11WilyCapsule.nsf', 'GBIV.nsf', 'MM3.nsf', 'MM&BKingBoss.nsf', 'MM3Wily.nsf', 'MM10WilyBoss.nsf', 'MM10WilyMachine.nsf', 'MM1Wily.nsf', 'MM10.nsf', 'MM2.nsf', 'MM5.nsf', 'BassMM8.nsf', 'MMPUWilyMachine.nsf', 'MM7Wily.nsf', 'MMWilyWars.nsf', 'MM8WoodMan.nsf', 'Duo.nsf', 'MM6Wily.nsf', 'MM11.nsf', 'BassMM7.nsf', 'MM5Wily.nsf', 'MM6.nsf', 'MM8CutMan.nsf', 'BossIntro.nsf', 'MM7.nsf', 'MMPU.nsf', 'MM4Wily.nsf', 'GBI.nsf', 'GBIVBalladeBoss.nsf', 'MM1.nsf', 'MM11WilyBoss.nsf', 'GBV.nsf', 'MM11WilyMachine.nsf', 'Midboss.nsf', 'GBII.nsf', 'MM9.nsf', 'MM8.nsf', 'MM8WilyCapsule.nsf', 'MM9Wily.nsf', 'EditorB.nsf', 'Credits.nsf', 'EditorD.nsf', '30XXTitle.nsf', 'Title.nsf', 'OnlineMenu.nsf', 'Menu.nsf', 'Editor.nsf', 'EditorC.nsf', 'ChallengeCredits.nsf', 'Tutorial.nsf', 'MagicMan.nsf', 'ColdMan.nsf', 'MM&BWily1.nsf', 'MM&BKing1.nsf', 'PirateMan.nsf', 'DynamoMan.nsf', 'GroundMan.nsf', 'AstroMan.nsf', 'Intro.nsf', 'TenguMan.nsf', 'CrystalGate.nsf', 'BurnerMan.nsf', 'BounceMan.nsf', 'FuseMan.nsf', 'TorchMan.nsf', 'Challenges.nsf', 'WilyStage.nsf', 'TundraMan.nsf', 'AcidMan.nsf', 'DrLightTrial.nsf', 'BlockMan.nsf', 'BlastMan.nsf', 'ImpactMan.nsf']
     return safe_files
 
+def boss_music():
+    boss_music = ['MM4.nsf', 'MM&B.nsf', 'MM8Wily.nsf', 'Sunstar - (GB V).nsf', 'MM7WilyCapsule.nsf', 'GBIVWilyBoss.nsf', 'MM10WilyCapsule.nsf', 'MMPUWilyBoss.nsf', 'MM11WilyCapsule.nsf', 'GBIV.nsf', 'MM3.nsf', 'MM&BKingBoss.nsf', 'MM3Wily.nsf', 'MM10WilyBoss.nsf', 'MM10WilyMachine.nsf', 'MM1Wily.nsf', 'MM10.nsf', 'MM2.nsf', 'MM5.nsf', 'BassMM8.nsf', 'MMPUWilyMachine.nsf', 'MM7Wily.nsf', 'MMWilyWars.nsf', 'MM8WoodMan.nsf', 'Duo.nsf', 'MM6Wily.nsf', 'MM11.nsf', 'BassMM7.nsf', 'MM5Wily.nsf', 'MM6.nsf', 'MM8CutMan.nsf', 'BossIntro.nsf', 'MM7.nsf', 'MMPU.nsf', 'MM4Wily.nsf', 'GBI.nsf', 'GBIVBalladeBoss.nsf', 'MM1.nsf', 'MM11WilyBoss.nsf', 'GBV.nsf', 'MM11WilyMachine.nsf', 'Midboss.nsf', 'GBII.nsf', 'MM9.nsf', 'MM8.nsf', 'MM8WilyCapsule.nsf', 'MM9Wily.nsf']
+    return boss_music
+
+def level_music():
+    level_music = ['MetalMan - (GB II).nsf', 'IceMan - (GB I).nsf', 'CutMan - (GB I).nsf', 'Neptune - (GB V).nsf', 'GBVWily2.nsf', 'WoodMan - (GB II).nsf', 'Saturn - (GB V).nsf', 'Jupiter - (GB V).nsf', 'Terra - (GB V).nsf', 'Venus - (GB V).nsf', 'GBIIIWily2.nsf', 'AirMan - (GB II).nsf', 'Uranus - (GB V).nsf', 'Pluto - (GB V).nsf', 'GBVWily1.nsf', 'Mercury - (GB V).nsf', 'GBIIWily2.nsf', 'MagnetMan - (GB II).nsf', 'Mars - (GB V).nsf', 'GBIIIWily1.nsf', 'ElecMan - (GB I).nsf', 'GBIWily2.nsf', 'TopMan - (GB II).nsf', 'CrashMan - (GB II).nsf', 'GBIWily1.nsf', 'Punk - (GB III).nsf', 'HardMan - (GB II).nsf', 'NeedleMan - (GB II).nsf', 'FireMan - (GB I).nsf', 'GBIVWily1.nsf', 'GBIIWily1.nsf', 'FreezeMan.nsf', 'JunkMan.nsf', 'SlashMan.nsf', 'CloudMan.nsf', 'SpringMan.nsf', 'Wily4.nsf', 'RobotMuseum.nsf', 'Wily1.nsf', 'Intro.nsf', 'Wily3.nsf', 'ShadeMan.nsf', 'ShadeManAlt.nsf', 'TurboMan.nsf', 'BurstMan.nsf', 'Wily2.nsf', 'WilyTower2.nsf', 'HyperStormH.nsf', 'WilyTower1.nsf', 'WilyTower3.nsf', 'MegaWaterS.nsf', 'WilyTower4.nsf', 'BusterRodG.nsf', 'GravityMan.nsf', 'WaveMan.nsf', 'StoneMan.nsf', 'MM5WilyCirno.nsf', 'MM5Wily.nsf', 'StarMan.nsf', 'NapalmMan.nsf', 'CrystalMan.nsf', 'ProtoCastle.nsf', 'ChargeMan.nsf', 'GyroMan.nsf', 'FlameMan.nsf', 'MrX.nsf', 'YamatoMan.nsf', 'CentaurMan.nsf', 'WindMan.nsf', 'PlantMan.nsf', 'MM6Wily.nsf', 'KnightMan.nsf', 'BlizzardMan.nsf', 'TomahawkMan.nsf', 'MM2Wily2.nsf', 'Crashman.nsf', 'MM2Wily1.nsf', 'MM2Wily6.nsf', 'Metalman.nsf', 'Flashman.nsf', 'Bubbleman.nsf', 'Woodman.nsf', 'Heatman.nsf', 'Quickman.nsf', 'Airman.nsf', 'Bombman.nsf', 'MM1Wily1.nsf', 'MM1Wily2.nsf', 'Fireman.nsf', 'Iceman.nsf', 'Cutman.nsf', 'Elecman.nsf', 'PUIntroStage.nsf', 'Timeman.nsf', 'Oilman.nsf', 'Gutsman.nsf', 'FrostMan.nsf', 'GrenadeMan.nsf', 'AquaMan.nsf', 'AstroMan.nsf', 'Wily4.nsf', 'TenguMan(SS).nsf', 'Wily1.nsf', 'IntroStage(Underwater).nsf', 'SwordMan.nsf', 'Wily3.nsf', 'SearchMan.nsf', 'IntroStage.nsf', 'ClownMan.nsf', 'Wily2.nsf', 'TenguMan(PSX).nsf', 'Megaman4.nsf', 'ToadMan.nsf', 'MM4Wily2.nsf', 'RingMan.nsf', 'MM4Wily1.nsf', 'DrillMan.nsf', 'BrightMan.nsf', 'DiveMan.nsf', 'SkullMan.nsf', 'DustMan.nsf', 'Cossack1.nsf', 'PharaohMan.nsf', 'Cossack2.nsf', 'MagnetMan.nsf', 'MM3Wily2.nsf', 'MM3Wily3.nsf', 'SnakeMan.nsf', 'Megaman3.nsf', 'MM3Wily1.nsf', 'GeminiMan.nsf', 'ShadowMan.nsf', 'HardMan.nsf', 'TopMan.nsf', 'NeedleMan.nsf', 'SparkMan.nsf', 'KazeYoTsutaete.nsf', 'MMB_-_King_Nix.nsf', '30XXTitleLooped.nsf', 'FakeMan.nsf', 'PlugMan.nsf', 'GalaxyMan.nsf', 'EndlessAttack.nsf', 'SplashWoman.nsf', 'Wily4.nsf', 'Wily1.nsf', 'MagmaMan.nsf', 'JewelMan.nsf', 'ConcreteMan.nsf', 'Wily3.nsf', 'TornadoMan.nsf', 'Wily2.nsf', 'HornetMan.nsf', 'StrikeMan.nsf', 'Challenges.nsf', 'Wily1Outside.nsf', 'SolarMan.nsf', 'EndlessAttack.nsf', 'Wily4.nsf', 'NitroMan.nsf', 'Wily1Inside.nsf', 'CommandoMan.nsf', 'Punk.nsf', 'Wily5.nsf', 'Wily3.nsf', 'Enker.nsf', 'Ballade.nsf', 'ChillMan.nsf', 'BladeMan.nsf', 'PumpMan.nsf', 'Wily2.nsf', 'SheepMan.nsf', 'MagicMan.nsf', 'ColdMan.nsf', 'MM&BWily1.nsf', 'MM&BKing1.nsf', 'PirateMan.nsf', 'DynamoMan.nsf', 'GroundMan.nsf', 'AstroMan.nsf', 'Intro.nsf', 'TenguMan.nsf', 'CrystalGate.nsf', 'BurnerMan.nsf', 'BounceMan.nsf', 'FuseMan.nsf', 'TorchMan.nsf', 'Challenges.nsf', 'WilyStage.nsf', 'TundraMan.nsf', 'AcidMan.nsf', 'DrLightTrial.nsf', 'BlockMan.nsf', 'BlastMan.nsf', 'ImpactMan.nsf']
+    return level_music
+
+# def generate_whitelist(megamaker_folder):
+#     subfolders = os.listdir(os.path.join(megamaker_folder, "Music"))
+#     subfolders.remove("Other")
+#     subfolders.remove("Misc")
+#     subfolders.remove("Boss")
+#     filenames = []
+#     for folder in subfolders:
+#         folder_filenames =os.listdir(os.path.join(megamaker_folder, "Music", folder))
+#         for filename in folder_filenames:
+#             filenames.append(filename)
+#
+#     print(filenames)
+
 def changed_nsf_files(megamaker_folder):
     megamaker_folder = os.path.abspath(megamaker_folder)
     music_folder = os.path.join(megamaker_folder, "original_specific_songs")
@@ -87,7 +108,7 @@ def create_mute_file(base64_str):
             file_to_save.write(decoded_image_data)
             return True
 
-def check_folder_structure(megamaker_folder, og_music=False):
+def check_folder_structure(megamaker_folder, og_music=False, whitelist=[]):
     megamaker_folder = os.path.abspath(megamaker_folder)
     subfolders = ([ f.path for f in os.scandir(megamaker_folder) if f.is_dir() ])
     music_folder = os.path.join(megamaker_folder, "Music")
@@ -105,19 +126,30 @@ def check_folder_structure(megamaker_folder, og_music=False):
                 print("skipping", files_no_path)
             else:
                 file_names.append(files_no_path)
-        if set(file_names) == set(safelist()):
+        if set(file_names) == set(safelist()) or set(file_names) == set(whitelist):
             return file_paths
         else:
             print("Safety check failed. Either you have other files in the music folder of mega maker, you are in the wrong folder, or there has been a new update.")
     else:
         return False
 
+##this needs an overhaul.
 def copy_music_to_ogmusic(path):
-    path = os.path.abspath(path)
     original_filepaths = check_folder_structure(path)
     if original_filepaths:
-        if not filecmp.cmp(os.path.join(path,"Music","MM1","Cutman.nsf"), os.path.join(path, "Music", "MM1", "Oilman.nsf"), shallow=False):
+
-            shutil.copytree(os.path.join(path, "Music"), os.path.join(path, "original_music"), dirs_exist_ok=True)
+        #if not filecmp.cmp(os.path.join(path,"Music","MM1","Cutman.nsf"), os.path.join(path, "Music", "MM1", "Oilman.nsf"), shallow=False):
+        for filepath in original_filepaths:
+            ogmusic_dirname = os.path.join(path, "original_music", os.path.basename(os.path.dirname(filepath)))
+            ogmusic_filepath = os.path.join(ogmusic_dirname, os.path.basename(filepath))
+            if not os.path.exists(ogmusic_dirname):
+                print("WORKING", ogmusic_dirname)
+                os.makedirs(ogmusic_dirname)
+            if os.path.exists(ogmusic_filepath):
+                print(os.path.basename(ogmusic_filepath), "EXISTS. SKIPPING.")
+            else:
+                shutil.copy(filepath, ogmusic_filepath)
+            #shutil.copytree(os.path.join(path, "Music"), os.path.join(path, "original_music"), dirs_exist_ok=True)
             if not os.path.exists(os.path.join(path, "Music")):
                 os.makedirs(os.path.join(path, "Music"))
         else:
@@ -166,20 +198,6 @@ def restore_song(old_song_name, megamaker_folder):
         except FileNotFoundError:
             return False
 
-# #this needs to check if original_music exists, and if it does restore_ogmusic before moving to og music.
-# def move_music_to_ogmusic(path):
-#     path = os.path.abspath(path)
-#     original_filepaths = check_folder_structure(path)
-#     if original_filepaths:
-#         if not filecmp.cmp(os.path.join(path,"Music","MM1","Cutman.nsf"), os.path.join(path, "Music", "MM1", "Oilman.nsf"), shallow=False):
-#             shutil.move(os.path.join(path, "Music"), os.path.join(path, "original_music"))
-#             if not os.path.exists(os.path.join(path, "Music")):
-#                 os.makedirs(os.path.join(path, "Music"))
-#         else:
-#             print("Cutman.nsf and Oilman.nsf match in Music folder.")
-#             return False
-
-
 def recreate_structure(new_nsf, original_paths_list):
     for path in original_paths_list:
         if not os.path.exists(os.path.dirname(path)):
@@ -196,24 +214,45 @@ def restore_ogmusic(path):
     if os.path.exists(os.path.join(path, "original_music")):
         if os.path.exists(os.path.join(path, 'Music')):
             if check_folder_structure(path):
-                if check_folder_structure(path, og_music=True):
+                #if check_folder_structure(path, og_music=True):
-                    if not filecmp.cmp(os.path.join(path, "Music", "MM1", 'Cutman.nsf'), os.path.join(path, "original_music", "MM1", "Cutman.nsf"), shallow=False):
+                #if not filecmp.cmp(os.path.join(path, "Music", "MM1", 'Cutman.nsf'), os.path.join(path, "original_music", "MM1", "Cutman.nsf"), shallow=False):
-                        shutil.rmtree(os.path.join(path, "Music"))
+                #shutil.rmtree(os.path.join(path, "Music"))
-                        shutil.move(os.path.join(path, "original_music", ""), os.path.join(path, "Music"))
+                ogmusic = os.path.join(path, "original_music")
-                        return True
+                music = os.path.join(path, "Music")
-                    else:
+                shutil.copytree(ogmusic, music, dirs_exist_ok=True)
-                        print("something went wrong...")
+                shutil.rmtree(ogmusic)
-                else:
+                #files = os.listdir(ogmusic)
-                    return False
+                #print(files)
+                #for filename in files:
+                    #filepath = os.path.join(music, os.path.basename(filename))
+                    #shutil.move(os.path.join(src, filename), os.path.join(path, "Music"))
+                return True
+                    #else:
+                    #    print("something went wrong...")
+                #else:
+                #    print("if check_folder_structure(path, og_music=True):")
+                #    return False
             else:
+                print("if check_folder_structure(path):")
                 return False
         else:
+            print("if os.path.exists(os.path.join(path, 'Music')):")
             return False
     else:
+        print("os.path.exists(os.path.join(path, 'original_music')):")
         return False
 
-def nsf_changer (nsf_path, megamaker_folder):
+def nsf_changer (nsf_path, megamaker_folder, whitelist=[]):
-    original_path = check_folder_structure(megamaker_folder)
+    original_path = check_folder_structure(megamaker_folder, whitelist=whitelist)
+
+    if whitelist:
+        full_original_path = original_path
+        original_path = []
+        for files in full_original_path:
+            files_no_path = os.path.basename(files)
+            if files_no_path in whitelist:
+                original_path.append(files)
+    #if not whitelist or files_no_path in whitelist:
     if original_path:
         try:
             common = os.path.commonpath([os.path.dirname(nsf_path), os.path.join(megamaker_folder, "Music")])
@@ -258,6 +297,7 @@ title = [
 
 change_all = [
     [sg.Button("Mute All"), sg.Button("Apply Music"), sg.Button("Restore Music")],
+    [sg.Button("Apply as Boss Music"), sg.Button("Apply as Level Music"), sg.Button("Mute Level Music"), sg.Button("Mute Boss Music")]
 ]
 
 change_one = [
@@ -312,6 +352,18 @@ while True:
                 restore_ogmusic(values["-OUT-"])
                 if not nsf_changer(os.path.join(os.path.dirname(__file__), "mmm_nsf_changer_mute.nsf"), values["-OUT-"]):
                     sg.popup_error("MegaMaker folder is not valid or definitions are out of date.")
+        if event == "Mute Boss Music":
+            create_mute_file(mute_nsf)
+            if not nsf_changer(os.path.join(os.path.dirname(__file__), "mmm_nsf_changer_mute.nsf"), values["-OUT-"], whitelist=boss_music()):
+                restore_ogmusic(values["-OUT-"])
+                if not nsf_changer(os.path.join(os.path.dirname(__file__), "mmm_nsf_changer_mute.nsf"), values["-OUT-"], whitelist=boss_music()):
+                    sg.popup_error("MegaMaker folder is not valid or definitions are out of date.")
+        if event == "Mute Level Music":
+            create_mute_file(mute_nsf)
+            if not nsf_changer(os.path.join(os.path.dirname(__file__), "mmm_nsf_changer_mute.nsf"), values["-OUT-"], whitelist=level_music()):
+                restore_ogmusic(values["-OUT-"])
+                if not nsf_changer(os.path.join(os.path.dirname(__file__), "mmm_nsf_changer_mute.nsf"), values["-OUT-"], whitelist=level_music()):
+                    sg.popup_error("MegaMaker folder is not valid or definitions are out of date.")
         if event == "Replace Song":
             if values["-DROPDOWN-"] == "":
                 sg.popup_error('Please select which Output NSF File to replace. If not, use "Replace *All* NSF Music".')
@@ -334,7 +386,17 @@ while True:
         if event == "Restore Music": # Needs testing
             # sg.popup_error("DOESNT WORK EITHER HAHAHAHAHAHAA")
             if not restore_ogmusic(values["-OUT-"]):
-                sg.popup_error("Unexpected folder structure or differing files")
+                sg.popup_error("Already Restored or wrong Mega Man Maker folder.")
+        if event == "Apply as Boss Music":
+            if not nsf_changer(values["-IN-"], values["-OUT-"], whitelist=boss_music()):
+                restore_ogmusic(values["-OUT-"])
+                if not nsf_changer(values["-IN-"], values["-OUT-"], whitelist=boss_music()):
+                    sg.popup_error("MegaMaker folder is not valid or definitions out of date.")
+        if event == "Apply as Level Music":
+            if not nsf_changer(values["-IN-"], values["-OUT-"], whitelist=level_music()):
+                restore_ogmusic(values["-OUT-"])
+                if not nsf_changer(values["-IN-"], values["-OUT-"], whitelist=level_music()):
+                    sg.popup_error("MegaMaker folder is not valid or definitions out of date.")
         if event == "Download and Restore Music":
             sg.popup_yes_no("WARNING: This will erase all custom songs.\nDo you want to continue?", location=window.current_location())
             window["-LAST-CHANGED-"].update("")