Update src/streamlit_app.py

src/streamlit_app.py

@@ -31,13 +31,18 @@ def process_image(image_path, gpu=False):
     return img, saved_text
 
 if uploaded_file:
+    st.success("✅ Image uploaded successfully!")
+
     # Save the uploaded file temporarily
     with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
         temp_file.write(uploaded_file.read())
         temp_path = temp_file.name
 
-    # Process image
-    img_result, extracted_text = process_image(temp_path)
+    with st.spinner("🔍 Running OCR..."):
+        # Process image
+        img_result, extracted_text = process_image(temp_path)
+
+    st.success("🎉 OCR complete!")
 
     # Show image with bounding boxes
     st.image(cv2.cvtColor(img_result, cv2.COLOR_BGR2RGB), caption="Detected Text", use_column_width=True)