Update app.py

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 import io
 
 from ocr_utils import extract_text
-from ocr_utils import highlight_content
+
 
 # Streamlit app layout
 st.title("OCR Text Extraction from Images")
@@ -33,21 +33,16 @@ if uploaded_file is not None:
         highlighted_content = highlight_content(full_text, keyword)
         st.subheader("Highlighted Search Results")
 
-      # CSS styles for highlighting
-        st.markdown('''
-        <style>
-        .highlight {
-            background-color: yellow;
-            color: black;
-            padding: 0.2em;
-            border-radius: 4px;
-        }
-        <style>
-        ''', unsafe_allow_html=True)
-
-        # Render the highlighted content with HTML
-        st.markdown(highlighted_content, unsafe_allow_html=True)
 
+    if keyword:
+        if keyword in full_text:
+            highlighted_text = full_text.replace(
+                keyword, f"<mark style='background-color: yellow; color: black;'>{keyword}</mark>")
+            st.subheader("Highlighted Search Results")
+            st.markdown(highlighted_text, unsafe_allow_html=True)
+        else:
+            st.subheader("Highlighted Search Results")
+            st.write(f"The keyword '{keyword}' was not found in the text.")
     else:
         st.subheader("Highlighted Search Results")
-        st.write("No keyword entered for highlighting.")
+        st.write("No keyword entered for highlighting.")