Fix bug

app.py

@@ -193,7 +193,7 @@ def visualize_structure(pil_img, structure_result):
         y2 = int((min_y + h / 2) * height)
         # print(x1, y1, x2, y2)
         
-        if score >= structure_class_map[structure_class_names[class_id]]:
+        if score >= structure_class_thresholds[structure_class_names[class_id]]:
             cv2.rectangle(image, (x1, y1), (x2, y2), color=(0, 0, 255))
             #cv2.putText(image, str(i)+'-'+str(class_id), (x1-10, y1), cv2.FONT_HERSHEY_SIMPLEX, fontScale=1, color=(0,0,255))
     
@@ -387,18 +387,18 @@ def main():
         else:
             print(filename)
             pil_img = PIL.Image.open(filename)
+            cols[0].image(pil_img)
 
             detection_result = table_detection(pil_img)
             crop_images, vis_det_img = crop_image(pil_img, detection_result)
-            cols[0].image(vis_det_img)
+            cols[1].image(vis_det_img)
 
-            str_cols = st.columns((len(crop_images), ) * 6)
+            str_cols = st.columns((len(crop_images), ) * 5)
             str_cols[0].subheader("Table image")
             str_cols[1].subheader("OCR result")
             str_cols[2].subheader("Structure result")
             str_cols[3].subheader("Cells result")
-            str_cols[4].subheader("HTML result")
-            str_cols[5].subheader("CSV result")
+            str_cols[4].subheader("CSV result")
 
             for img in crop_images:
                 ocr_result = ocr(img)
@@ -407,6 +407,7 @@ def main():
                 cells = extract_text_from_cells(cells)
                 html_result = cells_to_html(cells)
                 df, csv_result = cells_to_csv(cells)
+                print(df)
 
                 vis_ocr_img = visualize_ocr(img, ocr_result)
                 vis_str_img = visualize_structure(img, structure_result)
@@ -416,9 +417,10 @@ def main():
                 str_cols[1].image(vis_ocr_img)
                 str_cols[2].image(vis_str_img)
                 str_cols[3].image(vis_cells_img)
-                str_cols[4].markdown(html_result, unsafe_allow_html=True)
-                str_cols[5].dataframe(df)
-                str_cols[5].download_button("Download table", csv_result, "file.csv", "text/csv", key='download-csv')
+                #str_cols[4].dataframe(df)
+                str_cols[4].download_button("Download table", csv_result, "file.csv", "text/csv", key='download-csv')
+
+                st.markdown(html_result, unsafe_allow_html=True)
 
 
 if __name__ == '__main__':