Update app.py

app.py

@@ -3,22 +3,31 @@ import easyocr
 from PIL import Image
 import numpy as np
 
-# โหลด EasyOCR ด้วย GPU
-reader = easyocr.Reader(['en','th'], gpu=True)
+# โหลด EasyOCR ด้วย GPU (โหลดครั้งเดียวตอนเริ่มแอป)
+reader = easyocr.Reader(['en', 'th'], gpu=True)
 
+# --- Streamlit UI ---
+st.set_page_config(page_title="EasyOCR (GPU) - Fast OCR")
 st.title("EasyOCR (GPU) - Fast OCR")
 st.write("Upload an image to extract text using GPU acceleration")
 
+# อัปโหลดไฟล์
 uploaded_file = st.file_uploader("Choose an image...", type=["png", "jpg", "jpeg"])
 
 if uploaded_file is not None:
+    # เปิดภาพและแสดง
     image = Image.open(uploaded_file)
     st.image(image, caption='Uploaded Image', use_column_width=True)
 
     if st.button("Run OCR"):
-        # แปลงภาพเป็น numpy array
-        img_array = np.array(image)
-        # ใช้ detail=0 ให้คืนแค่ข้อความ
-        result = reader.readtext(img_array, detail=0, paragraph=True)
-        # แสดงผล
-        st.text_area("Detected Text", "\n".join(result), height=300)
+        with st.spinner("Processing..."):
+            # แปลงภาพเป็น numpy array
+            img_array = np.array(image)
+            # ใช้ detail=0 และ paragraph=True
+            result = reader.readtext(img_array, detail=0, paragraph=True)
+            
+            if result:
+                st.success("OCR Completed!")
+                st.text_area("Detected Text", "\n".join(result), height=300)
+            else:
+                st.warning("No text detected in the image.")