Update app.py

app.py

@@ -1,20 +1,31 @@
 import streamlit as st
 from PIL import Image
 import pytesseract
+import io
 
-st.set_page_config(page_title="OCR App", layout="centered")
+st.set_page_config(page_title="OCR - Image to Text", layout="centered")
 st.title("🖼️ OCR - Image to Text")
 
-uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
+st.write("Upload an image to extract text:")
 
-if uploaded_file:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
+uploaded_file = st.file_uploader("Upload Image", type=["png", "jpg", "jpeg"])
+
+if uploaded_file is not None:
+    try:
+        # Convert uploaded image to bytes and open with PIL
+        image = Image.open(uploaded_file)
+
+        # Show image preview
+        st.image(image, caption="Preview", use_column_width=True)
+
+        if st.button("Extract Text"):
+            with st.spinner("Extracting text..."):
+                text = pytesseract.image_to_string(image)
+            st.success("Text extracted successfully!")
+            st.text_area("Extracted Text", text, height=300)
+
+    except Exception as e:
+        st.error(f"⚠️ Failed to process the image: {e}")
 
-    if st.button("Extract Text"):
-        with st.spinner("Extracting..."):
-            text = pytesseract.image_to_string(image)
-        st.subheader("Extracted Text")
-        st.text_area("Result", text, height=300)