Update app.py

app.py

@@ -121,22 +121,24 @@ if uploaded_file is not None:
 
 # Step 9: Implement a search functionality on already extracted text
 if st.session_state.extracted_text:
-    with st.form(key='search_form'):
-        search_query = st.text_input("Enter keyword to search within the extracted text:")
-        search_button = st.form_submit_button("Search")
-
-    if search_button and search_query:
-        # Perform case-insensitive search and highlight the matches
-        extracted_text = st.session_state.extracted_text  # Use already extracted text
-        matches = re.finditer(re.escape(search_query), extracted_text, re.IGNORECASE)
-        
-        highlighted_text = extracted_text
-        result = ''
-        for match in matches:
-            start, end = match.span()
-            result = "**" + highlighted_text[start:end] + "**"
+    with st.form(key='text_search_form'):
+        search_input = st.text_input("Enter a keyword to search within the extracted text:")
+        search_action = st.form_submit_button("Search")
+
+    if search_action and search_input:
+        # Split the extracted text into lines for searching
+        full_text = st.session_state.exttracted_text
+        lines = full_text.split('\n')
+
+        results = []
+        # Search for keyword in each line and collect lines that contain the keyword
+        for line in lines:
+            if re.search(re.escape(search_input), line, re.IGNORECASE):
+                # Highlight keyword in the line
+                highlighted_line = re.sub(f"({re.escape(search_input)})", r"*\1*", line, flags=re.IGNORECASE)
+                results.append(highlighted_line)
         
         st.subheader("Search Results:")
-        if result == '':
+        if result == []:
             st.markdown('Not forund')
         st.markdown(result)