diff --git a/server.py b/server.py index 0bf558b..68a1083 100644 --- a/server.py +++ b/server.py @@ -32,10 +32,17 @@ def upload_to_s3(processed_data, userId): # Upload the buffer to AWS S3 s3.upload_fileobj(buffer, BUCKET_NAME, s3_key, ExtraArgs={'ContentType': 'audio/wav'}) - # Generate the URL of the uploaded file - file_url = f'https://{BUCKET_NAME}.s3.amazonaws.com/{s3_key}' + # Generate a pre-signed URL for the uploaded file + presigned_url = s3.generate_presigned_url( + 'getObject', + Params={ + 'Bucket': BUCKET_NAME, + 'Key': s3_key + }, + ExpiresIn=3600 # URL expiration time in seconds (1 hour) + ) - return file_url + return presigned_url def int16_to_float32(samples): return samples.astype(np.float32) / 32768.0