Added file persistence

main.py

@@ -5,7 +5,7 @@ import subprocess
 import os
 import uuid
 import logging
-import ffmpeg
+from fastapi.staticfiles import StaticFiles
 
 class Url(BaseModel):
     url: str
@@ -16,7 +16,7 @@ logging.basicConfig(format=format, level=logging.DEBUG,
 MODEL=os.environ['model']
 
 app = FastAPI()
-
+app.mount("/tracks", StaticFiles(directory="/tmp/holosubs/results"), name="tracks")
 @app.post("/captions/")
 def read_root(url: Url):
     # Download wav file and get filename
@@ -25,8 +25,9 @@ def read_root(url: Url):
     filename=ytaudio.filename
     # Resample file
     ytaudio.resample('16k')
-    # Generate subtitles 
-    output_file=os.path.join("/tmp/holosubs/results", str(uuid.uuid4()))
+    # Generate subtitles
+    captionFilename=str(uuid.uuid4())
+    output_file=os.path.join("/tmp/holosubs/results", captionFilename)
     logging.info(f'Writing to file {output_file}.vtt')
     cmd=['/usr/local/bin/whisper','-m',f'/home/user/root/models/ggml-{MODEL}.bin'
          ,'-f',filename, '-di', '-of', output_file, '-tr', '-ovtt', '-t', '8']
@@ -39,5 +40,5 @@ def read_root(url: Url):
         raise HTTPException(status_code=500, detail="Whisper translation failed")
     with open(output_file+".vtt", 'r') as f:
         raw_vtt=f.read()
-    os.remove(output_file+".vtt")
-    return {"captions": raw_vtt}
+    return {"captions": f"{captionFilename}.vtt"}
+