Update app.py

app.py

@@ -11,6 +11,11 @@ from st_keyup import st_keyup
 from st_img_pastebutton import paste
 from text_highlighter import text_highlighter
 
+if 'cleaned_text' not in st.session_state:
+    st.session_state.cleaned_text = ""
+if 'polished_text' not in st.session_state:
+    st.session_state.polished_text = ""
+
 # Page configuration
 st.set_page_config(page_title="DualTextOCRFusion",
                    page_icon="🔍", layout="wide")
@@ -174,13 +179,6 @@ if uploaded_file:
     os.makedirs(results_dir, exist_ok=True)
     result_path = os.path.join(
         results_dir, "temp_file_result.json" if clipboard_use else f"{uploaded_file.name}_result.json")
-
-    # Display extracted text
-    if 'cleaned_text' not in st.session_state:
-        st.session_state.cleaned_text = ""
-    if 'polished_text' not in st.session_state:
-        st.session_state.polished_text = ""
-
     # Handle predictions
     if predict_button:
         if os.path.exists(result_path):