Expanded search to multiple keywords

app.py

@@ -3,19 +3,23 @@ from main_got import extract_text
 import re
 
 
-def highlight_keywords(text: str, keyword: str) -> str:
+def highlight_keywords(text: str, keywords: str) -> str:
     # Split text into sentences
     sentences = text.split('. ')
     highlighted_text = ""
     
     for sentence in sentences:
-        if keyword.lower() in sentence.lower():
-            # Highlight sentence in red and the keyword in yellow
-            highlighted_sentence = sentence.replace(keyword, f'<mark style="background-color: yellow">{keyword}</mark>')
+        highlighted_sentence = sentence
+        flag = False
+        for keyword in keywords.split():
+            if keyword.lower() in sentence.lower():
+                # Highlight keyword in yellow and set flag to True to highlight whole sentence after cycle
+                highlighted_sentence = highlighted_sentence.replace(keyword, f'<mark style="background-color: yellow">{keyword}</mark>')
+                flag = True
+        if flag:
             highlighted_text += f'<span style="color: red">{highlighted_sentence}.</span> '
         else:
-            highlighted_text += sentence + ". "
-    
+            highlighted_text += sentence + '. '
     return highlighted_text
 
 
@@ -35,5 +39,6 @@ if uploaded_image is not None:
         # Search functionality
         search_query = st.text_input("Enter a keyword to search within the text")
         if search_query:
-            st.markdown(highlight_keywords(extracted_text, search_query), unsafe_allow_html=True)
+            for keyword in search_query.split():
+                st.markdown(highlight_keywords(extracted_text, search_query), unsafe_allow_html=True)