zfff

app.py

@@ -7,22 +7,31 @@ from gradio.outputs import Video
 from video_generator import generate_video, extract_last_frame
 
 def extract_lyrics(api_response):
+    lyrics_with_timing = []
+    CHUNK_DURATION = 10
+    current_chunk = ""
+    current_chunk_start_time = 0
+    current_chunk_end_time = 0
+    
     try:
         words_timing = api_response["results"]["channels"][0]["alternatives"][0]["words"]
-        lyrics_chunks = []
-        CHUNK_DURATION = 10
-        current_chunk = ""
-        current_chunk_start_time = 0
         for word_info in words_timing:
             word = word_info["word"]
             start_time = word_info["start"]
+            end_time = word_info["end"]
+
             if start_time >= current_chunk_start_time + CHUNK_DURATION:
-                lyrics_chunks.append(current_chunk.strip())
+                lyrics_with_timing.append((current_chunk_start_time, current_chunk_end_time, current_chunk.strip()))
                 current_chunk = ""
-                current_chunk_start_time += CHUNK_DURATION
+                current_chunk_start_time = start_time
+
             current_chunk += " " + word
-        lyrics_chunks.append(current_chunk.strip())
-        return lyrics_chunks
+            current_chunk_end_time = end_time
+
+        # Adding the last chunk
+        lyrics_with_timing.append((current_chunk_start_time, current_chunk_end_time, current_chunk.strip()))
+
+        return lyrics_with_timing
     except KeyError:
         print("Error in API response:", api_response)
         return []