[wip] enhance

Author: Luan Pham

.env

@@ -2,3 +2,4 @@ TABLE_DETECTION_PORT=9001
 TABLE_RECOGNITION_PORT=9002
 TEXT_DETECTION_PORT=9003
 TEXT_RECOGNITION_PORT=9004
+APP=8501

.gitignore

@@ -8,3 +8,4 @@ data
 .ipynb_checkpoints
 cache
 debug
+__pycache__

docker-compose.yaml

@@ -29,3 +29,18 @@ services:
     volumes:
       - ./text_recognition:/code
     restart: always
+  streamlit:
+    build: streamlit/
+    environment:
+      - TABLE_DETECTION_PORT="${TABLE_DETECTION_PORT}"
+      - TABLE_RECOGNITION_PORT="${TABLE_RECOGNITION_PORT}"
+      - TEXT_DETECTION_PORT="${TEXT_DETECTION_PORT}"
+      - TEXT_RECOGNITION_PORT="${TEXT_RECOGNITION_PORT}"
+    ports:
+      - "${APP}:8501"
+    volumes:
+      - ./streamlit:/code
+    restart: always
+    depends_on:
+      - text_recognition
+      - table_recognition

streamlit/Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.7
+
+WORKDIR /code
+
+COPY requirements.txt /code
+RUN pip install -r requirements.txt
+
+COPY streamlit_app.py classes.py utility.py /code/
+
+EXPOSE 8501
+CMD ["streamlit", "run", "streamlit_app.py"]

classes.py streamlit/classes.py

@@ -5,12 +5,12 @@
 import numpy as np
 import pandas as pd
 import requests
-import streamlit as st
-from dotenv import load_dotenv
-
 from classes import read_tables_from_list, read_texts_from_list
+from dotenv import load_dotenv
 from utility import dump_excel, get_random_color, merge_text_table
 
+import streamlit as st
+
 # load env var
 load_dotenv()
 TABLE_DETECTION_PORT = env["TABLE_DETECTION_PORT"]

requirements.txt streamlit/requirements.txt

@@ -5,7 +5,6 @@
 
 import cv2
 import numpy as np
-
 from classes import Table, Text