Update app.py

app.py

@@ -51,11 +51,11 @@ def search_and_highlight(full_text, keyword):
         )  
         results.append(highlighted_context)  
 
-        # Highlight for HTML output  
-        html_text = (  
-            html_text[:start] +  
-            f'<span style="background-color: yellow;">{html_text[start:end]}</span>' +  
-            html_text[end:]  
+        # Highlight for full text
+        highlighted_text = (
+            highlighted_text[:start] +
+            f'<span style="background-color: yellow;">{highlighted_text[start:end]}</span>' +
+            highlighted_text[end:]
         )  
 
     return results, html_text