app.py

import gradio as gr
from transformers import pipeline

# โหลดโมเดล
pipe = pipeline(
    "zero-shot-classification",
    model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli"
)

# ฟังก์ชันประมวลผลข้อความ
def extract_fields(msg):
    prompt = f"""กรุณาสรุปข้อมูลจากข้อความด้านล่าง แล้วแสดงผลเป็นรายการ
- วันที่ขึ้นของ
- วันที่ส่งของ
- ต้นทาง
- ปลายทาง

Message:
{msg}
[/INST]
"""
    result = pipe(prompt, max_new_tokens=300, do_sample=False, temperature=0.3)
    return result[0]["generated_text"]

# UI ด้วย Gradio
demo = gr.Interface(
    fn=extract_fields,
    inputs=gr.Textbox(lines=10, label="พิมพ์ข้อความจาก LINE"),
    outputs=gr.Textbox(label="ข้อมูลที่สรุปได้"),
    title="สรุปข้อมูลจากข้อความ LINE",
    description="ระบบจะดึงข้อมูล วันที่ขึ้นของ, วันที่ส่งของ, ต้นทาง, ปลายทาง จากข้อความที่ได้รับ"
)

demo.launch()