"""
BreadBuddy — AI 面包诊所 (v5)
单 Tab 诊断流程，原型稿对齐 UI，追问交互
"""
import os
import sys
import base64
import warnings
import gradio as gr
# Gradio 6.x 弃用警告（theme/css 已移到 launch()）
warnings.filterwarnings("ignore", category=UserWarning, module="gradio")
# ─── 将 deploy/ 目录加入 sys.path 以导入模块 ───
_deploy_dir = os.path.dirname(os.path.abspath(__file__))
if _deploy_dir not in sys.path:
sys.path.insert(0, _deploy_dir)
from config import API_URL, MODEL_NAME, API_KEY, DEFAULT_TIMEOUT
print(f"[APP STARTUP] API_URL={API_URL} MODEL={MODEL_NAME} TIMEOUT={DEFAULT_TIMEOUT}")
from llm import stream_llm, call_llm, _estimate_tokens
from parsers import parse_diagnosis_result, render_structured_result, render_empty_hint
from history import get_history, append_to_history, clear_history
from i18n import t, get_system_prompt, get_followup_suffix, get_user_fallback, set_lang, LANG
# ─── 诊断 System Prompt ───
DIAGNOSIS_SYSTEM_PROMPT = get_system_prompt()
FOLLOWUP_SYSTEM_SUFFIX = get_followup_suffix()
# ─── Handlers ───
def diagnose_bread(photo, description, temperature, humidity):
"""诊断面包问题 — 主 handler"""
if not description and not photo:
yield render_empty_hint(), "", "⏱️ --"
return
# 构建消息（新诊断不载入历史，追问才载入）
messages = [{"role": "system", "content": DIAGNOSIS_SYSTEM_PROMPT}]
user_text_parts = []
if description:
user_text_parts.append(f"Problem: {description}")
if temperature:
user_text_parts.append(f"Temperature: {temperature}°C")
if humidity:
user_text_parts.append(f"Humidity: {humidity}%")
user_text = "\n".join(user_text_parts) or "Please diagnose this bread photo"
if photo:
import os as _os
_photo_size = _os.path.getsize(photo) if _os.path.exists(photo) else -1
print(f"[DIAG DEBUG] photo type={type(photo)} value={repr(photo)}")
print(f"[DIAG DEBUG] file exists={_os.path.exists(photo)} size={_photo_size}")
with open(photo, "rb") as f:
img_bytes = f.read()
img_b64 = base64.b64encode(img_bytes).decode()
print(f"[DIAG] photo path={photo} size={_photo_size} b64_len={len(img_b64)}")
# Check image dimensions
try:
from PIL import Image
import io
pil_img = Image.open(io.BytesIO(img_bytes))
print(f"[DIAG DEBUG] image dimensions={pil_img.size} format={pil_img.format} mode={pil_img.mode}")
except Exception as e:
print(f"[DIAG DEBUG] image parse error: {e}")
messages.append({"role": "user", "content": [
{"type": "text", "text": user_text},
{"type": "image_url", "image_url": {"url": f"data:image/png;base64,{img_b64}"}}
]})
print(f"[DIAG] messages[-1] has image_url type + text, b64_preview={img_b64[:50]}...")
else:
messages.append({"role": "user", "content": user_text})
yield 'Connecting to AI baker...
First diagnosis may take 30-60s to warm up
🍞
')
gr.HTML('Snap a photo, let me check your bread
')
photo_in = gr.Image(
type="filepath", sources=["upload", "webcam"],
label="📷 Tap to take a photo or upload",
height=200,
)
desc_in = gr.Textbox(
label="", show_label=False,
placeholder="Describe your bread problem... (optional)",
lines=1, max_lines=3,
elem_id="descInput",
elem_classes=["desc-input"],
)
# 温湿度预设
gr.HTML("""
""")
temp_in = gr.Slider(0, 50, value=20, label="Temperature (°C)", elem_id="temp-slider")
humid_in = gr.Slider(0, 100, value=50, label="Humidity (%)", elem_id="humid-slider")
btn_diagnose = gr.Button("Start Diagnosis", variant="primary", elem_classes=["btn-diagnose"], elem_id="btnDiagnose")
# 结果区
with gr.Column(elem_id="bb-result-wrap"):
speed_out = gr.Markdown("⏱️ --", elem_classes=["speed-display"])
result_out = gr.HTML(render_empty_hint())
with gr.Accordion("🧠 Model Thinking (optional)", open=False):
reason_out = gr.Textbox(label="", lines=6, show_label=False)
# 操作按钮行
with gr.Row():
btn_rediagnose = gr.Button("← Restart", elem_classes=["btn-rediagnose"])
# 追问区（底部固定）
with gr.Column(elem_id="bb-followup-row"):
gr.HTML('🧠 Follow-up context:
')
gr.HTML('''
''')
with gr.Row():
followup_in = gr.Textbox(
label="", show_label=False,
placeholder="Ask a follow-up... e.g. Can you explain the first cause?",
lines=1, scale=5,
elem_id="followupInput",
)
btn_followup = gr.Button("➤", variant="primary", scale=1, elem_id="btnSend", elem_classes=["btn-send"])
# ─── 事件绑定 ───
btn_diagnose.click(
diagnose_bread,
[photo_in, desc_in, temp_in, humid_in],
[result_out, reason_out, speed_out],
)
btn_followup.click(
followup_diagnosis,
[followup_in, result_out],
[result_out, reason_out, speed_out],
).then(
lambda: "", # 清空追问输入框
outputs=[followup_in],
)
# Enter 键发送追问
followup_in.submit(
followup_diagnosis,
[followup_in, result_out],
[result_out, reason_out, speed_out],
).then(
lambda: "",
outputs=[followup_in],
)
# 重新诊断
btn_rediagnose.click(
do_rediagnose,
outputs=[result_out, reason_out, speed_out],
)
# Footer
gr.HTML("""
""")
# ─── JS 注入: 通过 demo.load() 在页面加载时执行 ───
_bb_js = """(function() {
var saved = localStorage.getItem('bb-theme');
var initTheme = saved || (window.matchMedia('(prefers-color-scheme: dark)').matches ? 'dark' : 'light');
document.body.classList.toggle('dark', initTheme === 'dark');
document.body.setAttribute('data-theme', initTheme);
document.addEventListener('click', function(e) {
var langSpan = e.target.closest('.bb-lang');
if (langSpan) {
var next = langSpan.textContent === 'EN' ? '中文' : 'EN';
langSpan.textContent = next;
location.reload();
return;
}
var toggleBtn = e.target.closest('.bb-theme-toggle, #themeToggle, .theme-toggle');
if (toggleBtn) {
document.body.classList.add('bb-theme-transitioning');
var isDark = document.body.classList.toggle('dark');
var newTheme = isDark ? 'dark' : 'light';
document.body.setAttribute('data-theme', newTheme);
localStorage.setItem('bb-theme', newTheme);
setTimeout(function() {
document.body.classList.remove('bb-theme-transitioning');
}, 350);
return;
}
var presetBtn = e.target.closest('.bb-preset-btn');
if (presetBtn) {
var presets = document.getElementById('bb-presets');
if (!presets) return;
if (presetBtn.classList.contains('active')) {
presetBtn.classList.remove('active');
var sr = document.getElementById('bb-slider-row');
if (sr) sr.classList.remove('visible');
return;
}
presets.querySelectorAll('.bb-preset-btn').forEach(function(b) { b.classList.remove('active'); });
presetBtn.classList.add('active');
var preset = presetBtn.dataset.preset;
var sr2 = document.getElementById('bb-slider-row');
if (preset === 'custom') {
if (sr2) sr2.classList.add('visible');
} else {
if (sr2) sr2.classList.remove('visible');
var t = parseInt(presetBtn.dataset.temp);
var h = parseInt(presetBtn.dataset.humid);
var ts = document.querySelector('#temp-slider input[type=range]');
var hs = document.querySelector('#humid-slider input[type=range]');
if (ts) { ts.value = t; ts.dispatchEvent(new Event('input', {bubbles:true})); }
if (hs) { hs.value = h; hs.dispatchEvent(new Event('input', {bubbles:true})); }
}
return;
}
/* ── Preset Questions ── */
var presetQ = e.target.closest('.bb-preset-q');
if (presetQ) {
var input = document.querySelector('#followupInput textarea, #followupInput input');
if (input) {
var nativeSetter = Object.getOwnPropertyDescriptor(
window.HTMLTextAreaElement.prototype, 'value'
).set;
nativeSetter.call(input, presetQ.dataset.q);
input.dispatchEvent(new Event('input', { bubbles: true }));
input.focus();
}
return;
}
var recipeHdr = e.target.closest('.recipe-header, #recipeHeader');
if (recipeHdr) {
var acc = document.getElementById('recipeAccordion');
if (!acc) return;
acc.classList.toggle('open');
var span = recipeHdr.querySelector('span:first-child');
if (span) {
var count = span.textContent.match(/\\d+/);
var n = count ? count[0] : '';
span.textContent = acc.classList.contains('open')
? '\U0001f35e 推荐 ' + n + ' 个配方 · 点击收起'
: '\U0001f35e 推荐 ' + n + ' 个配方 · 点击展开';
}
return;
}
});
function bbInitFollowup() {
var resultWrap = document.getElementById('bb-result-wrap');
var banner = document.getElementById('followupBanner');
if (!resultWrap || !banner || banner._bbBound) return;
banner._bbBound = true;
var observer = new MutationObserver(function() {
if (banner.classList.contains('visible')) return;
var card = resultWrap.querySelector('.result-card');
if (card && card.querySelectorAll('.cause-chip').length > 0) {
banner.classList.add('visible');
}
});
observer.observe(resultWrap, { childList: true, subtree: true });
}
var _obs = new MutationObserver(function() { bbInitFollowup(); });
_obs.observe(document.body, { childList: true, subtree: true });
bbInitFollowup();
/* ═══ Virtual keyboard fix (mobile) ═══ */
if (window.visualViewport) {
var followupWrap = document.querySelector('.bb-followup-wrap');
var rafId = null;
var adjustViewport = function() {
if (rafId) return;
rafId = requestAnimationFrame(function() {
rafId = null;
if (followupWrap) {
var diff = window.innerHeight - window.visualViewport.height;
followupWrap.style.bottom = Math.max(diff, 0) + 'px';
}
});
};
window.visualViewport.addEventListener('resize', adjustViewport);
window.visualViewport.addEventListener('scroll', adjustViewport);
if (followupWrap) {
followupWrap.addEventListener('focusout', function() {
setTimeout(function() {
followupWrap.style.bottom = '0px';
}, 350);
});
}
}
return null;
})()"""
demo.load(fn=None, inputs=None, outputs=None, js=_bb_js)
if __name__ == "__main__":
print(f"[STARTUP] BreadBuddy starting on 0.0.0.0:7860")
print(f"[STARTUP] API_URL={API_URL}")
print(f"[STARTUP] Language: {LANG}")
demo.launch(
server_name="0.0.0.0",
server_port=7860,
share=False
)