DisasterGPT API

curl -N https://sw-kim.com/api/chat/stream \
  -H "Content-Type: application/json" \
  -H "Accept: text/event-stream" \
  -H "X-API-KEY: YOUR_API_KEY" \
  -d '{
    "question": "호우 특보 시 지하공간 대피 절차는?",
    "history": [],
    "selected_pdf": "all",
    "conversation_id": "browser-session-uuid"
  }'

X-API-KEY: YOUR_API_KEY

data: {
  "type": "refs",
  "refs": [{ "docId": 0, "name": "풍수해 매뉴얼", "page": 37 }],
  "tables": [{
    "table_id": "TABLE_p038_0",
    "page": 38,
    "title": "재난대응 단계",
    "grade": "A",
    "rows": [["구분", "상황"], ["비상 1단계", "호우주의보 발표"]],
    "markdown": "| 구분 | 상황 |\n|---|---|\n| 비상 1단계 | 호우주의보 발표 |"
  }]
}

function renderMarkdown(target, markdown) {
  const source = (markdown ?? '').replace(/^[\u200B-\u200F\uFEFF]/, '');
  const unsafeHtml = marked.parse(source, { gfm: true, breaks: true });
  target.innerHTML = DOMPurify.sanitize(unsafeHtml, {
    USE_PROFILES: { html: true }
  });
  target.classList.add('markdown-body');
}

// chunk.content는 누적값: append가 아니라 replace
if (event.type === 'chunk') answer = event.content;
if (streamFinished) renderMarkdown(answerElement, answer);

function renderEvidenceTable(table) {
  const figure = document.createElement('figure');
  const caption = document.createElement('figcaption');
  caption.textContent = `${table.title || '근거 표'} · p.${table.page}`;

  const grid = document.createElement('table');
  const head = grid.createTHead();
  const body = grid.createTBody();
  table.rows.forEach((row, rowIndex) => {
    const tr = document.createElement('tr');
    row.forEach(value => {
      const cell = document.createElement(rowIndex === 0 ? 'th' : 'td');
      cell.textContent = String(value ?? '');
      tr.appendChild(cell);
    });
    (rowIndex === 0 ? head : body).appendChild(tr);
  });

  figure.append(caption, grid);
  return figure;
}

function stripAttachedTableMarkdown(answer, tables) {
  if (!tables.length) return answer;
  const marker = '\n\n── 근거: ';
  const markerIndex = answer.indexOf(marker);
  return markerIndex >= 0 ? answer.slice(0, markerIndex) : answer;
}

if (event.type === 'refs') {
  refs = event.refs ?? [];
  tables = event.tables ?? [];
}

<button id="prev" type="button">이전</button>
<span id="pageInfo">— / —</span>
<button id="next" type="button">다음</button>
<img id="pdfPage" alt="PDF 근거 페이지">

let docId = null, page = 0, totalPages = 0;

async function loadPage(nextDocId, nextPage) {
  const res = await fetch(`/api/documents/${nextDocId}/page/${nextPage}`, {
    headers: { 'X-API-KEY': API_KEY }
  });
  if (!res.ok) throw new Error(`PDF page: ${res.status}`);
  const data = await res.json();

  docId = nextDocId;
  page = data.page;
  totalPages = data.total_pages;
  pdfPage.src = `data:image/png;base64,${data.image}`;
  pageInfo.textContent = `${page + 1} / ${totalPages}`;
  prev.disabled = page === 0;
  next.disabled = page >= totalPages - 1;
}

prev.onclick = () => loadPage(docId, page - 1);
next.onclick = () => loadPage(docId, page + 1);
function jumpToRef(ref) { return loadPage(ref.docId, ref.page); }

function makeReferenceButton(ref) {
  const button = document.createElement('button');
  button.type = 'button';
  button.textContent = `${ref.name} · p.${ref.page + 1}`;
  button.onclick = () => jumpToRef(ref);
  return button;
}

async function streamChat(question) {
  const response = await fetch('https://sw-kim.com/api/chat/stream', {
    method: 'POST',
    headers: {
      'Content-Type': 'application/json',
      'Accept': 'text/event-stream',
      'X-API-KEY': API_KEY
    },
    body: JSON.stringify({ question, history: [], selected_pdf: 'all' })
  });
  if (!response.ok) throw new Error(`Chat API: ${response.status}`);

  const reader = response.body.getReader();
  const decoder = new TextDecoder();
  let buffer = '', answer = '', refs = [], tables = [];

  while (true) {
    const { value, done } = await reader.read();
    buffer += decoder.decode(value || new Uint8Array(), { stream: !done });
    const lines = buffer.split('\n');
    buffer = lines.pop() || '';

    for (const line of lines) {
      if (!line.startsWith('data:')) continue;
      const raw = line.slice(5).trim();
      if (!raw || raw === '[DONE]') continue;
      const event = JSON.parse(raw);
      if (event.type === 'chunk') answer = event.content; // 누적값 교체
      if (event.type === 'refs') {
        refs = event.refs || [];
        tables = event.tables || [];
      }
      if (event.type === 'error') throw new Error(event.content);
    }
    if (done) break;
  }

  // 구조화 표를 그릴 때 답변 끝의 동일한 Markdown 표는 제거한다.
  renderMarkdown(answerElement, stripAttachedTableMarkdown(answer, tables));
  tableArea.replaceChildren();
  referenceArea.replaceChildren();
  tables.forEach(table => tableArea.append(renderEvidenceTable(table)));
  refs.forEach(ref => referenceArea.append(makeReferenceButton(ref)));
}