Minor changes + fix bugs

app.py

@@ -211,7 +211,7 @@ def visualize_ocr(pil_img, ocr_result):
         text = result['text']
         rect = patches.Rectangle(bbox[:2], bbox[2]-bbox[0], bbox[3]-bbox[1], linewidth=2, edgecolor='red', facecolor='none', linestyle='-')
         ax.add_patch(rect)
-        ax.text(bbox[2], bbox[3], text, horizontalalignment='left', verticalalignment='bottom', color='blue', fontsize=7)
+        ax.text(bbox[0], bbox[1], text, horizontalalignment='left', verticalalignment='bottom', color='blue', fontsize=7)
 
     plt.xticks([], [])
     plt.yticks([], [])
@@ -300,11 +300,11 @@ def visualize_structure(pil_img, structure_result):
     for class_name in structure_class_names[:-1]:
         color, alpha, linewidth, hatch = get_bbox_decorations('recognition', structure_class_map[class_name])
         legend_elements.append(
-            Patch(facecolor='none', edgecolor=color, linewidth=linewidth, linestyle='--', label=class_name, hatch=hatch)
+            Patch(facecolor=color, edgecolor=color, linestyle='--', label=class_name, hatch=hatch)
         )
 
     plt.legend(handles=legend_elements, bbox_to_anchor=(0.5, -0.02), loc='upper center', borderaxespad=0,
-                    fontsize=10, ncol=6)
+                    fontsize=10, ncol=3)
     plt.gcf().set_size_inches(10, 10)
     plt.axis('off')
     img_buf = io.BytesIO()
@@ -572,7 +572,6 @@ def main():
 
     st.set_page_config(layout='wide')
     st.title('Table Extraction Demo')
-    st.write('\n')
 
     filename = st.file_uploader('Upload image', type=['png', 'jpeg', 'jpg'])
 
@@ -656,5 +655,6 @@ def main():
                     st.subheader(f'HTML Table {idx + 1}')
                     st.markdown(html_result, unsafe_allow_html=True)
 
+
 if __name__ == '__main__':
     main()