Update app.py

app.py

@@ -22,30 +22,33 @@ if uploaded_file is not None:
     # Display the image  
     st.image(image, caption='Uploaded Image', use_column_width=True)  
     
-    # Perform OCR  
-    text = pytesseract.image_to_string(opencv_image)  
-    
-    st.subheader("Extracted Text:")  
-    st.write(text)  
-
-    # Search functionality  
-    search_keyword = st.text_input("Enter a keyword to search in the extracted text:")  
-    if search_keyword:  
-        pattern = re.compile(re.escape(search_keyword), re.IGNORECASE)  
-        matches = list(pattern.finditer(text))  
+    try:  
+        # Perform OCR  
+        text = pytesseract.image_to_string(opencv_image)  
         
-        if matches:  
-            st.markdown("### Keyword Found:")  
-            for match in matches:  
-                start, end = match.span()  
-                context_start = max(0, start - 50)  
-                context_end = min(len(text), end + 50)  
-                context = text[context_start:context_end]  
-                highlighted_text = (  
-                    context[:start-context_start] +  
-                    f"<span style='background-color: yellow;'>{context[start-context_start:end-context_start]}</span>" +  
-                    context[end-context_start:]  
-                )  
-                st.markdown(f"...{highlighted_text}...")  
-        else:  
-            st.write(f"Keyword '{search_keyword}' not found in the extracted text.")
+        st.subheader("Extracted Text:")  
+        st.write(text)  
+
+        # Search functionality  
+        search_keyword = st.text_input("Enter a keyword to search in the extracted text:")  
+        if search_keyword:  
+            pattern = re.compile(re.escape(search_keyword), re.IGNORECASE)  
+            matches = list(pattern.finditer(text))  
+            
+            if matches:  
+                st.markdown("### Keyword Found:")  
+                for match in matches:  
+                    start, end = match.span()  
+                    context_start = max(0, start - 50)  
+                    context_end = min(len(text), end + 50)  
+                    context = text[context_start:context_end]  
+                    highlighted_text = (  
+                        context[:start-context_start] +  
+                        f"<span style='background-color: yellow;'>{context[start-context_start:end-context_start]}</span>" +  
+                        context[end-context_start:]  
+                    )  
+                    st.markdown(f"...{highlighted_text}...")  
+            else:  
+                st.write(f"Keyword '{search_keyword}' not found in the extracted text.")  
+    except Exception as e:  
+        st.error(f"An error occurred while processing the image: {str(e)}")