mend

main.py

@@ -19,15 +19,15 @@ 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)
 
-class CutRequestInput(BaseModel):
-    data: str
-
 class CutRequest(BaseModel):
     quote_filename: str
     segments: list[tuple[float, float]]
     news_name: str
     news_id: int
 
+class CutRequestInput(BaseModel):
+    data: CutRequest
+
 def download_file(news_name: str, quote_filename: str, new_filename: str = "source.mp3"):
     s3_directory = f'{aws_env}/{news_name}'
     s3_object_key = f'{s3_directory}/{quote_filename}'
@@ -54,10 +54,14 @@ def download_file(news_name: str, quote_filename: str, new_filename: str = "sour
 
 @app.post("/cut-audio")
 async def cut_audio(request: CutRequestInput):
+    print(f"Received request: {request}")
+
     try:
-        data = json.loads(request.data)
+        
+        cut_request = request.data
+
+        print(f"Received request: {cut_request}")
 
-        cut_request = CutRequest(**data)
 
         download_file(cut_request.news_name, cut_request.quote_filename)
         for i, segment in enumerate(cut_request.segments):