Update app.py

app.py

@@ -119,36 +119,50 @@ def create_medical_report(pt_name, pt_id, diagnosis, conf):
         print(f"PDF Error: {e}")
         return None
 
-# --- Chat Function (Added Missing Function) ---
+# --- Chat Function ---
 def chat_fn(message, history, crop_img, info_text, diagnosis):
-    if history is None: history = []
+    # ถ้า history เป็น None ให้เริ่มเป็น List ว่าง
+    if history is None: 
+        history = []
     
+    # 1. ใส่ข้อความฝั่ง User เข้าไปก่อน
+    history.append({"role": "user", "content": message})
+    
+    # เช็ค API Key
     if not GOOGLE_API_KEY:
-        response = "ระบบ Chat ไม่สามารถใช้งานได้เนื่องจากไม่มี API KEY (กรุณาตั้งค่า Secret 'GOOGLE_API_KEY')"
-        history.append((message, response))
+        error_msg = "❌ ไม่พบ API KEY: กรุณาไปที่ Settings > Secrets แล้วตั้งค่า 'GOOGLE_API_KEY'"
+        history.append({"role": "assistant", "content": error_msg})
         return history
 
     try:
-        # Construct Prompt with Context
+        # เตรียม Prompt
+        diag_str = diagnosis if diagnosis else "ยังไม่มีการวินิจฉัย"
+        info_str = info_text if info_text else "ไม่มีรายละเอียด"
+        
         context_prompt = f"""
-        คุณคือผู้ช่วยทางการแพทย์อัจฉริยะ (AI Medical Assistant) จากรามาธิบดี
-        ข้อมูลผลการตรวจปัจจุบันของผู้ป่วย:
-        - การวินิจฉัยหลัก: {diagnosis}
-        - รายละเอียดทางเทคนิค: {info_text}
+        บทบาท: คุณคือผู้ช่วยทางการแพทย์อัจฉริยะ (AI Medical Assistant)
+        
+        ข้อมูลผู้ป่วยปัจจุบัน:
+        - ผลการวินิจฉัย: {diag_str}
+        - รายละเอียด: {info_str}
         
-        คำถามจากผู้ใช้: {message}
+        คำถาม: {message}
         
-        กรุณาตอบคำถามอย่างสุภาพ ให้ข้อมูลที่เป็นประโยชน์ แต่เน้นย้ำว่าผลการตรวจต้องยืนยันโดยแพทย์ผู้เชี่ยวชาญเสมอ
-        ตอบเป็นภาษาไทย
+        ตอบเป็นภาษาไทย กระชับ สุภาพ และต้องลงท้ายว่า "ผลการวินิจฉัยต้องยืนยันโดยแพทย์ผู้เชี่ยวชาญ"
         """
         
-        model = genai.GenerativeModel('gemini-pro')
+        # เรียก Gemini
+        model = genai.GenerativeModel('gemini-1.5-flash')
         response = model.generate_content(context_prompt)
         bot_reply = response.text
+
     except Exception as e:
-        bot_reply = f"เกิดข้อผิดพลาดในการเชื่อมต่อ AI: {str(e)}"
+        bot_reply = f"เกิดข้อผิดพลาด: {str(e)}"
+        print(f"Error: {e}")
 
-    history.append((message, bot_reply))
+    # 2. ใส่ข้อความฝั่ง AI (Assistant) เข้าไป
+    history.append({"role": "assistant", "content": bot_reply})
+    
     return history
 
 # ============================================
@@ -341,7 +355,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, title="Ovarian Tumor AI") as dem
     with gr.Column(elem_id="floating_container"):
         with gr.Column(elem_id="chat_window"):
             gr.HTML(f"<div style='background:linear-gradient(90deg, #0072ff, #00c6ff); color:white; padding:15px; border-radius:15px 15px 0 0;'><b>💬 ปรึกษาน้องดูแล</b></div>")
-            chatbot = gr.Chatbot(height=400, show_label=False, avatar_images=(None, LOGO_RAMA_URL))
+            chatbot = gr.Chatbot(height=400, show_label=False, avatar_images=(None, LOGO_RAMA_URL), type="messages")
             msg = gr.Textbox(placeholder="พิมพ์คำถามที่นี่...", show_label=False)
             btn_send = gr.Button("ส่งข้อความ", variant="primary")