Update app.py

app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, HTTPException
-from fastapi.responses import JSONResponse, StreamingResponse
+from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 import numpy as np
 import io
@@ -13,6 +13,11 @@ import magic  # For MIME type detection
 from pydub import AudioSegment
 import traceback
 from logging.handlers import RotatingFileHandler
+import os
+import boto3
+from botocore.exceptions import NoCredentialsError
+from urllib.parse import quote
+import time
 
 # Import functions from other modules
 from asr import transcribe, ASR_LANGUAGES
@@ -33,6 +38,20 @@ logger.addHandler(file_handler)
 
 app = FastAPI(title="MMS: Scaling Speech Technology to 1000+ languages")
 
+# S3 Configuration
+S3_BUCKET = "afri"
+S3_REGION = "eu-west-3"
+S3_ACCESS_KEY_ID = os.environ.get("AWS_ACCESS_KEY_ID")
+S3_SECRET_ACCESS_KEY = os.environ.get("AWS_SECRET_ACCESS_KEY")
+
+# Initialize S3 client
+s3_client = boto3.client(
+    's3',
+    aws_access_key_id=S3_ACCESS_KEY_ID,
+    aws_secret_access_key=S3_SECRET_ACCESS_KEY,
+    region_name=S3_REGION
+)
+
 # Define request models
 class AudioRequest(BaseModel):
     audio: str  # Base64 encoded audio or video data
@@ -138,17 +157,28 @@ async def synthesize_speech(request: TTSRequest):
         buffer.seek(0)
         logger.info(f"Buffer size: {buffer.getbuffer().nbytes} bytes")
         
-        logger.info("Preparing StreamingResponse")
-        response = StreamingResponse(
-            buffer, 
-            media_type="audio/wav",
-            headers={
-                "Content-Disposition": "attachment; filename=synthesized_audio.wav"
-            }
-        )
-        logger.info("StreamingResponse prepared successfully")
+        # Generate a unique filename
+        filename = f"synthesized_audio_{int(time.time())}.wav"
+        
+        # Upload to S3
+        try:
+            s3_client.upload_fileobj(buffer, S3_BUCKET, filename)
+            logger.info(f"File uploaded successfully to S3: {filename}")
+            
+            # Generate a presigned URL
+            url = s3_client.generate_presigned_url('get_object',
+                                                    Params={'Bucket': S3_BUCKET,
+                                                            'Key': filename},
+                                                    ExpiresIn=3600)  # URL expires in 1 hour
+            
+            encoded_url = quote(url, safe=':/?&=')
+            logger.info(f"Presigned URL generated: {encoded_url}")
+            
+            return JSONResponse(content={"audio_url": encoded_url})
         
-        return response
+        except NoCredentialsError:
+            logger.error("AWS credentials not available")
+            raise HTTPException(status_code=500, detail="Could not upload file to S3")
 
     except ValueError as ve:
         logger.error(f"ValueError in synthesize_speech: {str(ve)}", exc_info=True)