Update app.py

app.py

@@ -1,28 +1,29 @@
 import streamlit as st
 from PIL import Image
 import pytesseract
-import shutil
+import io
 
-# Streamlit app title
-st.title("Image to Text Extraction App 🖼️📝")
+def perform_ocr(image):
+    try:
+        text = pytesseract.image_to_string(image)
+        return text
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
-# Prompt for image upload
-uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
+def main():
+    st.title("Image to Text Converter")
+    st.write("Upload an image and get the extracted text!")
 
-# Check if tesseract is installed and in PATH
-pytesseract.pytesseract.tesseract_cmd = shutil.which("tesseract") or None
+    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
-# If an image is uploaded, perform OCR
-if uploaded_file is not None:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
-    
-    # Perform OCR using Tesseract
-    with st.spinner("Extracting text..."):
-        text = pytesseract.image_to_string(image)
-        
-    # Display the extracted text
-    st.subheader("Extracted Text:")
-    st.write(text)
-else:
-    st.warning("Please upload an image file to extract text.")
+    if uploaded_file is not None:
+        image = Image.open(uploaded_file)
+        st.image(image, caption='Uploaded Image', use_column_width=True)
+        st.write("")
+        st.write("Extracting text...")
+        text = perform_ocr(image)
+        st.write("Extracted Text:")
+        st.write(text)
+
+if __name__ == "__main__":
+    main()