Initial Draft

lib/api/registry.py

@@ -0,0 +1,44 @@
+import logging
+import traceback
+import signal
+import sys
+
+from flask import Flask
+from flask_cors import CORS
+from flask_restful import Api
+
+import lib.api.util.CONFIG as CONFIG
+import lib.api.util.LogWrapper as LogWrapper
+from lib.api.endpoints.VectorStoreAPI import VectorStoreAPI
+
+lv_logger = logging.getLogger(__name__)
+
+def fn_signal_handler(sig, frame):
+    print('Flask server shutting down...')
+    sys.exit(0)
+
+def start_rest_app():
+    LogWrapper.fn_start_logging()
+
+    try:
+        lv_app = Flask(__name__)
+        CORS(lv_app)
+        lv_api = Api(lv_app)
+        lv_api.add_resource(
+                            VectorStoreAPI, 
+                            '/api/vectordb/create'
+                           )
+        lv_logger.info("Running FlaskServer")
+        signal.signal(signal.SIGTERM, fn_signal_handler)
+        signal.signal(signal.SIGINT, fn_signal_handler)
+        lv_app.run( 
+                    use_reloader=True,
+                    host='0.0.0.0',
+                    port=int(CONFIG.SERVER_PORT),
+                    debug=True
+                  )
+    except Exception as e:
+        lv_logger.error(e)
+        lv_logger.error(traceback.format_exc())
+        lv_logger.error("Unable to start FlaskServer")
+        return None