From 27776f8e880c13faf5b4212219336a79bb94cddb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcin=20W=C4=85troba?= <markowanga@gmail.com>
Date: Tue, 17 Aug 2021 13:46:22 +0200
Subject: [PATCH] Add web embeddings client
---
sziszapangma/integration/asr_processor.py | 2 +-
.../integration/base_asr_service/__init__.py | 0
.../base_asr_service/asr_processor.py | 67 -------------------
.../base_asr_service/asr_result.py | 8 ---
4 files changed, 1 insertion(+), 76 deletions(-)
delete mode 100644 sziszapangma/integration/base_asr_service/__init__.py
delete mode 100644 sziszapangma/integration/base_asr_service/asr_processor.py
delete mode 100644 sziszapangma/integration/base_asr_service/asr_result.py
diff --git a/sziszapangma/integration/asr_processor.py b/sziszapangma/integration/asr_processor.py
index 8d79b6a..83cecdb 100644
--- a/sziszapangma/integration/asr_processor.py
+++ b/sziszapangma/integration/asr_processor.py
@@ -30,7 +30,7 @@ class AsrWebClient(AsrProcessor):
if self._auth_token is not None
else dict()
)
- res = requests.post(self._url, files=files, headers=headers)
+ res = requests.post(self._url, files=files, headers=headers, timeout=600)
json_response = res.json()
print(json_response)
return json_response
diff --git a/sziszapangma/integration/base_asr_service/__init__.py b/sziszapangma/integration/base_asr_service/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/sziszapangma/integration/base_asr_service/asr_processor.py b/sziszapangma/integration/base_asr_service/asr_processor.py
deleted file mode 100644
index df684d8..0000000
--- a/sziszapangma/integration/base_asr_service/asr_processor.py
+++ /dev/null
@@ -1,67 +0,0 @@
-import os
-import traceback
-import uuid
-from abc import ABC, abstractmethod
-from pathlib import Path
-from typing import Optional
-
-from flask import Flask, Response, jsonify, request
-from flask_httpauth import HTTPTokenAuth
-
-from sziszapangma.integration.base_asr_service.asr_result import AsrResult
-
-_TEMP_DIRECTORY = "asr_processing"
-_AUTH_TOKEN = "AUTH_TOKEN"
-_SERVICE_PORT = "SERVICE_PORT"
-
-
-class AsrProcessor(ABC):
- user_token: str
-
- def __init__(self):
- self.user_token = os.environ[_AUTH_TOKEN]
-
- @abstractmethod
- def process_asr(self, audio_file_path: str) -> AsrResult:
- """Method to call for ASR results."""
-
- def process_request(self) -> Response:
- file_tag = str(uuid.uuid4())
- f = request.files["file"]
- if f is not None and f.filename is not None:
- file_extension = f.filename.split(".")[-1]
- file_name = f"{file_tag}.{file_extension}"
- file_path = f"{_TEMP_DIRECTORY}/{file_name}"
- f.save(file_path)
- try:
- transcription = self.process_asr(file_path)
- os.remove(file_path)
- result_object = jsonify(
- {"transcription": transcription.words, "full_text": transcription.full_text}
- )
- except Exception as exception:
- print(exception)
- traceback.print_exc()
- result_object = jsonify({"error": "Error on asr processing"})
- else:
- result_object = jsonify({"error": "Error on asr processing"})
- return result_object
-
- def is_token_correct(self, token: str) -> Optional[str]:
- if token == self.user_token:
- return "asr_client"
- else:
- return None
-
- def health_check(self) -> Response:
- return jsonify({"status": "running"})
-
- def start_processor(self):
- app = Flask(__name__)
- auth = HTTPTokenAuth(scheme="Bearer")
- auth.verify_token(self.is_token_correct)
- Path(_TEMP_DIRECTORY).mkdir(parents=True, exist_ok=True)
- app.route("/process_asr", methods=["POST"])(auth.login_required(self.process_request))
- app.route("/health_check", methods=["GET"])(self.health_check)
- port = int(os.environ[_SERVICE_PORT]) if _SERVICE_PORT in os.environ else 5000
- app.run(debug=True, host="0.0.0.0", port=port)
diff --git a/sziszapangma/integration/base_asr_service/asr_result.py b/sziszapangma/integration/base_asr_service/asr_result.py
deleted file mode 100644
index cdf7c20..0000000
--- a/sziszapangma/integration/base_asr_service/asr_result.py
+++ /dev/null
@@ -1,8 +0,0 @@
-from dataclasses import dataclass
-from typing import List
-
-
-@dataclass(frozen=True)
-class AsrResult:
- words: List[str]
- full_text: str
--
GitLab