dasemu/Transcriptarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update subgen.py

Browse Source
This commit is contained in:
McCloudS
2024-07-13 14:16:36 -06:00
committed by GitHub
parent 72553746fb
commit 8c310427c6
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
subgen.py
View File

@@ -219,7 +219,7 @@ def appendLine(result):
@app.get("/plex") @app.get("/plex")
@app.get("/webhook") @app.get("/webhook")
@app.get("/jellyfin") @app.get("/jellyfin")
@app.get("/asr") @app.get("/")
@app.get("/emby") @app.get("/emby")
@app.get("/detect-language") @app.get("/detect-language")
@app.get("/tautulli") @app.get("/tautulli")
@@ -416,7 +416,7 @@ def batch(
# idea and some code for asr and detect language from https://github.com/ahmetoner/whisper-asr-webservice # idea and some code for asr and detect language from https://github.com/ahmetoner/whisper-asr-webservice
@app.post("//asr") @app.post("//asr")
@app.post("/asr") @app.post("/asr")
def asr( async def asr(
task: Union[str, None] = Query(default="transcribe", enum=["transcribe", "translate"]), task: Union[str, None] = Query(default="transcribe", enum=["transcribe", "translate"]),
language: Union[str, None] = Query(default=None), language: Union[str, None] = Query(default=None),
initial_prompt: Union[str, None] = Query(default=None), #not used by Bazarr initial_prompt: Union[str, None] = Query(default=None), #not used by Bazarr
@@ -454,7 +454,7 @@ def asr(
except Exception as e: except Exception as e:
logging.info(f"Error processing or transcribing Bazarr {audio_file.filename}: {e}") logging.info(f"Error processing or transcribing Bazarr {audio_file.filename}: {e}")
finally: finally:
audio_file.close() await audio_file.close()
task_queue.task_done() task_queue.task_done()
delete_model() delete_model()
if result: if result:
@@ -469,7 +469,7 @@ def asr(
@app.post("//detect-language") @app.post("//detect-language")
@app.post("/detect-language") @app.post("/detect-language")
def detect_language( async def detect_language(
audio_file: UploadFile = File(...), audio_file: UploadFile = File(...),
#encode: bool = Query(default=True, description="Encode audio first through ffmpeg") # This is always false from Bazarr #encode: bool = Query(default=True, description="Encode audio first through ffmpeg") # This is always false from Bazarr
): ):
@@ -493,7 +493,7 @@ def detect_language(
logging.info(f"Error processing or transcribing Bazarr {audio_file.filename}: {e}") logging.info(f"Error processing or transcribing Bazarr {audio_file.filename}: {e}")
finally: finally:
audio_file.close() await audio_file.close()
task_queue.task_done() task_queue.task_done()
delete_model() delete_model()