Update app.py

app.py

@@ -12,6 +12,7 @@ import re
 import json
 import base64
 from groq import Groq
+from st_keyup import st_keyup
 
 # Page configuration
 st.set_page_config(page_title="DualTextOCRFusion", page_icon="🔍", layout="wide")
@@ -51,7 +52,7 @@ def clean_extracted_text(text):
 
 # Polish the text using a model
 def polish_text_with_ai(cleaned_text):
-    prompt = f"Remove unwanted spaces between and inside words to join incomplete words, creating a meaningful sentence in either Hindi, English, or Hinglish without altering any words from the given extracted text. Then, return the corrected text with adjusted spaces, keeping it as close to the original as possible. Extracted Text : {cleaned_text}"
+    prompt = f"Remove unwanted spaces between and inside words to join incomplete words, creating a meaningful sentence in either Hindi, English, or Hinglish without altering any words from the given extracted text. Then, return the corrected text with adjusted spaces, keeping it as close to the original as possible, along with relevant details or insights that an AI can provide about the extracted text.  Extracted Text : {cleaned_text}"
     client = Groq(api_key="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")
     chat_completion = client.chat.completions.create(
         messages=[
@@ -86,11 +87,13 @@ def extract_text_qwen(image_file, model, processor):
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-# Highlight keyword search
+# Function to highlight the keyword in the text
 def highlight_text(text, search_term):
-    if not 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)
 
 # Title and UI
@@ -193,14 +196,10 @@ if uploaded_file:
             else:
                 st.session_state["highlighted_result"] = extracted_text
 
-        search_query = st.text_input(
-            "Search in extracted text:",
-            key="search_query",
-            placeholder="Type to search...",
-            on_change=update_search,
-            disabled=not uploaded_file
-        )
+        # 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)