Update app.py

app.py

@@ -24,12 +24,15 @@ def run_GOT(image, tokenizer, model):
     image_path = f"{unique_id}.png"
     
     image.save(image_path)
+    st.write(f"Saved image to {image_path}")
 
     try:
         # Use the model to extract text
         res = model.chat(tokenizer, image_path, ocr_type='ocr')  # Extract plain text
+        st.write(f"Raw result: {res}")  # Debug output
         return res
     except Exception as e:
+        st.error(f"Error: {str(e)}")  # Display any errors
         return f"Error: {str(e)}"
     finally:
         # Clean up the saved image
@@ -39,12 +42,7 @@ def run_GOT(image, tokenizer, model):
 # Function to highlight keyword in text
 def highlight_keyword(text, keyword):
     if keyword:
-        # Use a case-insensitive search for highlighting
-        highlighted_text = text
-        highlighted_text = highlighted_text.replace(keyword, f"<mark>{keyword}</mark>")
-        highlighted_text = highlighted_text.replace(keyword.lower(), f"<mark>{keyword.lower()}</mark>")
-        highlighted_text = highlighted_text.replace(keyword.upper(), f"<mark>{keyword.upper()}</mark>")
-        highlighted_text = highlighted_text.replace(keyword.capitalize(), f"<mark>{keyword.capitalize()}</mark>")
+        highlighted_text = text.replace(keyword, f"<mark>{keyword}</mark>")
         return highlighted_text
     return text
 
@@ -87,3 +85,4 @@ if uploaded_image:
                     st.markdown(highlighted_text, unsafe_allow_html=True)
                 else:
                     st.error(result_text)
+