add more logs and upgrade dependencies

Dockerfile-cpu

@@ -4,7 +4,7 @@ WORKDIR /usr/src/app
 
 COPY requirements-cpu.txt ./
 
-RUN pip install --no-cache-dir -r requirements-cpu.txt
+RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements-cpu.txt
 
 COPY . .
 

requirements-cpu.txt

@@ -1,7 +1,7 @@
-Flask==1.0.2
-Flask-Cors==3.0.9
-gevent==1.2.2
-greenlet==0.4.13
+Flask==2.0.3
+Flask-Cors==3.0.10
+gevent==22.10.2
+greenlet==2.0.2
 numpy
 tensorboard==1.15
 tensorflow==1.15

server.py

@@ -1,24 +1,34 @@
+
+print("Program starting ...", flush=True)
+
 import sys
 
 from gevent import monkey
+
+print("Patching monkey ...", flush=True)
 monkey.patch_all()
 
+print("Importing library ...", flush=True)
 from flask import Flask, jsonify, request
 from flask_cors import CORS, cross_origin
-from model import Model
-from gevent.wsgi import WSGIServer
+from gevent.pywsgi import WSGIServer
 from logging.handlers import RotatingFileHandler
 import logging
 
+print("Importing model ...", flush=True)
+from model import Model
 
 
+print("Creating Flask app ...", flush=True)
 app = Flask(__name__)
 CORS(app)
 
 vocab_file = '/data/dl-data/couplet/vocabs'
 model_dir = '/data/dl-data/models/tf-lib/output_couplet_prod'
 
 
+print("Setting up logging ...", flush=True)
+
 def log_setup():
     log_handler = RotatingFileHandler(
         "/logs/service.log",
@@ -39,6 +49,7 @@ def log_setup():
 SPLIT_CHARS = ['，', '、', ',', '.', '。', '!', '！', '?', '？', ' ']
 CENSOR_WORDS_DICT = "/data/censor_words.txt"
 
+logging.info("Loading censor words...")
 with open(CENSOR_WORDS_DICT, encoding='utf-8') as censor_words_file:
     censor_words = [word[:-1] for word in censor_words_file.readlines()]
 logging.info("Loaded %s censor_words" % (len(censor_words)))