Add application file

app.py

@@ -1,24 +1,31 @@
 import gradio as gr
 from transformers import pipeline
 
+# โหลดโมเดล
 pipe = pipeline("text-generation", model="openchat/openchat_3.5")
 
-# Prompt แบบ role-play
-prompt = """[INST] Extract structured data from the following message using this form:
+# ฟังก์ชันประมวลผลข้อความ
+def extract_fields(msg):
+    prompt = f"""[INST] Extract structured data from the following message using this form:
 - วันที่ขึ้นของ
 - วันที่ส่งของ
 - ต้นทาง
 - ปลายทาง
 
 Message:
-เข้ารับของวัน พุธ  ที่  28/05/2568
-ส่งโหลด วัน พฤหัส  ที่ 29/05/2568 
-ห้วยกระเจา - BBL
-เข้ารับของวัน พฤหัส  ที่  29/05/2568
-ส่งโหลด วัน ศุกร์  ที่ 30/05/2568 
-ห้วยกระเจา - BBL
+{msg}
 [/INST]
 """
+    result = pipe(prompt, max_new_tokens=300, do_sample=False, temperature=0.3)
+    return result[0]["generated_text"]
 
-result = pipe(prompt, max_new_tokens=300, do_sample=False)
-print(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()