integrate json_stripper into /cut_audio

list.txt

@@ -1,2 +1,2 @@
 file '/tmp/genBase.mp3'
-file '/tmp/quote.mp3'
+file '/tmp/quote.mp3'

main.py

@@ -24,7 +24,7 @@ class CutRequest(BaseModel):
     news_name: str
     news_id: int
 
-async def json_stripper(strigified_json):
+async def json_stripper(request: CutRequest):
     raw_body = await request.body()
     try:
         # Decode the raw body
@@ -39,10 +39,7 @@ async def json_stripper(strigified_json):
         # Parse the JSON
         body = json.loads(body_str)
         
-        cut_request = CutRequest(**body)
-        
-        print(cut_request)
-        return {"message": "Request processed successfully"}
+        return body
     except json.JSONDecodeError as e:
         print(f"JSON Decode Error: {e}")
         raise HTTPException(status_code=400, detail=f"Invalid JSON: {str(e)}")
@@ -75,10 +72,19 @@ def download_file(news_name: str, quote_filename: str, new_filename: str = "sour
 
 
 @app.post("/cut-audio")
-def cut_audio(request: CutRequest):
-    request = json_stripper(request)
-    download_file(request.news_name,request.quote_filename)
-    for i, (start, end) in enumerate(request.segments):
+async def cut_audio(request: CutRequest):
+    # Use json_stripper to parse the request body
+    body = await json_stripper(request)
+    
+    # Validate the body against CutRequest model
+    try:
+        cut_request = CutRequest(**body)
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+    # Now you can use cut_request in your function
+    download_file(cut_request.news_name, cut_request.quote_filename)
+    for i, (start, end) in enumerate(cut_request.segments):
         output_file = f"/tmp/cut_quote_{i}.mp3"
         try:
             (
@@ -90,7 +96,7 @@ def cut_audio(request: CutRequest):
         except ffmpeg.Error as e:
             raise HTTPException(status_code=500, detail=str(e))
         try:
-            s3_output_key = f'{aws_env}/{request.news_id}/genBase_segment_{i}.mp3' 
+            s3_output_key = f'{aws_env}/{cut_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}")