Update app.py

app.py

@@ -107,9 +107,9 @@ def extract_text_qwen(image_file, model, processor):
 # Function to highlight the keyword in the text
 
 
-def highlight_text(cleaned_text, start, end):
+def highlight_text(text_sentence, start, end):
     text_highlighter(
-        text=cleaned_text,
+        text=text_sentence,
         labels=[("KEYWORD", "#0000FF")],
         annotations=[
             {"start": start, "end": end, "tag": "KEYWORD"},
@@ -218,11 +218,11 @@ if uploaded_file:
         search_query = st_keyup("Search in extracted text:")
 
         if search_query:
-            index = cleaned_text.find(search_query)
+            index = extracted_text.find(search_query)
             start = index
             len = search_query.length
             end = index + len
             if index != -1:
-                highlight_text(cleaned_text, start, end)
+                highlight_text(extracted_text, start, end)
             else:
                 st.write("No Search Found.")