Implemet Lyrics

LRC.py

@@ -14,7 +14,7 @@ def safe_read(i, a):
 def get_line_info(linein):
     # Extract minutes and seconds, handle potential issues with slicing and conversion
     minutes = float(linein[1:3])
-    seconds = float(linein[5:9])
+    seconds = float(linein[4:9])
     lt = minutes * 60 + seconds
     line = ""
     i = 10

LRC2SRT.py

@@ -0,0 +1,82 @@
+import os
+import LRC as Parse
+
+SRT = []
+LRC = []
+
+def safe_read(i, a):
+    if i > len(a):
+        return None
+    else:
+        return a[i]
+
+def clear():
+    global SRT
+    SRT = []
+
+def convert_time(sec):
+    t = ""
+    t += str(int(sec//3600)).rjust(2, "0")
+    t += ':'
+    t += str(int(sec%3600//60)).rjust(2, "0")
+    t += ':'
+    t += str(int(sec%3600%60//1)).rjust(2, "0")
+    t += ','
+    t += str(round(sec%3600%60%1*1000)).rjust(3, "0")
+    return t
+
+def add_chunk(line, start, end, number):
+    global SRT
+    SRT.append(str(number))
+    SRT.append(convert_time(float(start)) + " --> " + convert_time(float(end)))
+    SRT.append(line)
+    SRT.append("")
+
+def load(input: str):
+    global LRC
+    LRC = input.split("\n")
+
+def convert_line(number, chunk):
+    m = chunk
+    info = Parse.get_line_info(LRC[number])
+    if number+1 < len(LRC):
+        info1 = Parse.get_line_info(LRC[number+1])
+    print(info)
+    tmp = ""
+    i = 0
+    if not info.get("wordbreaks"):
+        add_chunk(info.get('line'),info.get('time'),info1.get('time'),m)
+        return
+    tmp = info.get('line')[0:info.get('wordbreaks')[0]-1]
+    if tmp:
+        add_chunk(tmp,info.get('time'),info.get('wordtimes')[0],m)
+    while i <= len(info.get('wordtimes')) - 1:
+        if tmp:
+            m += 1
+
+        if i+1 > len(info.get('wordtimes')):
+            add_chunk(info.get('line'),info.get('wordtimes')[i],info1.get('time'),m)
+        else:
+            if i+1 < len(info.get('wordbreaks')):
+                tmp = info.get('line')[0:info.get('wordbreaks')[i+1]-1]
+                add_chunk(tmp,info.get('wordtimes')[i],info.get('wordtimes')[i+1],m)
+        i += 1
+
+def convert_to_srt(ly):
+    load(Parse.remove_metadata(ly))
+    i = 0
+    e = 1
+    while i < len(LRC):
+        convert_line(i,e)
+        i += 1
+        e = int(SRT[len(SRT)-4])+1
+
+if __name__ == "__main__":
+    kk = open(".lrc", encoding="UTF8")
+    
+    convert_to_srt(kk.read())
+    
+    print('\n'.join(SRT))
+    if os.path.exists("out.srt"):
+        os.remove("out.srt")
+    open("out.srt", mode='x', encoding='UTF8').write('\n'.join(SRT))

app.py

@@ -15,6 +15,8 @@ import subprocess
 import traceback
 import time
 import shutil
+import LRC
+import LRC2SRT
 
 path = ""  # Update with your path
 
@@ -156,16 +158,33 @@ def RenderVid(af, n, fps=30):
      )
     gr.Interface.download(f"{n}.mp4")
 
-def main(file, name, fps=30, res: tuple=(1280,720), oscres=512, sr=11025):
+def main(file, name, fps=30, res: tuple=(1280,720), oscres=512, sr=11025, lyrics=None):
+    p = gr.Progress()
+    if os.path.exists("out.srt"):
+        os.remove("out.srt")
+    if lyrics:
+        p(0.5,"parsing lyrics")
+        try:
+            sf = open(lyrics, encoding="UTF8").read()
+            if sf[1] == '[':
+                rr = LRC.convert_to_json(lyrics)
+            elif sf[1].isdigit():
+                outf = open("out.srt",mode="x", encoding="UTF8").write()
+            else:
+                gr.Warning("Lyrics file is invalid, skipping")
+        except:
+            gr.Warning("Failed to parse lyrics, skipping")
+        
     os.makedirs(path+f'out/{name}/', exist_ok=True)
     global iii
     iii = 0
     # Load the audio file
+    p(0.25,"loading file")
     audio_path = file
     y, sr = librosa.load(audio_path, sr=sr)  # Resample to 11025 Hz
     y_u8 = (y * 128 + 128).astype('uint8')
     samples_array = y_u8.tolist()
-
+    p(0.5,"extracting metadata")
     # Extract cover image, title, and artist
     cover_img = extract_cover_image(audio_path)
     if cover_img is None:
@@ -187,7 +206,7 @@ def main(file, name, fps=30, res: tuple=(1280,720), oscres=512, sr=11025):
 
     # Prepare parameters for each frame
     params = [(n, samples_array, cover_img, title, artist, dominant_color, width, height, fps, name, oscres, sr) for n in range(num_frames)]
-    p = gr.Progress()
+    
     try:
         with Pool(cpu_count()) as pool: