From ebd2defd45ac7942ac1a67cb14109cec9662248a Mon Sep 17 00:00:00 2001
From: anjus <anjuskumar1313@gmail.com>
Date: Wed, 7 Aug 2024 14:16:32 -0400
Subject: [PATCH] Code cleanup

---
 app_doctr.py | 18 ++++--------------
 1 file changed, 4 insertions(+), 14 deletions(-)

diff --git a/app_doctr.py b/app_doctr.py
index 467b121..96202d4 100644
--- a/app_doctr.py
+++ b/app_doctr.py
@@ -2,7 +2,7 @@
 import copy
 import json
 import os
-
+import pandas as pd
 import requests
 import streamlit as st
 from requests.auth import HTTPBasicAuth
@@ -43,11 +43,6 @@
 PAGE_REVIEWED_INDICATOR = "✓"
 
 # Wrapper functions
-@st.cache_data(show_spinner=False)
-def get_results_wrapper(tally_sheet):
-    """A wrapper function for caching the get_results function."""
-    return ocr_functions.get_results(tally_sheet)
-
 @st.cache_resource
 def create_ocr():
     """
@@ -396,9 +391,7 @@ def authenticate():
             if 'page_nums' in st.session_state:
                 del st.session_state['page_nums']
             if 'pages_confirmed' in st.session_state:
-                del st.session_state['pages_confirmed']
-            if 'execute_once' in st.session_state:
-                del st.session_state['execute_once']    
+                del st.session_state['pages_confirmed']    
             st.rerun()
 
         # Sidebar for header data
@@ -489,9 +482,7 @@ def authenticate():
         if 'data_payload' not in st.session_state:
             st.session_state.data_payload = None
         if 'pages_confirmed' not in st.session_state:
-            st.session_state['pages_confirmed'] = False
-        if 'execute_once' not in st.session_state:
-            st.session_state['execute_once'] = True    
+            st.session_state['pages_confirmed'] = False  
 
         # Displaying the editable information
         # Used for multipage selection functionality
@@ -508,6 +499,7 @@ def authenticate():
         # Uploading the tables, adding columns for each name
         for i, (df, page_num) in enumerate(zip(st.session_state.table_dfs, st.session_state.page_nums)):
             if page_num != page_selected:
+                table_dfs[i] = pd.DataFrame(df)
                 continue
             int_page_num = int(page_num.replace(PAGE_REVIEWED_INDICATOR, "").strip())
             st.write(f"Table {i + 1}")
@@ -558,8 +550,6 @@ def authenticate():
                 save_st_table(table_dfs)
                 st.rerun()
                 
-            st.write(st.session_state.table_dfs)
-            st.write(table_dfs)
             # Generate and display key-value pairs button
             if st.button("Generate key value pairs", type="primary", disabled=not st.session_state.pages_confirmed):
                 try: