Added header

2 files changed, 26 insertions, 6 deletions

diff --git a/.SRCINFO b/.SRCINFO
index c7f453ad139b..4194363b23b7 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -13,18 +13,24 @@ pkgbase = stt-bin
 	provides = stt
 	conflicts = stt
 	source_x86_64 = https://github.com/coqui-ai/STT/stt/releases/download/v1.3.0/native_client.tflite.Linux.tar.xz
+    source_x86_64 = https://raw.githubusercontent.com/coqui-ai/STT/v1.3.0/native_client/coqui-stt.h
 	source_x86_64 = stt.pc
 	source_x86_64 = stt-tflite.pc
 	sha512sums_x86_64 = c7903de984f6af069c07a735c1fa0cda2f9f703313a663e18317e1d422a9f28aa9d879c21a428a4d74baaaa1d027b19d46e2abb15982e5262068a14a1d57e824
-	sha512sums_x86_64 = 30139b795537a9a21187889daac8f24fb0ab542363a8c59c6bd53ccf5883cdf49f300d8532e1c9ed9610f75237a22948da1ec6c5247855c62871a7484b2e34da
+    sha512sums_x86_64 = 1e4238c018d8c1092eb23c34650cd73e1a1940740054c9674e3f8a2f091c1e6665207c2b8ff8da3ca75c06a2fc7129668798e392f57c69a7b0ff095c90e179bc
+    sha512sums_x86_64 = 30139b795537a9a21187889daac8f24fb0ab542363a8c59c6bd53ccf5883cdf49f300d8532e1c9ed9610f75237a22948da1ec6c5247855c62871a7484b2e34da
 	sha512sums_x86_64 = 431f552a22e2a2030a99a47f00b2c2d56cbb7adc5e2d3e78431536ae209bcc76f2324f055889d3e7e371f0e7123b845050b4bfeb242bc766fc060a32354329f5
 	source_armv7h = https://github.com/coqui-ai/STT/stt/releases/download/v1.3.0/native_client.tflite.linux.armv7.tar.xz
-	source_armv7h = stt.pc
+    source_armv7h = https://raw.githubusercontent.com/coqui-ai/STT/v1.3.0/native_client/coqui-stt.h
+    source_armv7h = stt.pc
 	sha512sums_armv7h = 687fd924716ebcd53cf1f5ddf1efb44b4005f9a83f4af70756a1146c78b76d09c90add978dba40a4a7ba935a91704defbdbea3360426a09d4ac4c7f6f58248d2
-	sha512sums_armv7h = 30139b795537a9a21187889daac8f24fb0ab542363a8c59c6bd53ccf5883cdf49f300d8532e1c9ed9610f75237a22948da1ec6c5247855c62871a7484b2e34da
+    sha512sums_aemv7h = 1e4238c018d8c1092eb23c34650cd73e1a1940740054c9674e3f8a2f091c1e6665207c2b8ff8da3ca75c06a2fc7129668798e392f57c69a7b0ff095c90e179bc
+    sha512sums_armv7h = 30139b795537a9a21187889daac8f24fb0ab542363a8c59c6bd53ccf5883cdf49f300d8532e1c9ed9610f75237a22948da1ec6c5247855c62871a7484b2e34da
 	source_aarch64 = https://github.com/coqui-ai/STT/stt/releases/download/v1.3.0/native_client.tflite.linux.aarch64.tar.xz
+    source_aarch64 = https://raw.githubusercontent.com/coqui-ai/STT/v1.3.0/native_client/coqui-stt.h
 	source_aarch64 = stt.pc
 	sha512sums_aarch64 = 041400ee01bf0b863db60001972938aa1add4208df8e61a068bd4f872fe16fe9fccb23aa84c703e1181590ce19ded5453ace61670a57dfddbc18e5ef4b86c106
-	sha512sums_aarch64 = 30139b795537a9a21187889daac8f24fb0ab542363a8c59c6bd53ccf5883cdf49f300d8532e1c9ed9610f75237a22948da1ec6c5247855c62871a7484b2e34da
+    sha512sums_aarch64 = 1e4238c018d8c1092eb23c34650cd73e1a1940740054c9674e3f8a2f091c1e6665207c2b8ff8da3ca75c06a2fc7129668798e392f57c69a7b0ff095c90e179bc
+    sha512sums_aarch64 = 30139b795537a9a21187889daac8f24fb0ab542363a8c59c6bd53ccf5883cdf49f300d8532e1c9ed9610f75237a22948da1ec6c5247855c62871a7484b2e34da
 
 pkgname = stt-bin
diff --git a/PKGBUILD b/PKGBUILD
index 95baf6c49f8d..06510f3967ae 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -3,7 +3,7 @@ pkgname=stt-bin
 _pkgname=STT
 pkgver=1.3.0
 pkgrel=1
-pkgdesc="A TensorFlow implementation of Baidu's DeepSpeech architecture - C++ native client + devel files."
+pkgdesc="A TensorFlow implementation of Baidu's DeepSpeech architecture - Python native client + devel files."
 arch=('x86_64' 'armv7h' 'aarch64')
 url="https://github.com/coqui-ai/STT"
 license=('MPL2')
@@ -17,27 +17,34 @@ _src_base_url="https://github.com/coqui-ai/STT/releases/download/v$pkgver"
 _raw_base_url="https://raw.githubusercontent.com/coqui-ai/STT/v$pkgver"
 
 # Reused checksums.
+_header_sum='1e4238c018d8c1092eb23c34650cd73e1a1940740054c9674e3f8a2f091c1e6665207c2b8ff8da3ca75c06a2fc7129668798e392f57c69a7b0ff095c90e179bc'
 _pc_sum='30139b795537a9a21187889daac8f24fb0ab542363a8c59c6bd53ccf5883cdf49f300d8532e1c9ed9610f75237a22948da1ec6c5247855c62871a7484b2e34da'
 _pc_tflite_sum='431f552a22e2a2030a99a47f00b2c2d56cbb7adc5e2d3e78431536ae209bcc76f2324f055889d3e7e371f0e7123b845050b4bfeb242bc766fc060a32354329f5'
 
 # amd64
 source_x86_64=("$_src_base_url/native_client.tflite.Linux.tar.xz"
+               "$_raw_base_url/native_client/coqui-stt.h"
                "stt.pc"
                "stt-tflite.pc")
 sha512sums_x86_64=('c7903de984f6af069c07a735c1fa0cda2f9f703313a663e18317e1d422a9f28aa9d879c21a428a4d74baaaa1d027b19d46e2abb15982e5262068a14a1d57e824'
+                   "$_header_sum"
                    "$_pc_sum"
                    "$_pc_tflite_sum")
 
 # armv7h
 source_armv7h=("$_src_base_url/native_client.tflite.linux.armv7.tar.xz"
+               "$_raw_base_url/native_client/coqui-stt.h"
                "stt.pc")
 sha512sums_armv7h=('687fd924716ebcd53cf1f5ddf1efb44b4005f9a83f4af70756a1146c78b76d09c90add978dba40a4a7ba935a91704defbdbea3360426a09d4ac4c7f6f58248d2'
+                   "$_header_sum"
                    "$_pc_sum")
 
 # aarch64
 source_aarch64=("$_src_base_url/native_client.tflite.linux.aarch64.tar.xz"
+               "$_raw_base_url/native_client/coqui-stt.h"
                 "stt.pc")
 sha512sums_aarch64=('041400ee01bf0b863db60001972938aa1add4208df8e61a068bd4f872fe16fe9fccb23aa84c703e1181590ce19ded5453ace61670a57dfddbc18e5ef4b86c106'
+                    "$_header_sum"
                     "$_pc_sum")
 
 package() {
@@ -81,6 +88,13 @@ package() {
   # Install files.
   install -Dm755 libstt.so "$pkgdir/usr/lib/libstt.so.$pkgver"
   ln -s "/usr/lib/libstt.so.$pkgver" "$pkgdir/usr/lib/libstt.so"
-  #install -Dm644 stt.h "$pkgdir/usr/include"
+  install -Dm644 coqui-stt.h "$pkgdir/usr/include"
   install -Dm644 stt.pc "$pkgdir/usr/lib/pkgconfig/stt.pc"
+  
+  # Python bindings
+  PIP_CONFIG_FILE=/dev/null pip install --isolated --root="$pkgdir" --ignore-installed --no-deps python/dist/stt-*.whl
+  PIP_CONFIG_FILE=/dev/null pip install --isolated --root="$pkgdir" --ignore-installed --no-deps ctcdecode/dist/*.whl
+  #mv "$pkgdir/usr/bin/stt" "$pkgdir/usr/bin/stt_python"
+  cp -rv "${srcdir}/${_pkgname}-${pkgver}/training/coqui_stt_training" "$pkgdir"`python -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())"`
+
 }
\ No newline at end of file