Fix app.py

app.py

@@ -225,17 +225,26 @@ def main():
     st.set_option('deprecation.showfileUploaderEncoding', False)
     filename = st.sidebar.file_uploader("Upload files", type=['png', 'jpeg', 'jpg'])
 
-    cols[0].image(cv2.imread(filename))
 
-    ocr_res = ocr(filename)
-    structure_result = table_structure(filename)
-    table_structures, cells, confidence_score = convert_stucture(ocr_res, filename, structure_result)
-    visualize_cells(filename, cells, cols[1])
+    if st.sidebar.button("Analyze page"):
 
-    cells = extract_text_from_cells(filename, cells)
-    html_code = cells_to_html(cells)
+        if filename is None:
+            st.sidebar.write("Please upload an image")
 
-    cols[2].html(html_code)
+        else:
+            print(filename)
+
+            cols[0].image(cv2.imread(filename))
+
+            ocr_res = ocr(filename)
+            structure_result = table_structure(filename)
+            table_structures, cells, confidence_score = convert_stucture(ocr_res, filename, structure_result)
+            visualize_cells(filename, cells, cols[1])
+
+            cells = extract_text_from_cells(filename, cells)
+            html_code = cells_to_html(cells)
+
+            cols[2].html(html_code)
 
 
 if __name__ == '__main__':