Update inference/infer.py

inference/infer.py

@@ -120,11 +120,33 @@ stage1_output_set = []
 # Tips:
 # genre tags support instrumental，genre，mood，vocal timbr and vocal gender
 # all kinds of tags are needed
-with open(args.genre_txt) as f:
+# Ensure files exist
+if not os.path.exists(args.genre_txt):
+    raise FileNotFoundError(f"Genre file not found: {args.genre_txt}")
+if not os.path.exists(args.lyrics_txt):
+    raise FileNotFoundError(f"Lyrics file not found: {args.lyrics_txt}")
+
+# Read genre file safely
+with open(args.genre_txt, encoding="utf-8") as f:
     genres = f.read().strip()
-with open(args.lyrics_txt) as f:
-    lyrics = split_lyrics(f.read())
-# intruction
+
+# Read lyrics safely and debug format
+with open(args.lyrics_txt, encoding="utf-8") as f:
+    raw_lyrics = f.read()
+
+# Debugging: Check raw file content
+print("DEBUG: Lyrics before processing:", repr(raw_lyrics))
+
+# Normalize lyrics format
+normalized_lyrics = "\n\n".join([block.strip() for block in raw_lyrics.split("\n\n")])
+
+# Process lyrics
+lyrics = split_lyrics(normalized_lyrics)
+
+# Debugging: Check lyrics after splitting
+print("DEBUG: Lyrics after split_lyrics:", repr(lyrics))
+
+# Prepare prompt
 full_lyrics = "\n".join(lyrics)
 prompt_texts = [f"Generate music from the given lyrics segment by segment.\n[Genre] {genres}\n{full_lyrics}"]
 prompt_texts += lyrics