process from file and fix short audios

handler.py

@@ -10,18 +10,21 @@ import os
 
 def save_chunks_to_temp_files(url, chunk_length=5000):  # chunk_length in milliseconds
     # Download the audio file from the URL
-    response = requests.get(url)
-    response.raise_for_status()
+    if not url.startswith("file://"):
+        response = requests.get(url)
+        response.raise_for_status()
 
-    # Ensure the content type is audio
-    if "audio" not in response.headers["Content-Type"]:
-        raise ValueError("URL does not seem to be an audio file")
+        # Ensure the content type is audio
+        if "audio" not in response.headers["Content-Type"]:
+            raise ValueError("URL does not seem to be an audio file")
 
-    # Convert the downloaded bytes into a file-like object
-    audio_file = BytesIO(response.content)
+        # Convert the downloaded bytes into a file-like object
+        audio_file = BytesIO(response.content)
 
-    # Load audio into an AudioSegment
-    audio_segment = AudioSegment.from_file(audio_file)
+        # Load audio into an AudioSegment
+        audio_segment = AudioSegment.from_file(audio_file)
+    else:
+        audio_segment = AudioSegment.from_file(url[7:])
 
     # Split audio into 10-second chunks
     chunks = [
@@ -29,7 +32,8 @@ def save_chunks_to_temp_files(url, chunk_length=5000):  # chunk_length in millis
         for i in range(0, len(audio_segment), chunk_length)
     ]
 
-    chunks[-1] = audio_segment[-chunk_length:]  # Ensure last chunk is 10 seconds long
+    if len(chunks) > 1:
+        chunks[-1] = audio_segment[-chunk_length:]
 
     # Save each chunk to a temporary file and store file paths in a list
     temp_files = []

test_handler_files.py

@@ -0,0 +1,40 @@
+import json
+from handler import EndpointHandler
+
+# init handler
+my_handler = EndpointHandler()
+
+import os
+
+# Specify the folder path here
+folder_path = (
+    "/Users/apenasisso/pl/pl-bots/notebooks/analytics/data/elevenlabs_analysis"
+)
+
+# List all files in the folder
+for filename in os.listdir(folder_path):
+    full_filename = os.path.join(folder_path, filename)
+    if os.path.isfile(full_filename) and full_filename.endswith(".wav"):
+        print(full_filename)
+
+        holiday_payload = {
+            "inputs": f"file://{full_filename}"
+            # "inputs": "https://pl-bots-public-media.s3.amazonaws.com/5527999790371_8825d2c9-d87b-49c3-bf70-cb536328ba92.mp3"
+        }
+
+        full_filename_json = full_filename.replace(".wav", ".json")
+
+        if os.path.isfile(full_filename_json):
+            print("file exists", full_filename_json)
+            continue
+
+        # test the handler
+        payload = my_handler(holiday_payload)
+
+        # save payload to file
+
+        with open(full_filename_json, "w") as outfile:
+            json.dump(payload, outfile, indent=4)
+
+        # show results
+        print("holiday_payload", payload)

test_handler_url.py

@@ -0,0 +1,25 @@
+import json
+from handler import EndpointHandler
+
+# init handler
+my_handler = EndpointHandler()
+
+import os
+
+# Specify the folder path here
+folder_path = (
+    "/Users/apenasisso/pl/pl-bots/notebooks/analytics/data/elevenlabs_analysis"
+)
+
+# List all files in the folder
+
+holiday_payload = {
+    "inputs": "https://pl-bots-public-media.s3.amazonaws.com/5511996969344_093275df-8324-4425-be0a-6d933bbcd896.mp3"
+}
+
+# test the handler
+payload = my_handler(holiday_payload)
+
+
+# show results
+print("holiday_payload", payload)