mirror of
https://github.com/hoshikawa2/oci_dashboard_select_ai.git
synced 2026-03-03 16:09:37 +00:00
first commit
This commit is contained in:
7
config
Normal file
7
config
Normal file
@@ -0,0 +1,7 @@
|
||||
{
|
||||
"oci_profile": "DEFAULT",
|
||||
"WALLET_PATH": "./Wallet_ORADB23ai",
|
||||
"DB_ALIAS": "oradb23ai_high",
|
||||
"USERNAME": "admin",
|
||||
"PASSWORD": "Moniquinha1972"
|
||||
}
|
||||
899
select_ai_pro_v2.py
Normal file
899
select_ai_pro_v2.py
Normal file
@@ -0,0 +1,899 @@
|
||||
# app_select_ai.py
|
||||
# ---------------------------------------------------------
|
||||
# Chat + Dashboard (histórico) para SELECT AI no Autonomous Database
|
||||
# ---------------------------------------------------------
|
||||
|
||||
from flask import Flask, request, render_template_string, session
|
||||
import oracledb, os, json
|
||||
from flask import send_file, jsonify
|
||||
from io import BytesIO
|
||||
import base64
|
||||
from openpyxl import Workbook
|
||||
from openpyxl.utils import get_column_letter
|
||||
from openpyxl.drawing.image import Image as XLImage
|
||||
from PIL import Image as PILImage
|
||||
|
||||
# ======================
|
||||
# CONFIGURAÇÕES DO BANCO
|
||||
# ======================
|
||||
with open("./config", "r") as f:
|
||||
config_data = json.load(f)
|
||||
|
||||
WALLET_PATH = config_data["WALLET_PATH"]
|
||||
DB_ALIAS = config_data["DB_ALIAS"]
|
||||
USERNAME = config_data["USERNAME"]
|
||||
PASSWORD = config_data["PASSWORD"]
|
||||
os.environ["TNS_ADMIN"] = WALLET_PATH
|
||||
|
||||
PROFILE_NAME = "OCI_GENERATIVE_AI_PROFILE"
|
||||
|
||||
def set_select_ai_profile(conn, requested_tag):
|
||||
# Ativa o profile em cada sessão do pool
|
||||
with conn.cursor() as cur:
|
||||
cur.execute("BEGIN DBMS_CLOUD_AI.SET_PROFILE(:p); END;", p=PROFILE_NAME)
|
||||
|
||||
pool = oracledb.create_pool(
|
||||
user=USERNAME,
|
||||
password=PASSWORD,
|
||||
dsn=DB_ALIAS,
|
||||
config_dir=WALLET_PATH,
|
||||
wallet_location=WALLET_PATH,
|
||||
wallet_password=PASSWORD,
|
||||
min=1, max=5, increment=1,
|
||||
session_callback=set_select_ai_profile
|
||||
)
|
||||
|
||||
# ======================
|
||||
# APP FLASK
|
||||
# ======================
|
||||
app = Flask(__name__)
|
||||
app.secret_key = "troque-esta-chave" # necessário p/ sessão (histórico)
|
||||
|
||||
PAGE = """
|
||||
<!doctype html>
|
||||
<html lang="pt-br">
|
||||
<head>
|
||||
<meta charset="utf-8">
|
||||
<title>ER Analytics · Select AI</title>
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||||
<script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
|
||||
<script>
|
||||
Chart.defaults.font.size = 11;
|
||||
Chart.defaults.plugins.legend.labels.boxWidth = 10;
|
||||
</script>
|
||||
<!-- Export PDF -->
|
||||
<script src="https://cdn.jsdelivr.net/npm/jspdf@2.5.1/dist/jspdf.umd.min.js"></script>
|
||||
<script src="https://cdn.jsdelivr.net/npm/html2canvas@1.4.1/dist/html2canvas.min.js"></script>
|
||||
<!-- Export Excel (browser) -->
|
||||
<script src="https://cdn.jsdelivr.net/npm/xlsx-populate/browser/xlsx-populate.min.js"></script>
|
||||
<style>
|
||||
:root{
|
||||
--bg:#ffffff; /* fundo página */
|
||||
--card:#f9fafb; /* cartões/itens */
|
||||
--border:#e5e7eb; /* linhas/bordas */
|
||||
--muted:#6b7280; /* textos fracos */
|
||||
--text:#111827; /* texto principal */
|
||||
--brand:#0d6efd; /* botões/realces */
|
||||
}
|
||||
|
||||
*{box-sizing:border-box}
|
||||
body{
|
||||
margin:0;
|
||||
font-family:system-ui,-apple-system,Segoe UI,Roboto,Arial;
|
||||
background:var(--bg);
|
||||
color:var(--text);
|
||||
}
|
||||
|
||||
/* Layout */
|
||||
.page{ min-height:100vh; display:flex; flex-direction:column; }
|
||||
.wrap{ max-width:1200px; width:100%; margin:0 auto; padding:16px; flex:1; display:flex; flex-direction:column; gap:12px; }
|
||||
.header{ display:flex; align-items:center; gap:10px; }
|
||||
.logo{ width:34px; height:34px; border-radius:8px; background:linear-gradient(135deg,#60a5fa,#22d3ee); display:flex; align-items:center; justify-content:center; font-weight:800; color:#fff; }
|
||||
.title{ font-size:18px; font-weight:700 }
|
||||
.muted{ color:var(--muted); font-size:12px; }
|
||||
|
||||
.card{ background:var(--card); border:1px solid var(--border); border-radius:12px; padding:14px; }
|
||||
.kpi{ display:flex; gap:8px; flex-wrap:wrap; }
|
||||
.pill{ padding:4px 8px; border:1px solid var(--border); border-radius:999px; font-size:12px; background:#fff; color:var(--text); }
|
||||
|
||||
/* Histórico */
|
||||
#history{ flex:1; overflow:auto; display:flex; flex-direction:column; gap:10px; padding-right:4px; max-height:68vh; }
|
||||
.item{ background:var(--card); border:1px solid var(--border); border-radius:10px; padding:12px; }
|
||||
.sql{
|
||||
font-family: ui-monospace, SFMono-Regular, Menlo, Consolas, "Liberation Mono", monospace;
|
||||
font-size:12px; color:#1f2937; background:#ffffff; border:1px solid var(--border);
|
||||
border-radius:8px; padding:8px; overflow:auto;
|
||||
}
|
||||
.block{ margin-top:8px; }
|
||||
|
||||
table{ width:100%; border-collapse:collapse; font-size:14px; background:#fff; }
|
||||
th,td{ padding:8px; border-bottom:1px solid var(--border); text-align:left; }
|
||||
th{ background:#f3f4f6; position:sticky; top:0; }
|
||||
|
||||
/* Rodapé/input */
|
||||
.footer{ position:sticky; bottom:0; background:var(--bg); padding:12px 16px; border-top:1px solid var(--border); }
|
||||
.ask{ display:flex; gap:8px; flex-wrap:wrap; }
|
||||
input[type="text"]{
|
||||
flex:1; min-width:280px; padding:10px 12px; border-radius:10px;
|
||||
border:1px solid var(--border); background:#ffffff; color:var(--text);
|
||||
}
|
||||
button{
|
||||
background:var(--brand); color:#fff; border:none; border-radius:10px;
|
||||
padding:10px 14px; font-weight:600; cursor:pointer;
|
||||
}
|
||||
button:hover{ filter:brightness(1.05) }
|
||||
|
||||
/* Gráficos */
|
||||
.toolbar{ display:flex; gap:6px; flex-wrap:wrap; margin:6px 0 6px; }
|
||||
.pill{ background:#ffffff; border:1px solid var(--border); color:#374151; }
|
||||
.pill:hover{ border-color:#cbd5e1; }
|
||||
|
||||
.chart-box{
|
||||
height:220px; width:100%;
|
||||
border:1px solid var(--border); border-radius:10px;
|
||||
background:#ffffff; padding:8px;
|
||||
}
|
||||
@media (max-width:640px){ .chart-box{ height:180px; } }
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="page">
|
||||
<div class="wrap">
|
||||
<div class="header">
|
||||
<div class="logo">ER</div>
|
||||
<div>
|
||||
<div class="title">ER Analytics · Select AI</div>
|
||||
<div class="muted">Respostas e gráficos no topo; pergunta no rodapé. A última resposta fica visível automaticamente.</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{% if timeline %}
|
||||
<div class="card">
|
||||
<div class="kpi">
|
||||
<span class="pill">Conexão: <b>{{ db_alias }}</b></span>
|
||||
<span class="pill">Usuário: <b>{{ session_user }}</b></span>
|
||||
<span class="pill">Profile AI: <b>{{ profile or '—' }}</b></span>
|
||||
|
||||
<!-- Botões de export -->
|
||||
<span class="pill" style="cursor:auto; border:none;">|</span>
|
||||
<button type="button" class="pill" onclick="exportPDF()" title="Exporta todas as respostas (com gráficos e tabelas) para PDF">Exportar PDF</button>
|
||||
<button type="button" class="pill" onclick="exportExcelServer()">Exportar Excel</button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div id="history">
|
||||
{% for item in timeline %}
|
||||
<div class="item">
|
||||
<div><b>Pergunta:</b> {{ item.prompt }}</div>
|
||||
{% if item.sql %}<div class="block"><div class="sql">{{ item.sql }}</div></div>{% endif %}
|
||||
|
||||
{% if item.chart %}
|
||||
<div class="toolbar">
|
||||
<span class="pill" onclick="render_{{ loop.index0 }}('bar')">Barras</span>
|
||||
<span class="pill" onclick="render_{{ loop.index0 }}('line')">Linhas</span>
|
||||
<span class="pill" onclick="render_{{ loop.index0 }}('pie')">Pizza</span>
|
||||
{% if item.table %}
|
||||
<span class="pill" onclick="toggleTable_{{ loop.index0 }}()">Mostrar/ocultar Tabela</span>
|
||||
{% endif %}
|
||||
</div>
|
||||
|
||||
<div class="block chart-box">
|
||||
<canvas id="chart_{{ loop.index0 }}"></canvas>
|
||||
</div>
|
||||
|
||||
<script>
|
||||
(function(){
|
||||
const DATA_{{ loop.index0 }} = {
|
||||
type: '{{ item.chart["type"] }}',
|
||||
labels: {{ item.chart["labels"]|tojson }},
|
||||
values: {{ item.chart["values"]|tojson }},
|
||||
seriesLabel: '{{ item.chart["seriesLabel"] }}',
|
||||
title: '{{ item.chart["title"] }}',
|
||||
xLabel: '{{ item.chart["xLabel"] }}',
|
||||
yLabel: '{{ item.chart["yLabel"] }}'
|
||||
};
|
||||
|
||||
let ch_{{ loop.index0 }} = null;
|
||||
|
||||
function makeChart_{{ loop.index0 }}(kind){
|
||||
const ctx = document.getElementById('chart_{{ loop.index0 }}').getContext('2d');
|
||||
if (ch_{{ loop.index0 }}) ch_{{ loop.index0 }}.destroy();
|
||||
ch_{{ loop.index0 }} = new Chart(ctx, {
|
||||
type: kind,
|
||||
data: {
|
||||
labels: DATA_{{ loop.index0 }}.labels,
|
||||
datasets: [{
|
||||
label: DATA_{{ loop.index0 }}.seriesLabel,
|
||||
data: DATA_{{ loop.index0 }}.values,
|
||||
borderWidth: 1,
|
||||
pointRadius: 2
|
||||
}]
|
||||
},
|
||||
options: {
|
||||
responsive: true,
|
||||
maintainAspectRatio: false,
|
||||
layout: { padding: 4 },
|
||||
plugins: {
|
||||
legend: { display: true, labels: { boxWidth: 10, font: { size: 10 } } },
|
||||
title: { display: true, text: DATA_{{ loop.index0 }}.title, font: { size: 12 } },
|
||||
tooltip:{ bodyFont: { size: 11 }, titleFont: { size: 11 } }
|
||||
},
|
||||
scales: {
|
||||
x: { title: { display: true, text: DATA_{{ loop.index0 }}.xLabel, font: { size: 11 } },
|
||||
ticks: { font: { size: 10 }, maxRotation: 0, autoSkip: true } },
|
||||
y: { title: { display: true, text: DATA_{{ loop.index0 }}.yLabel, font: { size: 11 } },
|
||||
ticks: { font: { size: 10 } }, beginAtZero: true, grace: "5%" }
|
||||
},
|
||||
elements: { line: { tension: 0.2 } }
|
||||
}
|
||||
});
|
||||
// garante que a última resposta fica visível após render
|
||||
setTimeout(scrollHistoryBottom, 60);
|
||||
}
|
||||
|
||||
// expõe funções por-card
|
||||
window.render_{{ loop.index0 }} = makeChart_{{ loop.index0 }};
|
||||
{% if item.table %}
|
||||
window.toggleTable_{{ loop.index0 }} = function(){
|
||||
const el = document.getElementById('table_{{ loop.index0 }}');
|
||||
if (!el) return;
|
||||
el.style.display = (el.style.display === 'none') ? '' : 'none';
|
||||
setTimeout(scrollHistoryBottom, 40);
|
||||
};
|
||||
{% endif %}
|
||||
|
||||
// inicial: usa o tipo sugerido pelo backend
|
||||
makeChart_{{ loop.index0 }}(DATA_{{ loop.index0 }}.type);
|
||||
})();
|
||||
</script>
|
||||
{% endif %}
|
||||
|
||||
{% if item.table %}
|
||||
<div id="table_{{ loop.index0 }}" class="block" style="border:1px solid var(--border); border-radius:10px; overflow:auto; max-height:40vh;">
|
||||
<table>
|
||||
<thead><tr>{% for h in item.table.headers %}<th>{{ h }}</th>{% endfor %}</tr></thead>
|
||||
<tbody>
|
||||
{% for r in item.table.rows %}
|
||||
<tr>{% for c in r %}<td>{{ c }}</td>{% endfor %}</tr>
|
||||
{% endfor %}
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
{% endif %}
|
||||
</div>
|
||||
{% endfor %}
|
||||
<!-- ⬇️ marcador fixo de fim -->
|
||||
<div id="history-bottom"></div>
|
||||
</div>
|
||||
{% endif %}
|
||||
</div>
|
||||
|
||||
<!-- Rodapé com input SEMPRE embaixo -->
|
||||
<div class="footer">
|
||||
<form class="ask" method="post" onsubmit="markScrollToBottom()">
|
||||
<input type="text" name="frase" placeholder="Faça sua pergunta… (Enter para enviar)" required>
|
||||
<input type="text" name="tabela" value="{{ default_table }}" style="display:none;">
|
||||
<button type="submit">Enviar</button>
|
||||
</form>
|
||||
<div class="muted" style="margin-top:6px">Dica: cite colunas (ex.: <code>admitted</code>, <code>month_num</code>, <code>year_num</code>) para respostas mais precisas.</div>
|
||||
</div>
|
||||
<script>
|
||||
document.addEventListener("DOMContentLoaded", () => {
|
||||
const frase = document.querySelector('input[name="frase"]');
|
||||
frase.addEventListener("keydown", (e) => {
|
||||
if(e.key === "Enter"){
|
||||
e.preventDefault();
|
||||
frase.form.requestSubmit(); // envia o form
|
||||
}
|
||||
});
|
||||
});
|
||||
</script>
|
||||
</div>
|
||||
|
||||
<script>
|
||||
function scrollHistoryBottom(smooth=true){
|
||||
const end = document.getElementById('history-bottom');
|
||||
if (end) end.scrollIntoView({behavior: smooth ? 'smooth' : 'auto', block: 'end'});
|
||||
}
|
||||
|
||||
// Marca a intenção de rolar após o POST (reload da página)
|
||||
function markScrollToBottom(){
|
||||
try { localStorage.setItem('__scrollToBottom', '1'); } catch(e){}
|
||||
}
|
||||
|
||||
// Rola quando a página carrega (e re-rola depois que gráficos expandirem)
|
||||
function settleAndScroll(){
|
||||
scrollHistoryBottom(false); // imediato
|
||||
setTimeout(scrollHistoryBottom, 80); // após layout inicial
|
||||
setTimeout(scrollHistoryBottom, 250);// após Chart.js desenhar
|
||||
setTimeout(scrollHistoryBottom, 600);// salvaguarda final
|
||||
}
|
||||
|
||||
// Executa no load
|
||||
window.addEventListener('load', () => {
|
||||
// se veio de um submit, respeite a marca
|
||||
let must = false;
|
||||
try { must = localStorage.getItem('__scrollToBottom') === '1'; } catch(e){}
|
||||
if (must) {
|
||||
try { localStorage.removeItem('__scrollToBottom'); } catch(e){}
|
||||
settleAndScroll();
|
||||
} else {
|
||||
// mesmo sem submit, mantenha a última resposta à vista
|
||||
settleAndScroll();
|
||||
}
|
||||
});
|
||||
</script>
|
||||
<script>
|
||||
function _grabTableData(itemEl){
|
||||
const table = itemEl.querySelector('table');
|
||||
if(!table) return null;
|
||||
const headers = Array.from(table.querySelectorAll('thead th')).map(th => th.textContent.trim());
|
||||
const rows = Array.from(table.querySelectorAll('tbody tr')).map(tr =>
|
||||
Array.from(tr.querySelectorAll('td')).map(td => td.textContent)
|
||||
);
|
||||
return { headers, rows };
|
||||
}
|
||||
|
||||
function _grabQuestion(itemEl){
|
||||
const b = itemEl.querySelector('div>b');
|
||||
if(b && b.parentElement){
|
||||
return b.parentElement.textContent.replace(/^Pergunta:\s*/,'').trim();
|
||||
}
|
||||
return "Pergunta";
|
||||
}
|
||||
|
||||
function _grabSQL(itemEl){
|
||||
const sqlEl = itemEl.querySelector('.sql');
|
||||
return sqlEl ? sqlEl.textContent : "";
|
||||
}
|
||||
|
||||
async function exportExcelServer(){
|
||||
const items = Array.from(document.querySelectorAll('#history .item'));
|
||||
if(items.length === 0){
|
||||
alert('Não há respostas para exportar.');
|
||||
return;
|
||||
}
|
||||
|
||||
// monta payload
|
||||
const payload = { items: [] };
|
||||
for(const item of items){
|
||||
const question = _grabQuestion(item);
|
||||
const sql = _grabSQL(item);
|
||||
const table = _grabTableData(item);
|
||||
const canvas = item.querySelector('canvas');
|
||||
|
||||
let chartPng = null;
|
||||
if(canvas){
|
||||
try{
|
||||
chartPng = canvas.toDataURL('image/png'); // inclui prefixo data:
|
||||
}catch(e){
|
||||
chartPng = null;
|
||||
}
|
||||
}
|
||||
|
||||
payload.items.push({
|
||||
question, sql, table, chartPng
|
||||
});
|
||||
}
|
||||
|
||||
// envia para o servidor
|
||||
const res = await fetch("/export/xlsx", {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify(payload)
|
||||
});
|
||||
|
||||
if(!res.ok){
|
||||
const t = await res.text();
|
||||
console.error('Falha no export:', t);
|
||||
alert('Falha ao gerar Excel no servidor.');
|
||||
return;
|
||||
}
|
||||
|
||||
// baixa o arquivo
|
||||
const blob = await res.blob();
|
||||
const url = URL.createObjectURL(blob);
|
||||
const a = document.createElement("a");
|
||||
a.href = url;
|
||||
a.download = "respostas-select-ai.xlsx";
|
||||
document.body.appendChild(a);
|
||||
a.click();
|
||||
a.remove();
|
||||
setTimeout(()=>URL.revokeObjectURL(url), 1000);
|
||||
}
|
||||
// Utilitário para baixar Blob
|
||||
function downloadBlob(blob, filename){
|
||||
const url = URL.createObjectURL(blob);
|
||||
const a = document.createElement('a');
|
||||
a.href = url; a.download = filename; a.click();
|
||||
setTimeout(()=>URL.revokeObjectURL(url), 1000);
|
||||
}
|
||||
|
||||
// ============ Exportar PDF (com charts + tabelas como na tela) ============
|
||||
async function exportPDF(){
|
||||
const { jsPDF } = window.jspdf;
|
||||
const pdf = new jsPDF('p','mm','a4');
|
||||
const pageWidth = pdf.internal.pageSize.getWidth();
|
||||
const pageHeight = pdf.internal.pageSize.getHeight();
|
||||
const margin = 10;
|
||||
const maxW = pageWidth - margin*2;
|
||||
|
||||
const items = Array.from(document.querySelectorAll('#history .item'));
|
||||
if(items.length === 0){ alert('Não há respostas para exportar.'); return; }
|
||||
|
||||
for(let i=0;i<items.length;i++){
|
||||
const el = items[i];
|
||||
// usa html2canvas para “fotografar” o card, incluindo canvas e tabelas
|
||||
const canvas = await html2canvas(el, {backgroundColor: '#0f1115', scale: 2, useCORS: true});
|
||||
const imgData = canvas.toDataURL('image/png');
|
||||
// dimensiona para caber na página
|
||||
const imgW = maxW;
|
||||
const imgH = canvas.height * (imgW / canvas.width);
|
||||
|
||||
// se altura maior que a página, escala para caber verticalmente
|
||||
const effW = imgW;
|
||||
const effH = Math.min(imgH, pageHeight - margin*2);
|
||||
const scale = Math.min(imgW / canvas.width, (pageHeight - margin*2) / canvas.height);
|
||||
const drawW = canvas.width * scale;
|
||||
const drawH = canvas.height * scale;
|
||||
|
||||
if(i>0) pdf.addPage();
|
||||
pdf.addImage(imgData, 'PNG', margin, margin, drawW, drawH);
|
||||
}
|
||||
|
||||
pdf.save('respostas-select-ai.pdf');
|
||||
}
|
||||
|
||||
// ============ Exportar Excel (uma planilha por resposta; gráfico como imagem) ============
|
||||
// Helper para baixar Blob
|
||||
function downloadBlob(blob, filename){
|
||||
const url = URL.createObjectURL(blob);
|
||||
const a = document.createElement('a');
|
||||
a.href = url; a.download = filename; a.click();
|
||||
setTimeout(()=>URL.revokeObjectURL(url), 1000);
|
||||
}
|
||||
|
||||
// Helper: insere imagem tentando múltiplas assinaturas do XlsxPopulate
|
||||
async function insertImage(wb, sheet, base64Png, topLeftCell, width, height){
|
||||
// 1) Tentativa com 'base64' + 'anchor'
|
||||
try{
|
||||
await sheet.addImage({
|
||||
base64: base64Png, // sem prefixo data:
|
||||
name: 'chart',
|
||||
anchor: topLeftCell, // ex.: 'A12'
|
||||
width, height
|
||||
});
|
||||
return true;
|
||||
}catch(e1){
|
||||
// 2) Tentativa com 'image' (dataURL completo) + 'topLeftCell'
|
||||
try{
|
||||
const dataUrl = 'data:image/png;base64,' + base64Png;
|
||||
await sheet.addImage({
|
||||
image: dataUrl,
|
||||
topLeftCell: topLeftCell,
|
||||
width, height
|
||||
});
|
||||
return true;
|
||||
}catch(e2){
|
||||
// 3) Via workbook.addImage() + sheet.addImage({image})
|
||||
try{
|
||||
const img = wb.addImage({ base64: base64Png, extension: 'png' });
|
||||
await sheet.addImage({
|
||||
image: img,
|
||||
topLeftCell: topLeftCell,
|
||||
width, height
|
||||
});
|
||||
return true;
|
||||
}catch(e3){
|
||||
console.warn('Falha ao inserir imagem no Excel:', {e1, e2, e3});
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Exporta TODAS as respostas para Excel: dados + gráfico (PNG)
|
||||
async function exportExcel(){
|
||||
if (!window.XlsxPopulate) {
|
||||
alert("Biblioteca XlsxPopulate não carregada.");
|
||||
return;
|
||||
}
|
||||
|
||||
const items = Array.from(document.querySelectorAll('#history .item'));
|
||||
if(items.length === 0){
|
||||
alert('Não há respostas para exportar.');
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
const wb = await XlsxPopulate.fromBlankAsync();
|
||||
|
||||
for (let i = 0; i < items.length; i++) {
|
||||
const item = items[i];
|
||||
const name = ("Resp " + (i+1)).slice(0, 31);
|
||||
const sheet = wb.addSheet(name);
|
||||
|
||||
// --- Pergunta ---
|
||||
let question = 'Resposta ' + (i+1);
|
||||
const qTitle = item.querySelector('div>b');
|
||||
if (qTitle && qTitle.parentElement) {
|
||||
question = qTitle.parentElement.textContent.replace(/^Pergunta:\s*/,'').trim();
|
||||
}
|
||||
sheet.cell('A1').value('Pergunta:').style({ bold:true });
|
||||
sheet.cell('B1').value(question);
|
||||
sheet.row(1).height(22);
|
||||
|
||||
// --- SQL (se houver) ---
|
||||
const sqlEl = item.querySelector('.sql');
|
||||
if (sqlEl){
|
||||
sheet.cell('A2').value('SQL:').style({ bold:true });
|
||||
sheet.cell('B2').value(sqlEl.textContent);
|
||||
sheet.column('B').width(100);
|
||||
}
|
||||
|
||||
// --- Tabela (se houver) ---
|
||||
let nextRow = 4;
|
||||
const tableEl = item.querySelector('table');
|
||||
if (tableEl){
|
||||
const headers = Array.from(tableEl.querySelectorAll('thead th')).map(th => th.textContent.trim());
|
||||
const rows = Array.from(tableEl.querySelectorAll('tbody tr')).map(tr =>
|
||||
Array.from(tr.querySelectorAll('td')).map(td => td.textContent)
|
||||
);
|
||||
|
||||
if (headers.length){
|
||||
const lastColLetter = String.fromCharCode(64 + headers.length); // A..Z (ok p/ até 26 colunas)
|
||||
sheet.range(`A${nextRow}:${lastColLetter}${nextRow}`).value([headers]).style({ bold:true, fill: 'EFEFEF' });
|
||||
if (rows.length){
|
||||
sheet.range(`A${nextRow+1}:${lastColLetter}${nextRow+rows.length}`).value(rows);
|
||||
}
|
||||
// larguras aproximadas
|
||||
headers.forEach((h, idx) => sheet.column(idx+1).width(Math.max(12, Math.min(40, h.length + 4))));
|
||||
nextRow = nextRow + rows.length + 2;
|
||||
}
|
||||
}
|
||||
|
||||
// --- Gráfico (se houver) ---
|
||||
const canvasEl = item.querySelector('canvas');
|
||||
let embedded = false;
|
||||
|
||||
if (canvasEl){
|
||||
try{
|
||||
const dataUrl = canvasEl.toDataURL('image/png');
|
||||
const base64 = dataUrl.split(',')[1]; // remove prefixo
|
||||
embedded = await insertImage(wb, sheet, base64, `A${nextRow}`, 640, 300);
|
||||
if (embedded) nextRow += 18;
|
||||
}catch(err){
|
||||
console.warn('Falha ao capturar canvas:', err);
|
||||
}
|
||||
}
|
||||
|
||||
// Fallback: se não tinha canvas ou inserção falhou, captura o card inteiro (html2canvas)
|
||||
if (!embedded){
|
||||
try{
|
||||
const snap = await html2canvas(item, { backgroundColor: '#ffffff', scale: 2, useCORS: true });
|
||||
const base64 = snap.toDataURL('image/png').split(',')[1];
|
||||
await insertImage(wb, sheet, base64, `A${nextRow}`, 640, 360);
|
||||
nextRow += 20;
|
||||
}catch(err2){
|
||||
console.warn(`Sem gráfico na aba ${name} (seguindo só com dados).`, err2);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// (Opcional) renomeia a primeira planilha vazia
|
||||
try { wb.sheet(0).name('Resumo'); } catch(e){}
|
||||
|
||||
const blob = await wb.outputAsync();
|
||||
downloadBlob(blob, 'respostas-select-ai.xlsx');
|
||||
} catch (e) {
|
||||
console.error('Falha ao gerar Excel:', e);
|
||||
alert('Falha ao gerar Excel. Veja o console para detalhes.');
|
||||
}
|
||||
}
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
"""
|
||||
|
||||
def _auto_width(ws, start_col, end_col, extra=2):
|
||||
for col_idx in range(start_col, end_col+1):
|
||||
col_letter = get_column_letter(col_idx)
|
||||
max_len = 0
|
||||
for cell in ws[col_letter]:
|
||||
try:
|
||||
max_len = max(max_len, len(str(cell.value)) if cell.value is not None else 0)
|
||||
except:
|
||||
pass
|
||||
ws.column_dimensions[col_letter].width = min(50, max(12, max_len + extra))
|
||||
|
||||
@app.route("/export/xlsx", methods=["POST"])
|
||||
def export_xlsx():
|
||||
"""
|
||||
Espera JSON com uma lista 'items'. Cada item:
|
||||
{
|
||||
"question": str,
|
||||
"sql": str|None,
|
||||
"table": { "headers":[...], "rows":[[...], ...] } | None,
|
||||
"chartPng": "data:image/png;base64,..." | None
|
||||
}
|
||||
Retorna um arquivo XLSX com 1 aba por resposta.
|
||||
"""
|
||||
try:
|
||||
payload = request.get_json(force=True)
|
||||
items = payload.get("items", [])
|
||||
if not items:
|
||||
return jsonify({"error":"sem itens"}), 400
|
||||
|
||||
wb = Workbook()
|
||||
# use a folha ativa como “Resumo”
|
||||
ws0 = wb.active
|
||||
ws0.title = "Resumo"
|
||||
ws0["A1"] = "Export gerado pelo ER Analytics · Select AI"
|
||||
ws0["A2"] = f"Total de respostas: {len(items)}"
|
||||
|
||||
for i, it in enumerate(items, start=1):
|
||||
title = f"Resp {i}"
|
||||
ws = wb.create_sheet(title[:31])
|
||||
|
||||
# Pergunta
|
||||
ws["A1"] = "Pergunta:"
|
||||
ws["A1"].font = ws["A1"].font.copy(bold=True)
|
||||
ws["B1"] = it.get("question") or f"Resposta {i}"
|
||||
ws.row_dimensions[1].height = 22
|
||||
|
||||
# SQL
|
||||
sql = (it.get("sql") or "").strip()
|
||||
if sql:
|
||||
ws["A2"] = "SQL:"
|
||||
ws["A2"].font = ws["A2"].font.copy(bold=True)
|
||||
ws["B2"] = sql
|
||||
|
||||
row = 4
|
||||
|
||||
# Tabela
|
||||
table = it.get("table")
|
||||
if table and table.get("headers"):
|
||||
headers = table["headers"]
|
||||
rows = table.get("rows", [])
|
||||
for j, h in enumerate(headers, start=1):
|
||||
cell = ws.cell(row=row, column=j, value=h)
|
||||
cell.font = cell.font.copy(bold=True)
|
||||
cell.fill = cell.fill.copy()
|
||||
row += 1
|
||||
for r in rows:
|
||||
for j, val in enumerate(r, start=1):
|
||||
ws.cell(row=row, column=j, value=val)
|
||||
row += 1
|
||||
# largura de colunas
|
||||
_auto_width(ws, 1, len(headers))
|
||||
row += 1
|
||||
|
||||
# Gráfico como PNG (se vier)
|
||||
chart_png = it.get("chartPng")
|
||||
if chart_png:
|
||||
try:
|
||||
# aceita dataURL com prefixo
|
||||
if "," in chart_png:
|
||||
chart_png = chart_png.split(",", 1)[1]
|
||||
img_bytes = base64.b64decode(chart_png)
|
||||
# normaliza via PIL (corrige metadados)
|
||||
pil = PILImage.open(BytesIO(img_bytes)).convert("RGBA")
|
||||
buf = BytesIO()
|
||||
pil.save(buf, format="PNG")
|
||||
buf.seek(0)
|
||||
xl_img = XLImage(buf)
|
||||
anchor_cell = f"A{row}"
|
||||
ws.add_image(xl_img, anchor_cell)
|
||||
# empurra linhas para não sobrepor texto
|
||||
row += 20
|
||||
except Exception as e:
|
||||
# segue sem o gráfico
|
||||
pass
|
||||
|
||||
# remove folha “Sheet” se sobrar (caso libs criem extra)
|
||||
for sh in list(wb.sheetnames):
|
||||
if sh.lower().startswith("sheet") and sh != ws0.title:
|
||||
try:
|
||||
del wb[sh]
|
||||
except:
|
||||
pass
|
||||
|
||||
out = BytesIO()
|
||||
wb.save(out)
|
||||
out.seek(0)
|
||||
return send_file(
|
||||
out,
|
||||
mimetype="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
||||
as_attachment=True,
|
||||
download_name="respostas-select-ai.xlsx",
|
||||
)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
def _to_float(x):
|
||||
if x is None: return None
|
||||
s = str(x).strip().replace(',', '.')
|
||||
if s.endswith('%'): s = s[:-1]
|
||||
try:
|
||||
return float(s)
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
def build_chart(headers, rows):
|
||||
"""
|
||||
Gera metadados de gráfico + eixos/legenda:
|
||||
- 1x1 percentual -> pizza (Internados vs Não)
|
||||
- 2 colunas (cat, num) -> barras
|
||||
- 3 colunas com (categoria, count, percentage) -> usa percentage; senão count
|
||||
- 3 colunas com temporal (col contém MONTH/YEAR/DATE) -> linha
|
||||
Retorna dict {type, labels, values, seriesLabel, title, xLabel, yLabel} ou None
|
||||
"""
|
||||
if not headers or not rows: return None
|
||||
H = [h.upper() for h in headers]
|
||||
|
||||
# 1) 1x1 percentual
|
||||
if len(headers) == 1 and len(rows) == 1:
|
||||
v = _to_float(rows[0][0])
|
||||
if v is None: return None
|
||||
pct = v*100 if 0 <= v <= 1 else v
|
||||
if 0 <= pct <= 100:
|
||||
return {
|
||||
"type": "pie",
|
||||
"labels": ["Internados", "Não internados"],
|
||||
"values": [round(pct,2), round(100-pct,2)],
|
||||
"seriesLabel": "Percentual",
|
||||
"title": f"{headers[0]}",
|
||||
"xLabel": "Status",
|
||||
"yLabel": "%"
|
||||
}
|
||||
|
||||
# 2) 2 colunas: categoria × valor
|
||||
if len(headers) == 2:
|
||||
labels, values = [], []
|
||||
for r in rows:
|
||||
labels.append(str(r[0]))
|
||||
vf = _to_float(r[1])
|
||||
if vf is None: return None
|
||||
values.append(vf)
|
||||
return {
|
||||
"type": "bar",
|
||||
"labels": labels,
|
||||
"values": values,
|
||||
"seriesLabel": headers[1],
|
||||
"title": f"{headers[1]} por {headers[0]}",
|
||||
"xLabel": headers[0],
|
||||
"yLabel": headers[1]
|
||||
}
|
||||
|
||||
# 3) 3 colunas: temporal ou categoria + (count, percentage)
|
||||
if len(headers) == 3:
|
||||
# 3a) temporal
|
||||
idx_time = next((i for i,h in enumerate(H) if any(k in h for k in ["DATE","DATA","MONTH","YEAR","TIME"])), None)
|
||||
if idx_time is not None:
|
||||
idx_val = 2
|
||||
labels, values = [], []
|
||||
for r in rows:
|
||||
labels.append(str(r[idx_time]))
|
||||
vf = _to_float(r[idx_val])
|
||||
if vf is None: return None
|
||||
values.append(vf)
|
||||
return {
|
||||
"type": "line",
|
||||
"labels": labels,
|
||||
"values": values,
|
||||
"seriesLabel": headers[idx_val],
|
||||
"title": f"{headers[idx_val]} por {headers[idx_time]}",
|
||||
"xLabel": headers[idx_time],
|
||||
"yLabel": headers[idx_val]
|
||||
}
|
||||
# 3b) categoria + count + percentage
|
||||
idx_pct = next((i for i,h in enumerate(H) if "PERCENT" in h), None)
|
||||
idx_cnt = next((i for i,h in enumerate(H) if any(k in h for k in ["COUNT","NUMBER"])), None)
|
||||
idx_cat = 0
|
||||
idx_val = idx_pct if idx_pct is not None else idx_cnt
|
||||
if idx_val is not None:
|
||||
labels, values = [], []
|
||||
for r in rows:
|
||||
labels.append(str(r[idx_cat]))
|
||||
vf = _to_float(r[idx_val])
|
||||
if vf is None: return None
|
||||
values.append(vf)
|
||||
ylab = headers[idx_val] + (" (%)" if idx_val == idx_pct else "")
|
||||
return {
|
||||
"type": "bar" if idx_val == idx_cnt else "pie",
|
||||
"labels": labels,
|
||||
"values": values,
|
||||
"seriesLabel": headers[idx_val],
|
||||
"title": f"{headers[idx_val]} por {headers[idx_cat]}",
|
||||
"xLabel": headers[idx_cat],
|
||||
"yLabel": ylab
|
||||
}
|
||||
|
||||
return None
|
||||
|
||||
def format_table(headers, rows, limit=500):
|
||||
if not headers: return None
|
||||
rows = rows if len(rows) <= limit else rows[:limit]
|
||||
return {"headers": headers, "rows": rows}
|
||||
|
||||
def run_select_ai(nl_prompt, table_name):
|
||||
frase_segura = nl_prompt.replace("'", "''")
|
||||
sql = f"SELECT AI '{frase_segura}' FROM {table_name}"
|
||||
with pool.acquire() as conn:
|
||||
with conn.cursor() as cur:
|
||||
cur.execute("select user, dbms_cloud_ai.get_profile() from dual")
|
||||
user_, profile_ = cur.fetchone()
|
||||
cur.execute(sql)
|
||||
rows = cur.fetchall()
|
||||
headers = [d[0] for d in cur.description] if cur.description else []
|
||||
return sql, headers, rows, user_, profile_
|
||||
|
||||
def ensure_timeline():
|
||||
if "timeline" not in session:
|
||||
session["timeline"] = []
|
||||
return session["timeline"]
|
||||
|
||||
@app.route("/", methods=["GET","POST"])
|
||||
def index():
|
||||
default_table = "MEU_USUARIO.NLU_ED_ADMISSION"
|
||||
timeline = ensure_timeline()
|
||||
|
||||
if request.method == "POST":
|
||||
frase = request.form["frase"].strip()
|
||||
tabela = request.form.get("tabela", default_table).strip()
|
||||
|
||||
try:
|
||||
sql, headers, rows, user_, profile_ = run_select_ai(frase, tabela)
|
||||
table = format_table(headers, rows)
|
||||
chart = build_chart(headers, rows)
|
||||
|
||||
# empilha no histórico (mantém últimos 10)
|
||||
timeline.append({
|
||||
"prompt": frase,
|
||||
"sql": sql,
|
||||
"table": table,
|
||||
"chart": chart
|
||||
})
|
||||
if len(timeline) > 10:
|
||||
timeline[:] = timeline[-10:]
|
||||
session["timeline"] = timeline
|
||||
|
||||
return render_template_string(
|
||||
PAGE,
|
||||
timeline=timeline,
|
||||
default_table=default_table,
|
||||
db_alias=DB_ALIAS,
|
||||
session_user=user_,
|
||||
profile=profile_
|
||||
)
|
||||
except Exception as e:
|
||||
timeline.append({
|
||||
"prompt": frase,
|
||||
"sql": None,
|
||||
"table": {"headers": ["Erro"], "rows": [[str(e)]]},
|
||||
"chart": None
|
||||
})
|
||||
if len(timeline) > 10:
|
||||
timeline[:] = timeline[-10:]
|
||||
session["timeline"] = timeline
|
||||
return render_template_string(
|
||||
PAGE,
|
||||
timeline=timeline,
|
||||
default_table=default_table,
|
||||
db_alias=DB_ALIAS,
|
||||
session_user=USERNAME,
|
||||
profile=None
|
||||
)
|
||||
|
||||
# GET
|
||||
return render_template_string(
|
||||
PAGE,
|
||||
timeline=timeline,
|
||||
default_table=default_table,
|
||||
db_alias=DB_ALIAS,
|
||||
session_user=USERNAME,
|
||||
profile=None
|
||||
)
|
||||
|
||||
if __name__ == "__main__":
|
||||
app.run(debug=True, port=5001)
|
||||
Reference in New Issue
Block a user