Added tests

pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+pythonpath = server

server/audio_processing.py

@@ -104,7 +104,7 @@ def detect_ads(transcript):
         raise
 
 
-def remove_ads(audio, ad_segments):
+def remove_ads(audio, ad_segments, flag=None):
     """Removes ad segments from the audio file with optimized processing."""
     if not ad_segments:
         print("No ads to remove.")
@@ -121,7 +121,7 @@ def remove_ads(audio, ad_segments):
         start, end = segment["start"] * 1000, segment["end"] * 1000  # Convert to milliseconds
 
         # Edge Case 1: Adjust ads in the first 5 seconds
-        if start <= 5000:
+        if start <= 5000 and flag=="first":
             start = max(0, start - 1000)
 
         # Edge Case 2: Merge close ads
@@ -131,7 +131,7 @@ def remove_ads(audio, ad_segments):
             merged_ads.append({"start": start, "end": end})
 
     # Edge Case 3: Remove everything from the last ad if it's near the end
-    if merged_ads and merged_ads[-1]["end"] >= total_duration - 10000:
+    if merged_ads and merged_ads[-1]["end"] >= total_duration - 10000 and flag=="last":
         merged_ads[-1]["end"] = total_duration
 
     # Extract non-ad sections with proper updating of previous_end
@@ -189,10 +189,14 @@ def process_audio(audio_segment, url, streaming):
             logger.info(f"Transcription complete for chunk {i + 1}/{len(chunks)}: {url}")
 
             ad_segments = detect_ads(transcription)
-            logger.info(f"Ad-analysis complete for chunk {i + 1}/{len(chunks)}: {url}")
-
-            processed_chunk = remove_ads(chunk, ad_segments)
-            logger.info(f"Processing complete for chunk {i + 1}/{len(chunks)}: {url}")
+            logger.info(f"Ad-analysis complete for chunk {i+1}/{len(chunks)}: {url}")
+            flag = None
+            if i == 0:
+                flag = "first"
+            elif i == len(chunks) - 1:
+                flag = "last"
+            processed_chunk = remove_ads(chunk, ad_segments, flag=flag)
+            logger.info(f"Processing complete for chunk {i+1}/{len(chunks)}: {url}")
 
             if i == 0 and not streaming:
                 processed_chunk = intro + processed_chunk