cut audio into segments

main.py

@@ -1,3 +1,4 @@
+from pydantic import BaseModel
 from fastapi import FastAPI, UploadFile, HTTPException
 from ffmpeg import input, output, run
 import ffmpeg
@@ -11,14 +12,15 @@ aws_access_key_id = os.environ['AWS_ACCESS_KEY_ID']
 aws_secret_access_key = os.environ['AWS_SECRET_ACCESS_KEY']
 s3_bucket_name = os.environ['S3_BUCKET_NAME']
 aws_region = 'eu-central-1'
-
+temp_dir = '/tmp/'
 # Initialize an S3 client
 s3_client = boto3.client('s3', aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key, region_name=aws_region)
 
-@app.post("/merge_audio/{output_key}")
-async def merge_audio(output_key: str):
+class CutRequest(BaseModel):
+    segments: list[tuple[float, float]]
+
+def download_files(output_key: str):
     # Temporary directory to store uploaded audio files
-    temp_dir = '/tmp/'
 
     # Local paths for the fixed input audio files
     gen_base_file = temp_dir + 'genBase.mp3'
@@ -38,6 +40,27 @@ async def merge_audio(output_key: str):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to download files from S3: {e}")
 
+@app.post("/cut-audio/{output_key}")
+def cut_audio(request: CutRequest, output_key: str):
+    download_files(output_key)
+    for i, (start, end) in enumerate(request.segments):
+        output_file = f"/tmp/genBase_segment_{i}.mp3"
+        try:
+            (
+                ffmpeg
+                .input('/tmp/genBase.mp3', ss=start, to=end)
+                .output(output_file)
+                .run()
+            )
+        except ffmpeg.Error as e:
+            raise HTTPException(status_code=500, detail=str(e))
+    return {"message": "Audio cut successfully"}
+
+
+
+@app.post("/merge_audio/{output_key}")
+async def merge_audio(output_key: str):
+    s3_directory = output_key
     # Output file name
     output_file = temp_dir + 'output.mp3'