Update app.py

app.py

@@ -6,8 +6,8 @@ from PIL import Image
 processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
 model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
 
-# ฟังก์ชันหลักสำหรับตรวจจับชื่อธนาคาร
-def detect_bank(image):
+# ฟังก์ชันหลักสำหรับแสดงข้อความที่ตรวจจับได้จากภาพ
+def ocr_text(image):
     # ตรวจสอบว่าเป็น RGB
     if image.mode != "RGB":
         image = image.convert("RGB")
@@ -15,36 +15,18 @@ def detect_bank(image):
     # ทำ OCR ด้วย TrOCR
     pixel_values = processor(images=image, return_tensors="pt").pixel_values
     generated_ids = model.generate(pixel_values)
-    text = processor.batch_decode(generated_ids, skip_special_tokens=True)
+    text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
-    # ชื่อธนาคารที่รองรับ
-    banks = {
-        "SCB": "ธนาคารไทยพาณิชย์ (SCB)",
-        "กสิกร": "ธนาคารกสิกรไทย",
-        "กรุงเทพ": "ธนาคารกรุงเทพ",
-        "กรุงไทย": "ธนาคารกรุงไทย",
-        "ออมสิน": "ธนาคารออมสิน",
-        "กรุงศรี": "ธนาคารกรุงศรีอยุธยา",
-        "tmb": "ธนาคารทหารไทยธนชาต (ttb)",
-        "ttb": "ธนาคารทหารไทยธนชาต (ttb)",
-    }
-
-    # ตรวจจับชื่อธนาคาร
-    detected = [name for key, name in banks.items() if key.lower() in text.lower()]
-
-    # ตอบกลับตามผลลัพธ์
-    if detected:
-        return f"✅ ตรวจพบธนาคาร: {detected[0]}"
-    else:
-        return f"⚠️ ไม่พบชื่อธนาคารในภาพ\n\n📄 ข้อความที่พบในภาพ:\n{text}"
+    # แสดงข้อความที่ตรวจจับได้
+    return f"📄 ข้อความที่พบในภาพ:\n{text}"
 
 # สร้าง Gradio Interface
 iface = gr.Interface(
-    fn=detect_bank,
-    inputs=gr.Image(type="pil", label="อัปโหลดภาพสลิปหรือใบเสร็จ"),
-    outputs=gr.Textbox(lines=10, label="ผลลัพธ์"),
-    title="🧾 ตรวจจับชื่อธนาคารจากภาพ OCR",
-    description="ระบบจะอ่านข้อความจากภาพ แล้วบอกว่าพบธนาคารอะไรบ้าง หรือแสดงข้อความทั้งหมดถ้าไม่พบ"
+    fn=ocr_text,
+    inputs=gr.Image(type="pil", label="อัปโหลดภาพที่มีข้อความ"),
+    outputs=gr.Textbox(lines=10, label="ข้อความที่ตรวจจับได้"),
+    title="🧾 อ่านข้อความจากภาพ (OCR)",
+    description="ระบบจะอ่านข้อความจากภาพและแสดงผลลัพธ์ทั้งหมดโดยไม่กรอง"
 )
 
 # เปิดแอป