add news_id to request for output storage

main.py

@@ -22,6 +22,7 @@ class CutRequest(BaseModel):
     news_name: str
     quote_filename: str
     segments: list[tuple[float, float]]
+    news_id: int
 
 def download_file(news_name: str, quote_filename: str, new_filename: str = "source.mp3"):
     s3_directory = f'{aws_env}/{news_name}'
@@ -48,7 +49,7 @@ def download_file(news_name: str, quote_filename: str, new_filename: str = "sour
 
 
 @app.post("/cut-audio")
-def cut_audio(request: CutRequest, output_key: str):
+def cut_audio(request: CutRequest):
     download_file(request.news_name,request.quote_filename)
     for i, (start, end) in enumerate(request.segments):
         output_file = f"/tmp/cut_quote_{i}.mp3"
@@ -62,7 +63,7 @@ def cut_audio(request: CutRequest, output_key: str):
         except ffmpeg.Error as e:
             raise HTTPException(status_code=500, detail=str(e))
         try:
-            s3_output_key = f'{output_key}/genBase_segment_{i}.mp3' 
+            s3_output_key = f'{request.news_id}/genBase_segment_{i}.mp3' 
             s3_client.upload_file(output_file, s3_bucket_name, s3_output_key)
         except Exception as e:
             raise HTTPException(status_code=500, detail=f"Failed to upload file to S3: {e}")