Update app.py

app.py

@@ -175,8 +175,9 @@ 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
-    st.subheader("Extracted Text (Cleaned & Polished)")
     if 'cleaned_text' not in st.session_state:
         st.session_state.cleaned_text = ""
     if 'polished_text' not in st.session_state:
@@ -224,19 +225,21 @@ if uploaded_file:
         st.session_state.cleaned_text = cleaned_text
         st.session_state.polished_text = polished_text
 
-# Display extracted text
-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 = search_query.length
-    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)")
+        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
+            end = index + len
+            if index != -1:
+                highlight_text(st.session_state.cleaned_text, start, end)
+            else:
+                st.write("No Search Found.")