Update inference/infer.py

inference/infer.py

@@ -121,33 +121,24 @@ stage1_output_set = []
 # genre tags support instrumental，genre，mood，vocal timbr and vocal gender
 # all kinds of tags are needed
 # 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
+# Read genre file with UTF-8 encoding
 with open(args.genre_txt, encoding="utf-8") as f:
     genres = f.read().strip()
 
-# Read lyrics safely and debug format
+# Read lyrics file with UTF-8 encoding
 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")])
+# Format lyrics content and ensure proper newline spacing between blocks
+formatted_lyrics = "\n\n".join([block.strip() for block in raw_lyrics.split("\n\n")])
 
-# Process lyrics
-lyrics = split_lyrics(normalized_lyrics)
+# Now, split the formatted lyrics into blocks
+lyrics = split_lyrics(formatted_lyrics)
 
-# Debugging: Check lyrics after splitting
-print("DEBUG: Lyrics after split_lyrics:", repr(lyrics))
-
-# Prepare prompt
+# Join all lyrics into one single string for the full lyrics
 full_lyrics = "\n".join(lyrics)
+
+# Prepare the prompt with genre and full lyrics
 prompt_texts = [f"Generate music from the given lyrics segment by segment.\n[Genre] {genres}\n{full_lyrics}"]
 prompt_texts += lyrics