Update app.py

app.py

@@ -157,6 +157,12 @@ col1, col2 = st.columns([2, 1])
 cleaned_text = ""
 polished_text = ""
 
+# 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 = ""
+
 # Display image preview
 if uploaded_file:
     image = Image.open(uploaded_file)
@@ -178,12 +184,6 @@ if uploaded_file:
     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):
@@ -226,19 +226,21 @@ if uploaded_file:
         st.session_state.cleaned_text = cleaned_text
         st.session_state.polished_text = polished_text
 
-        # Display extracted text
-        st.subheader("Extracted Text (Cleaned & Polished)")
-        st.markdown(st.session_state.cleaned_text, unsafe_allow_html=True)
-        st.markdown(st.session_state.polished_text, unsafe_allow_html=True)
-        # Input search term with real-time update on key press
-        search_query = st_keyup("Search in extracted text:")
-
-        if search_query:
-            index = st.session_state.cleaned_text.find(search_query)
-            start = index
-            len = len(search_query)
-            end = index + len
-            if index != -1:
-                highlight_text(st.session_state.cleaned_text, start, end)
-            else:
-                st.write("No Search Found.")
+# Display extracted text
+st.subheader("Extracted Text (Cleaned & Polished)")
+if st.session_state.cleaned_text:
+    st.markdown(st.session_state.cleaned_text, unsafe_allow_html=True)
+if st.session_state.polished_text:
+    st.markdown(st.session_state.polished_text, unsafe_allow_html=True)
+# Input search term with real-time update on key press
+search_query = st_keyup("Search in extracted text:")
+
+if search_query:
+    index = st.session_state.cleaned_text.find(search_query)
+    start = index
+    len = len(search_query)
+    end = index + len
+    if index != -1:
+        highlight_text(st.session_state.cleaned_text, start, end)
+    else:
+        st.write("No Search Found.")