test: singular file download option

.gitignore

@@ -1 +1,2 @@
 .venv/
+ffmpgvenv/

main.py

@@ -11,6 +11,7 @@ app = FastAPI()
 aws_access_key_id = os.environ['AWS_ACCESS_KEY_ID']
 aws_secret_access_key = os.environ['AWS_SECRET_ACCESS_KEY']
 s3_bucket_name = os.environ['S3_BUCKET_NAME']
+aws_env = os.environ['AWS_ENV']
 aws_region = 'eu-central-1'
 temp_dir = '/tmp/'
 # Initialize an S3 client
@@ -19,25 +20,38 @@ s3_client = boto3.client('s3', aws_access_key_id=aws_access_key_id, aws_secret_a
 class CutRequest(BaseModel):
     segments: list[tuple[float, float]]
 
-def download_files(output_key: str):
-    # Temporary directory to store uploaded audio files
-
-    s3_directory = output_key
-
-    input_files = ['genBase.mp3', 'quote.mp3']
+def download_file(news_name: str,quote_filename: str):
+    # Define a function to download a single file
+    s3_directory = f'{aws_env}/{news_name}'
+    input_file = quote_filename
     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)
+        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}")
+        raise HTTPException(status_code=500, detail=f"Failed to download a file from S3: {e}")
+
+
+# def download_files(output_key: str):
+
+#     # 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}")
 def cut_audio(request: CutRequest, output_key: str):
     download_files(output_key)
     for i, (start, end) in enumerate(request.segments):
-        output_file = f"/tmp/genBase_segment_{i}.mp3"
+        output_file = f"/tmp/cut_quote_{i}.mp3"
+        # output_file = f"/tmp/genBase_segment_{i}.mp3"
         try:
             (
                 ffmpeg

requirements.txt

@@ -19,5 +19,5 @@ six==1.16.0
 sniffio==1.3.0
 starlette==0.32.0.post1
 typing_extensions==4.9.0
-urllib3==2.0.7
+urllib3
 uvicorn==0.25.0