Update app/main.py

app/main.py

@@ -3,6 +3,8 @@ import subprocess
 from tempfile import TemporaryDirectory
 from fastapi import FastAPI, UploadFile
 from .srt_to_ssml import sub_to_ssml
+from .s3_handler.py import upload_file
+from uuid import uuid4
 
 
 app = FastAPI()
@@ -14,7 +16,7 @@ async def main():
 @app.post("/")
 async def main(file: UploadFile):
     with TemporaryDirectory(dir=".") as tempdir:
-        text = ""
+        id = str(uuid4())
         audio_file = f'{tempdir}/audio.wav'
         srt_file = f'{tempdir}/audio.srt'
         ssml_file = f'{tempdir}/ssml.txt'
@@ -23,15 +25,15 @@ async def main(file: UploadFile):
         command = ["las", "gen", "-s", audio_file]
         subprocess.run(command)
         sub_to_ssml(srt_file, ssml_file, "hi-IN", "hi-IN-SwaraNeural", "Female", "en", "hi")
-        with open(ssml_file, "r") as f:
-            text = f.read()
-    return {"subtitle": text}
+        srt_url = upload_file(srt_file, "expressapi", "srt", f"{id}.srt")
+        ssml_url = upload_file(ssml_file, "expressapi", "ssml", f"{id}.txt")
+    return {"id":id, "srt_url":srt_url, "ssml_url": ssml_url}
 
     
 @app.post("/url")
 async def url(url: str):
     with TemporaryDirectory(dir=".") as tempdir:
-        text = ""
+        id = str(uuid4())
         input_audio = f'{tempdir}/audio.m4a'
         audio_file = f'{tempdir}/audio.wav'
         srt_file = f'{tempdir}/audio.srt'
@@ -41,9 +43,9 @@ async def url(url: str):
         command = ["las", "gen", "-s", audio_file]
         subprocess.run(command)
         sub_to_ssml(srt_file, ssml_file, "hi-IN", "hi-IN-SwaraNeural", "Female", "en", "hi")
-        with open(ssml_file, "r") as f:
-            text = f.read()
-    return {"subtitle": text}
+        srt_url = upload_file(srt_file, "expressapi", "srt", f"{id}.srt")
+        ssml_url = upload_file(ssml_file, "expressapi", "ssml", f"{id}.txt")
+    return {"id":id, "srt_url":srt_url, "ssml_url": ssml_url}
 
 def download_audio(link, output):
     command = ["yt-dlp", "-f", "ba*[ext=m4a]", "-o", output, link]