Update app.py

app.py

@@ -11,6 +11,7 @@ import base64
 from groq import Groq
 from st_keyup import st_keyup
 from st_img_pastebutton import paste
+from text_highlighter import text_highlighter
 
 # Page configuration
 st.set_page_config(page_title="DualTextOCRFusion",
@@ -102,13 +103,14 @@ def extract_text_qwen(image_file, model, processor):
 # Function to highlight the keyword in the text
 
 
-def highlight_text(text, search_term):
-    if not search_term:  # If no search term is provided, return the original text
-        return text
-    # Use a regular expression to search for the term, case insensitive
-    pattern = re.compile(re.escape(search_term), re.IGNORECASE)
-    # Highlight matched terms with yellow background
-    return pattern.sub(lambda m: f'<span style="background-color: yellow;">{m.group()}</span>', text)
+def highlight_text(cleaned_text,start,end):
+    result = text_highlighter(
+    text=cleaned_text,
+    labels=[("KEYWORD", "#0000FF")],
+    annotations=[
+        {"start": start, "end": end, "tag": "KEYWORD"},
+    ],
+)
 
 
 # Title and UI
@@ -123,11 +125,11 @@ model_choice = st.sidebar.selectbox(
 
 # Upload Section
 uploaded_file = st.sidebar.file_uploader(
-    "Choose an image...", type=["png", "jpg", "jpeg"])
+    "Choose An Image : ", type=["png", "jpg", "jpeg"])
 
 # Input from clipboard
 # Paste image button
-image_data = paste(label="Paste From Clipboard", key="image_clipboard")
+image_data = st.sidebar.paste(label="Paste From Clipboard", key="image_clipboard")
 if image_data is not None:
     clipboard_use = True
     header, encoded = image_data.split(",", 1)
@@ -136,7 +138,7 @@ if image_data is not None:
     uploaded_file = img_stream
 
 # Input from camera
-camera_file = st.sidebar.camera_input("Capture from Camera")
+camera_file = st.sidebar.camera_input("Capture From Camera : ")
 if camera_file:
     uploaded_file = camera_file
 
@@ -206,20 +208,15 @@ if uploaded_file:
         st.markdown(cleaned_text, unsafe_allow_html=True)
         st.markdown(polished_text, unsafe_allow_html=True)
 
-        # Search functionality
-        def update_search():
-            if search_query:
-                highlighted_text = highlight_text(extracted_text, search_query)
-                st.session_state["highlighted_result"] = highlighted_text
-            else:
-                st.session_state["highlighted_result"] = extracted_text
-
         # Input search term with real-time update on key press
-        search_query = st_keyup(
-            "Search in extracted text:", key="search_key", on_change=update_search)
-
-        # Display highlighted results if they exist in session state
-        if "highlighted_result" in st.session_state:
-            st.markdown("### Highlighted Search Results:")
-            st.markdown(
-                st.session_state["highlighted_result"], unsafe_allow_html=True)
+        search_query = st_keyup("Search in extracted text:")
+
+        if search_query:
+            index = cleaned_text.find(search_query)
+            start = index
+            len = search_query.length
+            end = index + len
+            if index != -1:
+                highlight_text(cleaned_text,start,end)
+            else:
+                st.write("No Search Found.")