expand request to involve news name and quote name

main.py

@@ -1,6 +1,7 @@
 from pydantic import BaseModel
 from fastapi import FastAPI, UploadFile, HTTPException
 from ffmpeg import input, output, run
+from botocore.exceptions import ClientError
 import ffmpeg
 import boto3
 import os
@@ -18,44 +19,43 @@ temp_dir = '/tmp/'
 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)
 
 class CutRequest(BaseModel):
+    news_name: str
+    quote_filename: str
     segments: list[tuple[float, float]]
 
-def download_file(news_name: str,quote_filename: str):
-    # Define a function to download a single file
+def download_file(news_name: str, quote_filename: str, new_filename: str = "source.mp3"):
     s3_directory = f'{aws_env}/{news_name}'
-    input_file = quote_filename
+    s3_object_key = f'{s3_directory}/{quote_filename}'
+    local_file = os.path.join(temp_dir, new_filename)
+
     try:
-        s3_object_key = f'{s3_directory}/{input_file}'
-        local_file = temp_dir + input_file
+        # Ensure the download directory exists
+        os.makedirs(temp_dir, exist_ok=True)
+
         s3_client.download_file(s3_bucket_name, s3_object_key, local_file)
+        print(f"File downloaded and saved as: {local_file}")
+        
+        # Return the local_file path
+        return local_file
+
+    except ClientError as e:
+        if e.response['Error']['Code'] == "404":
+            raise HTTPException(status_code=404, detail=f"The file {quote_filename} does not exist in S3.")
+        else:
+            raise HTTPException(status_code=500, detail=f"Failed to download file from S3: {str(e)}")
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Failed to download a file from S3: {e}")
-
-
-# def download_files(output_key: str):
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
 
-#     # Temporary directory to store uploaded audio files
-#     s3_directory = output_key
 
-#     input_files = ['genBase.mp3', 'quote.mp3']
-#     try:
-#         for input_file in input_files:
-#             s3_object_key = f'{s3_directory}/{input_file}'
-#             local_file = temp_dir + input_file
-#             s3_client.download_file(s3_bucket_name, s3_object_key, local_file)
-#     except Exception as e:
-#         raise HTTPException(status_code=500, detail=f"Failed to download files from S3: {e}")
-
-@app.post("/cut-audio/{output_key}")
+@app.post("/cut-audio/")
 def cut_audio(request: CutRequest, output_key: str):
-    download_files(output_key)
+    download_file(request.news_name,request.quote_filename)
     for i, (start, end) in enumerate(request.segments):
         output_file = f"/tmp/cut_quote_{i}.mp3"
-        # output_file = f"/tmp/genBase_segment_{i}.mp3"
         try:
             (
                 ffmpeg
-                .input('/tmp/genBase.mp3', ss=start, to=end)
+                .input('/tmp/source.mp3', ss=start, to=end)
                 .output(output_file)
                 .run()
             )
@@ -69,8 +69,6 @@ def cut_audio(request: CutRequest, output_key: str):
 
     return {"message": "Audio cut successfully"}
 
-
-
 @app.post("/merge_audio/{output_key}")
 async def merge_audio(output_key: str):
     s3_directory = output_key