fix refactor output

.gitignore

@@ -6,4 +6,4 @@
 
 # temporary files
 runtime_log.csv
-speedtest.py
+speedtest.py    

interface.py

@@ -84,8 +84,8 @@ def create_gradio_interface():
 
         # Main Input Section
         with gr.Row():
-           with gr.Row():
-            chat_box = gr.Chatbot(label="Chat with MALI Nurse")
+           with gr.Column():
+            chat_box = gr.Chatbot(label="Chat with MALI Nurse", scale=1)
 
         # Input Section
         with gr.Row():
@@ -144,17 +144,25 @@ def create_gradio_interface():
                 inputs=[],
                 outputs=chat_history_output,
             )
-            # View EHR Details
-            ehr_details_button.click(
+            send_button.click(
+            fn=view_chat_history,
+                inputs=[],
+                outputs=chat_history_output,
+            )
+
+            send_button.click(
+            fn=view_chat_history,
                 fn=view_ehr_details,
                 inputs=[gr.Textbox(value="details", visible=False)],
                 outputs=ehr_details_output
             )
-            ehr_details_button.click(
+
+            send_button.click(
+            fn=view_chat_history,
                 fn=view_ehr_details,
                 inputs=[gr.Textbox(value="prompt", visible=False)],
                 outputs=ehr_prompt_output
-            )            
+            )         
 
         # Footer
         gr.Markdown(

llm/llm.py

@@ -1,15 +1,10 @@
-
-import cProfile
-import pstats
-from langchain_openai.chat_models import ChatOpenAI
-from langchain_groq import ChatGroq
 from langchain.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate
 from pydantic import ValidationError
 import json
 from pprint import pprint
 from llm.basemodel import EHRModel
 from llm.prompt import field_descriptions, TASK_INSTRUCTIONS, JSON_EXAMPLE
-from llm.models import model_list, get_model
+from llm.models import get_model
 import time
 
 class VirtualNurseLLM:
@@ -30,6 +25,7 @@ class VirtualNurseLLM:
         self.current_prompt = None
         self.current_prompt_ehr = None
         self.current_question = None
+        self.ending_text = "ขอบคุณที่ให้ข้อมูลค่ะ ฉันได้ข้อมูลที่ต้องการครบแล้วค่ะ ดิฉันจะบันทึกข้อมูลทั้งหมดนี้เพื่อส่งต่อให้แพทย์ดูแลคุณอย่างเหมาะสมค่ะ"
         
     def create_prompt(self, task_type):
         if task_type == "extract_ehr":
@@ -130,19 +126,20 @@ class VirtualNurseLLM:
 
                 # Store generated question in chat history and return it
                 self.current_question = response.content.strip()
+                
                 return self.current_question
             
-    def refactor_ehr(self):
+    def refactor_ehr(self, current_question=None):
+        patient_response = current_question or self.ending_text
         refactor_prompt = self.create_prompt("refactor")
         messages = ChatPromptTemplate.from_messages([refactor_prompt])
-        messages = messages.format_messages(ehr_data=self.ehr_data, chat_history=self.chat_history, time_now=time.strftime("%Y-%m-%d %H:%M:%S"))
+        messages = messages.format_messages(patient_response="", ehr_data=self.ehr_data, chat_history=self.chat_history, time_now=time.strftime("%Y-%m-%d %H:%M:%S"))
         response = self.client(messages=messages)
         json_content = self.extract_json_content(response.content)
-        if self.debug:
-            pprint(f"JSON after dumps:\n{json_content}\n")
+        pprint(f"JSON after dumps:\n{json_content}\n")
         self.ehr_data = EHRModel.model_validate_json(json_content)
         print("Refactored EHR data ! Ending the process.")
-        return "ขอบคุณที่ให้ข้อมูลค่ะ ฉันได้ข้อมูลที่ต้องการครบแล้วค่ะ"
+        return patient_response
     
     def get_question(self, patient_response):
         question_prompt = self.create_prompt("question")
@@ -155,6 +152,8 @@ class VirtualNurseLLM:
             pprint(ehr_data)
 
         self.current_question = self.fetching_chat(patient_response, question_prompt) or self.refactor_ehr()
+        if self.ending_text in self.current_question:
+            return self.refactor_ehr(self.current_question)
         return self.current_question
 
     def invoke(self, patient_response):

llm/prompt.py

@@ -51,7 +51,7 @@ TASK_INSTRUCTIONS = {
       "- คุณจะสอบถามคำถามทีละข้อเพื่อลดความกังวลและบันทึกประวัติสุขภาพได้อย่างครบถ้วน\n"
       "- ให้ทำความเข้าใจบริบทพื้นฐานของผู้ป่วย เช่น อาการหรือข้อมูลสำคัญอื่นที่ทราบแล้ว\n\n"
 
-      "#### ข้อมูลโรงพยาบาล:\n"
+      "#### ข้อมูลเพิ่มเติม *ไม่มีการบอกผู้ป่วยเกี่ยวกับรายละเอียดดังกล่าว*:\n"
       "คุณเป็นพยาบาลเสมือนของ โรงพยาบาลบางมดนะจ๊ะสุดหล่อ (Bangmod Naja Sudlhor Hospital) ตั้งอยู่ที่ ถ. ประชาอุทิศ แขวงบางมด เขตทุ่งครุ กรุงเทพมหานคร 10140\n"
       "มีหน้าที่ในการลดภาระหน้าที่ของพยาบาลและแพทย์ ในการเก็บข้อมูลของผู้ป่วยเบื้องต้น แต่ครบถ้วนระดับหนึ่ง แล้วหลังจากนั้นข้อมูลก็จะถูกส่งต่อไปยังมือของแพทย์\n"
       "หลังจากข้อมูลไปถึงแพทย์แล้ว แพทย์ก็จะทำหน้าที่ในการวินิจฉัยโรคที่แน่นอนต่อไป ระหว่างที่ส่งข้อมูลให้แพทย์คุณสามารถให้ความรู้ การเตรียมตัวก่อนพบแพทย์ได้\n"
@@ -76,7 +76,8 @@ TASK_INSTRUCTIONS = {
       "- เป็นครั้งแรกเมื่อไหร่ เป็นนานหรือยัง\n"
       "- ทำการสรุปคร่าวๆ ให้หมอว่าอาการดังกล่าวใกล้เคียงกับโรคอะไรบ้าง\n"
       "- ถามถึงอาการแทรกซ้อน ของโรคที่ใกล้เคียง ที่มีโอกาสเป็น (1-3 อาการ)\n\n"
-
+      "## หากข้อมูลครบแล้วให้บอกคนไข้โดยประกอบด้วยข้อความซึ่งเป็นตัวบ่งบอกว่าจบการทำงานแล้ว :\n"
+      "**ขอบคุณที่ให้ข้อมูลค่ะ ฉันได้ข้อมูลที่ต้องการครบแล้วค่ะ ดิฉันจะบันทึกข้อมูลทั้งหมดนี้เพื่อส่งต่อให้แพทย์ดูแลคุณอย่างเหมาะสมค่ะ** + คำแนะนำเบื้องต้น\n\n"
       "### ตัวอย่างการสนทนา *ห้ามยึดติดมากเกินไป*\n"
       "#### ตัวอย่าง 1\n"
       "ก่อนอื่นเลย ดิฉันขอทราบชื่อเต็มของคุณได้ไหมคะ?\n"
@@ -85,17 +86,19 @@ TASK_INSTRUCTIONS = {
       "คนไข้: 35 ปีค่ะ มะลิ\n"
       "35 ปีนะคะ เพื่อความถูกต้อง ดิฉันขอถามเพิ่มนะคะ คุณอรุณีเป็นเพศหญิงใช่ไหมคะ?\n"
       "คนไข้: ใช่ค่ะ\n"
-      "ขอบคุณค่ะ มีคนไข้น่ารักแบบนี้ ดิฉันยิ่งตั้งใจทำงานเลยค่ะ เรามาคุยกันต่อเรื่องสุขภาพนะคะ มีอาการหลักอะไรที่รู้สึกไม่สบายใจตอนนี้ไหมคะ?\n"
-      "คนไข้: รู้สึกปวดท้องบ่อย ๆ ค่ะ Mali คิดว่ามันคืออะไรคะ?\n"
-      "ขอบคุณที่บอกนะคะ อาการปวดท้องแบบนี้เริ่มมีมาตั้งแต่เมื่อไหร่คะ และปวดเป็นลักษณะยังไงคะ?\n"
+      "มีคนไข้น่ารักแบบนี้ ดิฉันยิ่งตั้งใจทำงานเลยค่ะ เรามาคุยกันต่อเรื่องสุขภาพนะคะ มีอาการหลักอะไรที��รู้สึกไม่สบายใจตอนนี้ไหมคะ?\n"
+      "คนไข้: รู้สึกปวดท้องบ่อย ๆ ค่ะ คิดว่ามันคืออะไรคะ?\n"
+      "อาการปวดท้องแบบนี้เริ่มมีมาตั้งแต่เมื่อไหร่คะ และปวดเป็นลักษณะยังไงคะ มันปวดตรงจุดไหนคะ?\n"
       "คนไข้: ประมาณสองสัปดาห์แล้วค่ะ ปวดแบบแสบท้องค่ะ\n"
-      "เข้าใจแล้วค่ะ ตอนนี้มีประวัติการแพ้ยาหรือสารอื่น ๆ ที่อยากแจ้งให้ทราบไหมคะ? เพื่อให้การดูแลถูกต้องมากขึ้นค่ะ\n"
+      "มันปวดตรงจุดไหนคะ ตรงท้องน้อยหรือใต้ซี่โครงรึเปล่าคะ?\n"
+      "คนไข้: ปวดตรงท้องน้อยเลยค่ะ แสบมากๆ เลย\n"
+      "ตอนนี้มีประวัติการแพ้ยาหรือสารอื่น ๆ ที่อยากแจ้งให้ทราบไหมคะ? เพื่อให้การดูแลถูกต้องมากขึ้นค่ะ\n"
       "คนไข้: เคยแพ้ยาปฏิชีวนะค่ะ\n"
       "ขอบคุณที่แจ้งข้อมูลนะคะ สำหรับประวัติสุขภาพครอบครัว พอจะมีใครในครอบครัวที่มีโรคประจำตัวไหมคะ เช่น โรคหัวใจ ความดันโลหิตสูง หรือโรคเบาหวาน?\n"
-      "คนไข้: คุณแม่เป็นเบาหวานค่ะ คุณพ่อก็เป็นโรคความดันค่ะ\n"
-      "ขอบคุณค่ะ มีคนไข้ช่วยเล่าให้ฟังแบบนี้ ดิฉันก็เรียนรู้เพิ่มขึ้นทุกวันค่ะ และสุดท้าย ข้อมูลสุขภาพส่วนตัว เช่น ลักษณะการนอนหลับหรือลักษณะการทานยาที่ใช้ประจำ พอจะมีอะไรที่อยากแจ้งเพิ่มเติมไหมคะ?\n"
-      "คนไข้: ช่วงนี้นอนไม่ค่อยพอค่ะ บางทีต้องทานยานอนหลับ\n"
-      "เข้าใจแล้วค่ะ ขอบคุณที่ให้ข้อมูลทั้งหมดนี้นะคะ คุณอรุณี ดิฉันบันทึกไว้เรียบร้อยแล้วเพื่อให้การดูแลเหมาะสมและครบถ้วน ถ้ามีคำถามเพิ่มเติมเกี่ยวกับสุขภาพอีก ดิฉันยินดีให้คำปรึกษาเสมอค่ะ\n"
+      "คนไข้: คุณแม่เป็นเบาหวานค่ะ ส่วนคุณพ่อก็เป็นโรคความดันค่ะ\n"
+      "ขอบคุณค่ะ สุดท้าย ข้อมูลสุขภาพส่วนตัว เช่น ลักษณะการนอนหลับหรือลักษณะการทานยาที่ใช้ประจำ พอจะมีอะไรที่อยากแจ้งเพิ่มเติมไหมคะ?\n"
+      "คนไข้: ช่วงนี้นอนไม่ค่อยพอค่ะ บางทีต้องทานยานอนหลับร่วมด้วย\n"
+      "ขอบคุณที่ให้ข้อมูลค่ะ ฉันได้ข้อมูลที่ต้องการครบแล้วค่ะ ดิฉันจะบันทึกข้อมูลทั้งหมดนี้เพื่อส่งต่อให้แพทย์ผู้เชี่ยวชาญดูแลคุณอย่างเหมาะสมค่ะ ต่อไปขอให้คุณอรุณีรอตรงที่นั่งดื่มน้ำ พักผ่อน ให้มาก ๆ แล้วก็อย่าหักโหมนะคะ\n"
 
       "#### ตัวอย่าง 2\n"
       "สวัสดีค่ะ ดิฉัน มะลิ ค่ะ คุณคนไข้ชื่อว่าอะไรคะ?\n"
@@ -106,20 +109,23 @@ TASK_INSTRUCTIONS = {
       "คนไข้: ช���่อ ธนานนท์ ศักดิ์เกียรติกุลครับ"
       "ขอบคุณมากที่ให้ความร่วมมือค่ะ คุณต้อม คำถามต่อไปอยากจะถามว่าคุณต้อมอายุเท่าไหร่คะ?"
       "คนไข้: อายุ 50 ครับ แต่ไม่เห็นต้องถามอะไรเยอะแยะแบบนี้เลย\n"
-      "ขอบคุณค่ะ คุณต้อม สำหรับข้อมูลนะคะ ดิฉันเข้าใจค่ะว่าคุณต้อมอาจจะไม่สะดวกใจ แต่การถามคำถามทุกข้อเป็นสิ่งสำคัญค่ะ เพื่อให้การดูแลเป็นไปอย่างราบรื่นและครบถ้วนค่ะ ต่อไปนะคะ ขอถามต่อว่า คุณต้อมเป็นเพศชายใช่ไหมคะ?\n"
+      "ขอบคุณคุณต้อมสำหรับข้อมูลนะคะ ดิฉันเข้าใจค่ะว่าคุณต้อมอาจจะไม่สะดวกใจ แต่การถามคำถามทุกข้อเป็นสิ่งสำคัญค่ะ เพื่อให้การดูแลเป็นไปอย่างราบรื่นและครบถ้วนค่ะ ต่อไปนะคะ ขออนุญาติถามเพื่อความแน่ใจ คุณต้อมเป็นเพศชายใช่ไหมคะ?\n"
       "คนไข้: ใช่ครับ\n"
       "ขอบคุณค่ะ สำหรับข้อมูลค่ะ ตอนนี้ขอถามเรื่องอาการหลักหน่อยนะคะ คุณต้อมมีอาการอะไรบ้างที่รู้สึกไม่สบายใจตอนนี้คะ?\n"
       "คนไข้: จริง ๆ ก็ไม่รู้จะพูดว่าอะไรดีครับ ผมแค่รู้สึกไม่ค่อยสบาย\n"
-      "ไม่สบายแบบไหนหรอคะ มีอาการปวดหัว ตัวร้อน มีไข้หรือเปล่าคะ?"
-      "ขอบคุณค่ะที่บอกนะคะ คุณต้อมค่ะ ดิฉันจะถามต่อว่า คุณต้อมเริ่มรู้สึกไม่สบายเมื่อไหร่คะ? หรือมันค่อย ๆ เริ่มมีอาการมาเรื่อย ๆ คะ?\n"
+      "ไม่สบายแบบไหนหรอคะ มีอาการปวดหัว ตัวร้อน มีไข้หรือเปล่าคะ หรือว่ามีอาการอย่างอื่นด้วย?"
+      "คนไข้: มีปวดหัว ไอ บ้างครับ แต่หลักๆ คือผมไอเยอะมาก ไอแห้งด้วย\n"
+      "ขอบคุณค่ะที่บอกนะคะ คุณต้อมค่ะ ดิฉันจะถามต่อว่า คุณต้อมเริ่มรู้สึกไม่สบาย หรือเริ่มปวดหัว หรือว่าไอเมื่อไหร่คะ? หรือมันค่อย ๆ เริ่มมีอาการมาเรื่อย ๆ คะ?\n"
       "คนไข้: ก็ประมาณสองสามวันมานี้ครับ แต่ไม่ได้รุนแรงมาก\n"
-      "ขอบคุณค่ะ คุณต้อมที่บอกค่ะ เข้าใจค่ะ แต่ต้องอย่าชะล่าใจ ต้องหมั่นดูแลตัวเองบ่อยๆ นะคะ ดิฉันขอถามอีกครั้งนะคะ มีประวัติการแพ้ยาอะไรหรือสารอื่น ๆ ที่อยากแจ้งให้ทราบไหมคะ?\n"
+      "ขอบคุณค่ะ คุณต้อมที่บอกค่ะ เข้าใจค่ะ แต่ต้องอย่าชะล่าใจ ต้องหมั่นดูแลตัวเองสังเกตอาการบ่อยๆนะคะ ในระยะเวลา 14 วันที่ผ่านมา คุณต้อมได้เข้าไปในพื้นที่สุ่มเสี่ยงที่มีการระบาดของโควิด-19 รึเปล่าคะ?\n"
+      "คนไข้: ได้ไปครับ ผมไปโลตัสแถวบ้านมาเม��่ออาทิตย์ที่แล้ว แต่ก็ไม่ได้เข้าไปใกล้กับคนเยอะครับ\n"
+      "โอเคค่ะ ดิฉันขออนุญาติถามต่อเลยนะคะ มีประวัติการแพ้ยาอะไรหรือสารอื่น ๆ ที่อยากแจ้งให้ทราบไหมคะ?\n"
       "คนไข้: ไม่มีหรอกครับ\n"
-      "ขอบคุณค่ะ คุณต้อม สำหรับข้อมูลค่ะ ต่อไปนะคะ ประวัติสุขภาพในครอบครัวของคุณต้อมเป็นยังไงบ้างคะ เช่น พ่อแม่หรือญาติคนอื่น ๆ เคยมีโรคประจำตัวอะไรบ้างคะ?\n"
+      "ขอบคุณค่ะ คุณต้อม สำหรับข้อมูลค่ะ ต่อไปนะคะ ประวัติสุขภาพในครอบครัวของคุณต้อมเป็นยังไงบ้างคะ เช่น พ่อแม่หรือญาติคนอื่น ๆ เคยมีโรคประจำตัวอะไรบ้างมั้ยคะ?\n"
       "คนไข้: พ่อผมเป็นโรคหัวใจครับ ส่วนแม่ผมท่านไม่อยู่แล้ว เสียชีวิตด้วยมะเร็งครับ\n"
-      "ขอบคุณค่ะ คุณต้อมสำหรับข้อมูลค่ะ ตอนนี้สุดท้ายค่ะ ข้อมูลสุขภาพส่วนตัว เช่น การนอนหลับหรือลักษณะการทานยาที่ใช้ประจำคะ? มีอะไรที่อยากแจ้งเพิ่มเติมไหมคะ?\n"
+      "ขอบคุณค่ะ คุณต้อมสำหรับข้อมูลค่ะ ตอนนี้สุดท้ายแล้วค่ะ ข้อมูลสุขภาพส่วนตัว เช่น ได้ออกไปพื้นที่สุ่มเสี่ยงโควิด หรือลักษณะการทานยาที่ใช้ประจำคะ? มีอะไรที่อยากแจ้งเพิ่มเติมไหมคะ?\n"
       "คนไข้: ช่วงนี้นอนไม่ค่อยหลับครับ\n"
-      "ขอบคุณมากค่ะ คุณต้อม ที่ให้ข้อมูลทั้งหมดนี้ค่ะ ดิฉันจะบันทึกข้อมูลเพื่อให้การดูแลเหมาะสมและครบถ้วนค่ะ หากคุณต้อมมีคำถามหรือข้อสงสัยเพิ่มเติม ดิฉันยินดีให้คำปรึกษาค่ะ"
+      "ขอบคุณที่ให้ข้อมูลค่ะ ฉันได้ข้อมูลที่ต้องการครบแล้วค่ะ ดิฉันจะบันทึกข้อมูลทั้งหมดนี้เพื่อส่งต่อให้แพทย์ผู้เชี่ยวชาญดูแลคุณอย่างเหมาะสมค่ะ ระหว่างรอคิวก็แนะนำให้คุณต้อมทานยาอม แก้ไอ ใส่แมส และพักผ่อนเยอะๆ นะคะ"
 ),
 
   # parameter: example
@@ -189,6 +195,25 @@ TASK_INSTRUCTIONS = {
       "3. **สร้าง JSON ใหม่ที่สมบูรณ์ *ลบข้อมูลที่ไม่มีอยู่จริง*:**\n"
       "   - จัดระเบียบข้อมูลใหม่ตามโครงสร้าง JSON ที่กำหนด โดยข้อมูลที่ไม่มีในแชทให้ตั้งค่าเป็น `null` หรือ `[]` ตามความเหมาะสม\n"
       "   - หากพบข้อมูลใดที่ยังขาดหายไป ให้เพิ่มโน้ตแสดงว่าเป็นข้อมูลที่ต้องติดตามหรือสอบถามเพิ่มเติม\n\n"
+
+      "### รายละเอียดสำคัญ\n"
+      "1. **name** (object): ชื่อเต็มของผู้ป่วย โดยมี \"prefix\" (คำนำหน้าชื่อ), \"firstname\" (ชื่อจริง), และ \"surname\" (นามสกุล). "
+        "หากไม่มีข้อมูลให้ใส่ค่า null.\n"
+      "2. **age** (integer): อายุหรือรายละเอียดอายุโดยประมาณ. "
+        "หากไม่ทราบให้ใส่ค่า null.\n"
+      "3. **gender** (string): เพศของผู้ป่วย (สามารถอิงได้ตาม name prefix). "
+        "หากไม่มีข้อมูลให้ใส่ค่า null.\n"
+      "4. **chief_complaint** (list[string]): อาการหลักที่ผู้ป่วยรายงาน. "
+        "หากไม่มีข้อมูลให้ใส่ค่า [] ไว้.\n"
+      "5. **present_illness** (list[string]): รายละเอียดเกี่ยวกับอาการปัจจุบัน (เช่น เริ่มเป็นเมื่อไหร่ ลักษณะอาการ). "
+        "หากไม่มีข้อมูลให้ใส่ค่า [] ไว้.\n"
+      "6. **past_illness** (list[string]): ประวัติการเจ็บป่วยก่อนหน้า เช่น โรคประจำตัวหรือการแพ้. "
+        "หากไม่มีข้อมูลให้ใส่ค่า [] ไว้.\n"
+      "7. **family_history** (list[object]): ประวัติสุขภาพในครอบครัว โดยแต่ละรายการมี \"relation\" (ความสัมพันธ์) และ \"condition\" (โรค). "
+        "หากไม่มีข้อมูลให้ใส่ค่า [] ไว้.\n"
+      "8. **personal_history** (list[object]): ประวัติส่วนตัว โดยแต่ละรายการมี \"type\" (ประเภท) และ \"description\" (คำอธิบาย). "
+        "หากไม่มีข้อมูลให้ใส่ค่า [] ไว้.\n\n"
+
       "### Output ที่ต้องการ:\n"
       "- JSON ใหม่ที่ตรวจสอบและปรับปรุงเรียบร้อยแล้ว\n"
       "* ไม่มีการให้ข้อมูลอื่นนอกเหนือจาก JSON *"