Update app.py

app.py

@@ -5,6 +5,18 @@ splt_client = Client("https://fffiloni-splittrack2musicgen.hf.space/")
 #whisper_client = Client("https://sanchit-gandhi-whisper-jax.hf.space/")
 whisper_client = Client("https://sanchit-gandhi-whisper-large-v2.hf.space/")
 
+import re
+
+def format_lyrics(text):
+    # Use regex to find parts that start with a capital letter
+    pattern = r'(^|\n)([A-Z][^\n]*)'
+    formatted_text = re.sub(pattern, r'\n\2', text)
+
+    # Remove any leading whitespace on each line
+    formatted_text = re.sub(r'^[ \t]+', '', formatted_text, flags=re.MULTILINE)
+
+    return formatted_text
+    
 def infer(audio_input):
 
     # STEP 1 | Split vocals from the song/audio file
@@ -27,7 +39,9 @@ def infer(audio_input):
 
     #return whisper_result[0] # if using JAX
 
-    return splt_result, whisper_result
+    lyrics = format_lyrics(whisper_result)
+
+    return splt_result, lyrics
 
 css = """
 #col-container {max-width: 510px; margin-left: auto; margin-right: auto;}