Did stuff

extractor_service/extraction.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python3
+import json
+from json import JSONEncoder
+import numpy
+import os
+import os.path
+import logging
+import requests
+from pyAudioAnalysis import audioBasicIO
+from pyAudioAnalysis import ShortTermFeatures
+
+
+class NumpyArrayEncoder(JSONEncoder):
+    def default(self, obj):
+        if isinstance(obj, numpy.ndarray):
+            return obj.tolist()
+        return JSONEncoder.default(self, obj)
+
+
+def do_extraction(file_path: str):
+    logging.info("Running extraction...")
+
+    [Fs, x] = audioBasicIO.read_audio_file(file_path)
+    F, f_names = ShortTermFeatures.feature_extraction(x, Fs, 0.050 * Fs, 0.025 * Fs)
+
+    return {"F": F, "f_names": f_names}
+
+
+def run_everything(parameters: dict):
+    tag = parameters['tag']
+    logging.info(f"Downloading sample: {tag}")
+
+    file_path = os.path.join("/tmp/extractor-service/", f"{tag}.wav")
+    r = requests.get(f"http://storage-service/object/{tag}")
+    with open(file_path, 'wb') as f:
+        f.write(r.content)
+
+    # download done. Do extraction magic
+    try:
+        results = do_extraction(file_path)
+    finally:
+        os.remove(file_path)
+
+    logging.info(f"Pushing results to AI service...")
+
+    response = {
+        "tag": tag,
+        "results": results
+    }
+
+    logging.debug(f"Data being pushed: {str(response)}")
+
+    # r = requests.post('http://ai-service/asd', data=json.dumps(results, cls=NumpyArrayEncoder), headers={'Content-Type': 'application/json'})
+
+    # r.raise_for_status()