Update app.py

app.py

@@ -19,6 +19,8 @@
 import streamlit as st
 import streamlit.components.v1 as components
 from PIL import ImageOps, Image
+import cv2
+import numpy as np
 import torch
 from ultralytics import YOLO
 from src.tiles import *
@@ -197,6 +199,14 @@ if st.session_state.screen == 2:
     with row5_col2:
         # st.image(st.session_state.image, use_column_width=True)
         st.session_state.boxed_image = cv2.imread(st.session_state.image)
+        # st.session_state.boxed_image = cv2.cvtColor(st.session_state.boxed_image, cv2.COLOR_BGR2RGB)
+        height, width, channels = st.session_state.boxed_image.shape
+        st.session_state.blank_img = np.zeros((height + 640, width + 640, 3), np.uint8)
+        st.session_state.blank_img[0:height, 0:width] = st.session_state.boxed_image
+        for i in range(0, height, 600):
+            for j in range(0, width, 600):
+                cv2.rectangle(st.session_state.blank_img, (j, i), (j + 640, i + 640), (0, 255, 0), 5)
+                cv2.putText(st.session_state.blank_img, str(int(i / 600) * 10 + int(j / 600)), (j + 200, i + 320), cv2.FONT_HERSHEY_SIMPLEX, 10, (0, 255, 0), 10)        
         st.image(st.session_state.boxed_image, use_column_width=True) 
     
     with row5_col3: