From 3ccebc9a7c93f9eb71c87ad99524d716d9a543db Mon Sep 17 00:00:00 2001
From: Rahul Sharma <rahul@propertyguru.com.sg>
Date: Tue, 9 Jun 2020 13:57:34 +0800
Subject: [PATCH] Add docker compose

---
 Dockerfile       |  6 +++---
 requirements.txt |  1 +
 src/main.py      | 12 ++++++++++++
 3 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index bf09bf2..9c89906 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 FROM elasticsearch:7.3.1
-RUN yum install -y https://centos7.iuscommunity.org/ius-release.rpm
+RUN yum install -y https://dl.iuscommunity.org/pub/ius/stable/CentOS/7/x86_64/ius-release-2-1.el7.ius.noarch.rpm
 
 RUN yum clean all && yum update -y && \
     yum install -y \
@@ -8,6 +8,6 @@ RUN yum clean all && yum update -y && \
         python36u-devel \
         python36u-pip
 
-WORKDIR /usr/share/elasticsearch/
+WORKDIR /usr/share/elasticsearch/text-embeddings/
 COPY . /usr/share/elasticsearch/text-embeddings/
-RUN python3.6 -m pip install -r /usr/share/elasticsearch/text-embeddings/requirements.txt
\ No newline at end of file
+RUN python3.6 -m pip install -r /usr/share/elasticsearch/text-embeddings/requirements.txt
diff --git a/requirements.txt b/requirements.txt
index e63d486..b1c1169 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -19,3 +19,4 @@ termcolor==1.1.0
 urllib3==1.25.3
 Werkzeug==0.15.5
 wrapt==1.11.2
+flask==1.1.1
\ No newline at end of file
diff --git a/src/main.py b/src/main.py
index 451754f..4fa610c 100644
--- a/src/main.py
+++ b/src/main.py
@@ -4,10 +4,22 @@
 from elasticsearch.helpers import bulk
 import tensorflow.compat.v1 as tf
 import tensorflow_hub as hub
+from flask import Flask
+
+##### API #####
+
+app = Flask(__name__)
+
+@app.route("/")
+def hello():
+    return "Hello, World!"
 
 ##### INDEXING #####
 
 def index_data():
+    if client.indices.exists(index=INDEX_NAME):
+        return
+
     print("Creating the 'posts' index.")
     client.indices.delete(index=INDEX_NAME, ignore=[404])