Update app.py

app.py

@@ -175,8 +175,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")
 
-    cleaned_text = ""
-    polished_text = ""
     # Display extracted text
     if 'cleaned_text' not in st.session_state:
         st.session_state.cleaned_text = ""
@@ -225,19 +223,18 @@ if uploaded_file:
         st.session_state.cleaned_text = cleaned_text
         st.session_state.polished_text = polished_text
 
+    if os.path.exists(result_path) and st.session_state.cleaned_text != "" and st.session_state.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)
-
-    if os.path.exists(result_path):
         # 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 = search_query.length
+            len = len(search_query)
             end = index + len
             if index != -1:
                 highlight_text(st.session_state.cleaned_text, start, end)