update app.py

.gitattributes

@@ -1,4 +1,5 @@
 *.7z filter=lfs diff=lfs merge=lfs -text
+*.ttf filter=lfs diff=lfs merge=lfs -text
 *.arrow filter=lfs diff=lfs merge=lfs -text
 *.bin filter=lfs diff=lfs merge=lfs -text
 *.bz2 filter=lfs diff=lfs merge=lfs -text

.streamlit/config.toml

@@ -4,6 +4,10 @@ showErrorDetails = "none"
 [logger]
 level = "warning"
 
+[ui]
+hideTopBar = true
+hideSidebarNav = true
+
 [server]
 maxUploadSize = 2
 enableXsrfProtection = false

src/NotoSans.ttf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2e68d43ae2c504f4e302a9cf522ecc3f06ef66d724cade58bbe13a3a4af70512
+size 17805476

src/app.py

@@ -5,23 +5,47 @@ logging.basicConfig(
     format="%(asctime)s [%(levelname)s] %(name)s: %(message)s"
 )
 import streamlit as st
+import streamlit.components.v1 as components
 from myocr.pipelines import CommonOCRPipeline
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageFont
 import tempfile
 import pandas as pd
+import os
+
+logger = logging.getLogger(__name__)
+
+st.set_page_config(layout="wide")
+st.title("MyOCR Demo")
 
 # init pipeline
-pipeline = CommonOCRPipeline("cpu")
+@st.cache_resource
+def load_pipeline():
+    pipeline = CommonOCRPipeline("cpu")
+    return pipeline
+
+pipeline = load_pipeline()
+font = ImageFont.truetype("src/NotoSans.ttf", 12) 
+
+def google_analytics():
+    # Google Analytics
+    GA_MEASUREMENT_ID = os.getenv("GOOGLE_ANALYTICS") or "NO_ID"
+    google_analytics_code = f"""
+    <!-- Google tag (gtag.js) -->
+    <script async src="https://www.googletagmanager.com/gtag/js?id={GA_MEASUREMENT_ID}"></script>
+    <script>
+      window.dataLayer = window.dataLayer || [];
+      function gtag(){{dataLayer.push(arguments);}}
+      gtag('js', new Date());
+      gtag('config', '{GA_MEASUREMENT_ID}');
+    </script>
+    """
+    components.html(google_analytics_code, height=0, width=0)
 
 def main():
-    st.set_page_config(layout="wide")
-    st.title("MyOCR Demo")
-    
+    google_analytics()
     left_col, right_col = st.columns([2, 1.5])
-
     with left_col:
-        with st.form("my-form", clear_on_submit=True):
-            uploaded_file = st.file_uploader("", type=["png", "jpg", "jpeg"])
+        uploaded_file = st.file_uploader("", type=["png", "jpg", "jpeg"])
         spinner_container = st.empty()
         image_slot = st.empty()
         if uploaded_file:
@@ -48,7 +72,7 @@ def main():
             top_left = (bbox.left,bbox.top)
             bottom_right = (bbox.right, bbox.bottom )
             draw.rectangle([top_left, bottom_right], outline="red", width=2)
-            draw.text(top_left, text, fill="red")
+            draw.text((bbox.left, 0 if bbox.top - 18<0 else bbox.top - 18), text, font=font, fill="green")
             table_data.append((text, item.confidence))
         image_slot.image(image_with_boxes, use_container_width=True)
     
@@ -58,10 +82,9 @@ def main():
             df = pd.DataFrame(table_data, columns=["Text", "Confidence"])
             st.table(df)
 
-
 if __name__ == "__main__":
     try:
         main()
     except Exception as e:
-        logger.error(f"Error: {e}")
+        logger.error("Error: ", e)
         st.error(f"Internal Error!")