Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import math
 import time
 import numpy as np
+import csv
 from pydub import AudioSegment
 import io
 
@@ -36,25 +37,28 @@ def stream(audio, chunk_length_s):
     start_time = time.time()
     sampling_rate, array = audio
     chunk_length = int(chunk_length_s * sampling_rate)
-    time_length = chunk_length_s / 2  # always stream outputs faster than it takes to process
+    time_length = chunk_length_s / 2  # Always stream outputs faster than it takes to process
+    max_duration = 30  # Only process the first 30 seconds
+    max_samples = int(max_duration * sampling_rate)
+
+    # Truncate audio array to 30 seconds
+    array = array[:max_samples]
     audio_length = len(array)
     num_batches = math.ceil(audio_length / chunk_length)
 
-    # Initialize a list to store timestamps for a 30-second timespan
+    # Initialize a list to store timestamps
     timestamps = []
 
     for idx in range(num_batches):
-        time.sleep(time_length)
         start_pos = idx * chunk_length
         end_pos = min((idx + 1) * chunk_length, audio_length)
         chunk_start_time = start_pos / sampling_rate
         chunk_end_time = end_pos / sampling_rate
 
-        # Save timestamps for a 30-second window
-        if chunk_start_time < 30:
-            timestamps.append((chunk_start_time, chunk_end_time))
+        # Save timestamps for the current chunk
+        timestamps.append((chunk_start_time, chunk_end_time))
 
-        chunk = array[start_pos: end_pos]
+        chunk = array[start_pos:end_pos]
         chunk_mp3 = numpy_to_mp3(chunk, sampling_rate=sampling_rate)
 
         if idx == 0:
@@ -63,7 +67,13 @@ def stream(audio, chunk_length_s):
 
         yield chunk_mp3, first_time, run_time
 
-    # Print the timestamps after streaming
+    # Save timestamps to a CSV file
+    with open("timestamps_30sec.csv", mode="w", newline="") as file:
+        writer = csv.writer(file)
+        writer.writerow(["Start Time (s)", "End Time (s)"])
+        writer.writerows(timestamps)
+
+    print(f"Timestamps saved to 'timestamps_30sec.csv'")
     print("Timestamps for the first 30 seconds:")
     for start, end in timestamps:
         print(f"Start: {start:.2f}s, End: {end:.2f}s")
@@ -82,4 +92,4 @@ with gr.Blocks() as demo:
 
     run_button.click(fn=stream, inputs=[audio_in, chunk_length], outputs=[audio_out, first_time, run_time])
 
-demo.launch()
+demo.launch()