Update app.py

app.py

@@ -12,9 +12,8 @@ from openai import OpenAI
 from reportlab.lib import colors
 from reportlab.lib.pagesizes import A4
 from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph, Spacer
-from reportlab.lib.styles import getSampleStyleSheet
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.pdfbase import pdfmetrics
-from reportlab.pdfbase.ttfonts import TTFont
 from reportlab.pdfbase.cidfonts import UnicodeCIDFont
 
 # --------------------
@@ -26,7 +25,6 @@ FALLBACK_TEXT = """
 ### 1. 可能相關疾病
 - 胃食道逆流（GERD）：胃酸回流到食道，可能引起燒灼感和胃痛。
 - 胃潰瘍：胃內部的黏膜受損，形成潰瘍，可能導致疼痛和不適。
-- 胃炎：胃黏膜的炎症，通常伴隨疼痛、噁心和消化不良。
 ### 2. 飲食營養建議
 - 纖維素：有助於消化，減少便秘和腸道不適。
 - 維生素C：幫助修復胃黏膜，增強免疫系統。
@@ -43,7 +41,7 @@ FALLBACK_TEXT = """
 """.strip()
 
 # --------------------
-# GPT 呼叫
+# GPT 呼叫（加速版）
 # --------------------
 def call_gpt_system(prompt: str) -> str:
     if client is None:
@@ -51,8 +49,9 @@ def call_gpt_system(prompt: str) -> str:
     try:
         resp = client.chat.completions.create(
             model="gpt-4o-mini",
+            max_tokens=500,         # 限制字數以加快速度
+            temperature=0.6,        # 適度隨機
             messages=[{"role": "user", "content": prompt}],
-            temperature=0.7,
         )
         text = resp.choices[0].message.content
         return text if text and text.strip() else FALLBACK_TEXT
@@ -61,7 +60,7 @@ def call_gpt_system(prompt: str) -> str:
         return FALLBACK_TEXT
 
 # --------------------
-# 解析函式
+# 區塊解析
 # --------------------
 def extract_block(text: str, starts: list, ends: list = None) -> str:
     if not text:
@@ -159,7 +158,7 @@ def robust_parse(text: str) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
     return df_disease, df_nutrition, df_meals
 
 # --------------------
-# PDF 生成
+# PDF 生成（修正版：文字換行）
 # --------------------
 def has_chinese(s: str) -> bool:
     return bool(re.search(r"[\u4e00-\u9fff]", str(s)))
@@ -175,12 +174,8 @@ def generate_pdf_file(df_disease, df_nutrition, df_meals, basic_info=None):
     df_nutrition = clean_dataframe(df_nutrition)
     df_meals = clean_dataframe(df_meals)
 
-    font_name, font_ok = None, False
-    try:
-        pdfmetrics.registerFont(UnicodeCIDFont("STSong-Light"))
-        font_name = "STSong-Light"; font_ok = True
-    except:
-        pass
+    pdfmetrics.registerFont(UnicodeCIDFont("STSong-Light"))
+    font_name = "STSong-Light"
 
     file_prefix = basic_info.get("name", "個人") if basic_info else "個人"
     safe_name = re.sub(r"[^\w\u4e00-\u9fff]", "_", file_prefix)
@@ -192,9 +187,9 @@ def generate_pdf_file(df_disease, df_nutrition, df_meals, basic_info=None):
     styles = getSampleStyleSheet()
     title_style = styles["Title"].clone("title")
     heading_style = styles["Heading2"].clone("h2")
-    normal_style = styles["Normal"].clone("normal")
-    if font_ok and font_name:
-        title_style.fontName = heading_style.fontName = normal_style.fontName = font_name
+    normal_style = ParagraphStyle("normal", fontName=font_name, fontSize=10, leading=14)
+
+    title_style.fontName = heading_style.fontName = font_name
 
     # 標題
     elements.append(Paragraph(f"健康分析與飲食建議報告 — {file_prefix}", title_style))
@@ -214,10 +209,8 @@ def generate_pdf_file(df_disease, df_nutrition, df_meals, basic_info=None):
         style = TableStyle([
             ("GRID", (0, 0), (-1, -1), 0.8, colors.black),
             ("BACKGROUND", (0, 0), (-1, 0), colors.lightgrey),
-            ("VALIGN", (0, 0), (-1, -1), "TOP"),
+            ("FONTNAME", (0, 0), (-1, -1), font_name),
         ])
-        if font_ok:
-            style.add("FONTNAME", (0, 0), (-1, -1), font_name)
         t.setStyle(style)
         elements.append(t)
         elements.append(Spacer(1, 16))
@@ -225,45 +218,27 @@ def generate_pdf_file(df_disease, df_nutrition, df_meals, basic_info=None):
     # 疾病表
     if not df_disease.empty:
         elements.append(Paragraph("一、可能相關疾病", heading_style))
-        table_data = [df_disease.columns.tolist()] + df_disease.values.tolist()
+        table_data = [df_disease.columns.tolist()] + [[Paragraph(str(c), normal_style) for c in row] for row in df_disease.values.tolist()]
         t = Table(table_data, colWidths=[150, 330])
-        style = TableStyle([
-            ("GRID", (0, 0), (-1, -1), 0.8, colors.black),
-            ("BACKGROUND", (0, 0), (-1, 0), colors.lightgrey),
-        ])
-        if font_ok:
-            style.add("FONTNAME", (0, 0), (-1, -1), font_name)
-        t.setStyle(style)
+        t.setStyle(TableStyle([("GRID", (0, 0), (-1, -1), 0.8, colors.black), ("BACKGROUND", (0, 0), (-1, 0), colors.lightgrey), ("FONTNAME", (0, 0), (-1, -1), font_name)]))
         elements.append(t)
         elements.append(Spacer(1, 12))
 
     # 營養建議表
     if not df_nutrition.empty:
         elements.append(Paragraph("二、飲食營養建議", heading_style))
-        table_data = [df_nutrition.columns.tolist()] + df_nutrition.values.tolist()
+        table_data = [df_nutrition.columns.tolist()] + [[Paragraph(str(c), normal_style) for c in row] for row in df_nutrition.values.tolist()]
         t = Table(table_data, colWidths=[150, 330])
-        style = TableStyle([
-            ("GRID", (0, 0), (-1, -1), 0.8, colors.black),
-            ("BACKGROUND", (0, 0), (-1, 0), colors.lightgrey),
-        ])
-        if font_ok:
-            style.add("FONTNAME", (0, 0), (-1, -1), font_name)
-        t.setStyle(style)
+        t.setStyle(TableStyle([("GRID", (0, 0), (-1, -1), 0.8, colors.black), ("BACKGROUND", (0, 0), (-1, 0), colors.lightgrey), ("FONTNAME", (0, 0), (-1, -1), font_name)]))
         elements.append(t)
         elements.append(Spacer(1, 12))
 
-    # 餐點表
+    # 餐點表（換行修正）
     if not df_meals.empty:
         elements.append(Paragraph("三、一日三餐（中式 / 西式）", heading_style))
-        table_data = [df_meals.columns.tolist()] + df_meals.values.tolist()
+        table_data = [df_meals.columns.tolist()] + [[Paragraph(str(c), normal_style) for c in row] for row in df_meals.values.tolist()]
         t = Table(table_data, colWidths=[240, 240])
-        style = TableStyle([
-            ("GRID", (0, 0), (-1, -1), 0.8, colors.black),
-            ("BACKGROUND", (0, 0), (-1, 0), colors.lightgrey),
-        ])
-        if font_ok:
-            style.add("FONTNAME", (0, 0), (-1, -1), font_name)
-        t.setStyle(style)
+        t.setStyle(TableStyle([("GRID", (0, 0), (-1, -1), 0.8, colors.black), ("BACKGROUND", (0, 0), (-1, 0), colors.lightgrey), ("FONTNAME", (0, 0), (-1, -1), font_name)]))
         elements.append(t)
 
     doc.build(elements)
@@ -300,7 +275,7 @@ css_code = """
 """
 
 with gr.Blocks(css=css_code) as demo:
-    gr.Markdown("## 🥗 AI 營養師 — 客製化飲食建議")
+    gr.Markdown("## 🩺AI食療專家")
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### 請輸入基本資料")