Fix bug

app.py

@@ -169,6 +169,7 @@ def visualize_ocr(pil_img, ocr_result):
         x2 = int(bbox[2])
         y2 = int(bbox[3])
         cv2.rectangle(image, (x1, y1), (x2, y2), color=(0, 255, 0))
+        cv2.putText(image, res['text'], (x1, y1), cv2.FONT_HERSHEY_SIMPLEX, fontScale=0.5, color=(0, 0, 255))
     return cv_to_PIL(image)
 
 
@@ -192,7 +193,7 @@ def visualize_structure(pil_img, structure_result):
         # print(x1, y1, x2, y2)
 
         if score >= structure_class_thresholds[structure_class_names[class_id]]:
-            cv2.rectangle(image, (x1, y1), (x2, y2), color=(0, 0, 255))
+            cv2.rectangle(image, (x1, y1), (x2, y2), color=(0, 255, 0))
             #cv2.putText(image, str(i)+'-'+str(class_id), (x1-10, y1), cv2.FONT_HERSHEY_SIMPLEX, fontScale=1, color=(0,0,255))
 
     return cv_to_PIL(image)
@@ -369,16 +370,16 @@ def main():
     st.title('Table Extraction Demo')
     st.write('\n')
 
+    filename = st.file_uploader('Upload image', type=['png', 'jpeg', 'jpg'])
+
     tabs = st.tabs(
         ['Table Detection', 'Table Structure Recognition']
     )
 
-    filename = st.file_uploader('Upload image', type=['png', 'jpeg', 'jpg'])
-
     if st.button('Analyze image'):
 
         if filename is None:
-            st.sidebar.write('Please upload an image')
+            st.write('Please upload an image')
 
         else:
             print(filename)
@@ -415,12 +416,7 @@ def main():
                     str_cols[1].image(vis_ocr_img)
                     str_cols[2].image(vis_str_img)
                     str_cols[3].image(vis_cells_img)
-
-                    try:
-                        str_cols[4].dataframe(df)
-                    except:
-                        pass
-
+                    #str_cols[4].dataframe(df)
                     str_cols[4].download_button('Download table', csv_result, f'table-{i}.csv', 'text/csv', key=f'download-csv-{i}')
 
                     st.write('\n')