Update app.py

app.py

@@ -24,8 +24,8 @@ from reportlab.pdfbase.ttfonts import TTFont
 # 1) Configuration & Setup
 # ============================================
 
-# 🔑 API KEY
-GOOGLE_API_KEY = os.getenv("AIzaSyBYksOq03N5V2MjSYicHdsk4ESdyR9FABw") 
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+
 if GOOGLE_API_KEY:
     genai.configure(api_key=GOOGLE_API_KEY)
 
@@ -48,17 +48,23 @@ def force_download_font(url, filename):
     if not os.path.exists(filename):
         print(f"📥 Downloading {filename}...")
         try:
-            r = requests.get(url, allow_redirects=True)
-            with open(filename, 'wb') as f:
-                f.write(r.content)
+            session = requests.Session()
+            session.headers.update({'User-Agent': 'Mozilla/5.0'})
+            r = session.get(url, allow_redirects=True)
+            if r.status_code == 200 and len(r.content) > 1000:
+                with open(filename, 'wb') as f:
+                    f.write(r.content)
+            else:
+                print(f"❌ Failed to download {filename} (Status: {r.status_code})")
+                return False
         except Exception as e:
             print(f"❌ Error downloading {filename}: {e}")
             return False
     return True
 
 font_urls = [
-    ("https://github.com/nutjunkie/thaifonts_sipa/raw/master/sipa_fonts/THSarabunNew/THSarabunNew.ttf", "THSarabunNew.ttf"),
-    ("https://github.com/nutjunkie/thaifonts_sipa/raw/master/sipa_fonts/THSarabunNew/THSarabunNew%20Bold.ttf", "THSarabunNew-Bold.ttf")
+    ("https://raw.githubusercontent.com/nutjunkie/thaifonts_sipa/master/sipa_fonts/THSarabunNew/THSarabunNew.ttf", "THSarabunNew.ttf"),
+    ("https://raw.githubusercontent.com/nutjunkie/thaifonts_sipa/master/sipa_fonts/THSarabunNew/THSarabunNew%20Bold.ttf", "THSarabunNew-Bold.ttf")
 ]
 
 for url, fname in font_urls:
@@ -98,8 +104,6 @@ def create_medical_report(pt_name, pt_id, diagnosis, conf):
     try:
         filename = tempfile.mktemp(suffix=".pdf")
         c = canvas.Canvas(filename, pagesize=A4)
-        
-        # Use registered font if available, else standard
         font_name = 'THSarabun-Bold' if 'THSarabun-Bold' in pdfmetrics.getRegisteredFontNames() else 'Helvetica-Bold'
         
         c.setFont(font_name, 24)
@@ -119,22 +123,20 @@ def create_medical_report(pt_name, pt_id, diagnosis, conf):
         return None
 
 # ============================================
-# [FIXED] Chat Function 
+# [FIXED] Chat Function (Dictionary Format)
 # ============================================
 def chat_fn(message, history, crop_img, info_text, diagnosis):
-    # เปลี่ยนจาก history = [] เป็นการรับค่า list ของ dict
     if history is None: history = []
     
-    # 1. เช็ค API Key
+    # 1. API Key Check
     if not GOOGLE_API_KEY:
-        response = "❌ ไม่พบ API KEY: กรุณาไปที่ Settings > Secrets แล้วตั้งค่า 'GOOGLE_API_KEY' จากนั้นกด Restart Space"
-        # เพิ่มข้อความลง history แบบ Dictionary (ตามมาตรฐานใหม่ Huggingface/Gradio)
+        response = "❌ ไม่พบ API KEY: กรุณาตรวจสอบการตั้งค่า GOOGLE_API_KEY ในไฟล์ app.py"
         history.append({"role": "user", "content": message})
         history.append({"role": "assistant", "content": response})
         return history, ""
 
     try:
-        # 2. สร้าง Prompt
+        # 2. Context Prompt
         context_prompt = f"""
         บทบาท: คุณคือผู้ช่วยทางการแพทย์อัจฉริยะ (AI Medical Assistant)
         
@@ -150,8 +152,8 @@ def chat_fn(message, history, crop_img, info_text, diagnosis):
         - *สำคัญ*: ต้องลงท้ายเสมอว่า "ควรปรึกษาแพทย์ผู้เชี่ยวชาญเพื่อการวินิจฉัยที่แม่นยำที่สุด"
         """
         
-        # 3. เรียกใช้ Gemini
-        model = genai.GenerativeModel('gemini-1.5-flash')
+        # 3. Call Gemini
+        model = genai.GenerativeModel('gemini-2.5-flash')
         response = model.generate_content(context_prompt)
         bot_reply = response.text
 
@@ -159,11 +161,10 @@ def chat_fn(message, history, crop_img, info_text, diagnosis):
         bot_reply = f"เกิดข้อผิดพลาด (System Error): {str(e)}"
         print(f"DEBUG ERROR: {e}")
 
-    # [FIX] เพิ่มข้อความลงใน History แบบ Dictionary
+    # Append Dictionary format
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": bot_reply})
     
-    # คืนค่า history และ string ว่าง ("") เพื่อลบข้อความในช่องพิมพ์
     return history, ""
 
 # ============================================
@@ -344,13 +345,13 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, title="Ovarian Tumor AI") as dem
         with gr.Tab("3. Gallery History"):
             gallery_ui = gr.Gallery(columns=4, height=600)
 
-    # --- [FIXED] Floating Chatbot ---
+    # --- Floating Chatbot ---
     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>")
             
-            # [FIX] เพิ่ม type="messages" เพื่อบอก Gradio ว่าใช้ format ใหม่
-            chatbot = gr.Chatbot(height=400, show_label=False, avatar_images=(None, LOGO_RAMA_URL), type="messages")
+            # Chatbot: No type param needed here, but data passed will be dicts
+            chatbot = gr.Chatbot(height=400, show_label=False, avatar_images=(None, LOGO_RAMA_URL))
             
             msg = gr.Textbox(placeholder="พิมพ์คำถามที่นี่...", show_label=False)
             btn_send = gr.Button("ส่งข้อความ", variant="primary")
@@ -374,7 +375,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, title="Ovarian Tumor AI") as dem
 
     btn_pdf.click(pdf_wrapper, [inp_pt_name, inp_pt_id, state_diag, state_conf], out_pdf)
 
-    # [FIX] Chat interaction: เพิ่ม outputs ตัวที่ 2 (msg) เพื่อล้างข้อความ
+    # Chat interactions
     btn_send.click(chat_fn, [msg, chatbot, state_crop, state_info, state_diag], [chatbot, msg])
     msg.submit(chat_fn, [msg, chatbot, state_crop, state_info, state_diag], [chatbot, msg])