Versão em produção - código extraído do container

main.py

@@ -0,0 +1,136 @@
+import uuid
+from fastapi import FastAPI, Request, UploadFile, File
+from fastapi.templating import Jinja2Templates
+from fastapi.responses import HTMLResponse, JSONResponse
+from fastapi.staticfiles import StaticFiles
+import os, shutil
+from fastapi.responses import StreamingResponse
+from io import BytesIO
+import pandas as pd
+from sqlalchemy.future import select
+from database import AsyncSessionLocal
+from models import Fatura
+from processor import (
+    fila_processamento,
+    processar_em_lote,
+    status_arquivos,
+    limpar_arquivos_processados
+)
+
+app = FastAPI()
+templates = Jinja2Templates(directory="templates")
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+UPLOAD_DIR = "uploads/temp"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
+@app.get("/", response_class=HTMLResponse)
+def dashboard(request: Request):
+    indicadores = [
+        {"titulo": "Total de Faturas", "valor": 124},
+        {"titulo": "Faturas com ICMS", "valor": "63%"},
+        {"titulo": "Valor Total", "valor": "R$ 280.000,00"},
+    ]
+
+    analise_stf = {
+        "antes": {"percentual_com_icms": 80, "media_valor": 1200},
+        "depois": {"percentual_com_icms": 20, "media_valor": 800},
+    }
+
+    return templates.TemplateResponse("dashboard.html", {
+        "request": request,
+        "cliente_atual": "",
+        "clientes": ["Cliente A", "Cliente B"],
+        "indicadores": indicadores,
+        "analise_stf": analise_stf
+    })
+
+@app.get("/upload", response_class=HTMLResponse)
+def upload_page(request: Request):
+    return templates.TemplateResponse("upload.html", {"request": request})
+
+@app.get("/relatorios", response_class=HTMLResponse)
+def relatorios_page(request: Request):
+    return templates.TemplateResponse("relatorios.html", {"request": request})
+
+@app.get("/parametros", response_class=HTMLResponse)
+def parametros_page(request: Request):
+    return templates.TemplateResponse("parametros.html", {"request": request})
+
+@app.post("/upload-files")
+async def upload_files(files: list[UploadFile] = File(...)):
+    for file in files:
+        temp_path = os.path.join(UPLOAD_DIR, f"{uuid.uuid4()}_{file.filename}")
+        with open(temp_path, "wb") as f:
+            shutil.copyfileobj(file.file, f)
+        await fila_processamento.put({
+            "caminho_pdf": temp_path,
+            "nome_original": file.filename
+        })
+    return {"message": "Arquivos enviados para fila"}
+
+@app.post("/process-queue")
+async def process_queue():
+    resultados = await processar_em_lote()
+    return {"message": "Processamento concluído", "resultados": resultados}
+
+@app.get("/get-status")
+async def get_status():
+    files = []
+    for nome, status in status_arquivos.items():
+        files.append({
+            "nome": nome,
+            "status": status,
+            "mensagem": "---" if status == "Concluído" else status
+        })
+    is_processing = not fila_processamento.empty()
+    return JSONResponse(content={"is_processing": is_processing, "files": files})
+
+@app.post("/clear-all")
+async def clear_all():
+    limpar_arquivos_processados()
+    for f in os.listdir(UPLOAD_DIR):
+        os.remove(os.path.join(UPLOAD_DIR, f))
+    return {"message": "Fila e arquivos limpos"}
+
+@app.get("/export-excel")
+async def export_excel():
+    async with AsyncSessionLocal() as session:
+        result = await session.execute(select(Fatura))
+        faturas = result.scalars().all()
+
+        dados = []
+        for f in faturas:
+            dados.append({
+                "Nome": f.nome,
+                "UC": f.unidade_consumidora,
+                "Referência": f.referencia,
+                "Nota Fiscal": f.nota_fiscal,
+                "Valor Total": f.valor_total,
+                "ICMS (%)": f.icms_aliq,
+                "ICMS (R$)": f.icms_valor,
+                "Base ICMS": f.icms_base,
+                "PIS (%)": f.pis_aliq,
+                "PIS (R$)": f.pis_valor,
+                "Base PIS": f.pis_base,
+                "COFINS (%)": f.cofins_aliq,
+                "COFINS (R$)": f.cofins_valor,
+                "Base COFINS": f.cofins_base,
+                "Consumo (kWh)": f.consumo,
+                "Tarifa": f.tarifa,
+                "Cidade": f.cidade,
+                "Estado": f.estado,
+                "Distribuidora": f.distribuidora,
+                "Data Processamento": f.data_processamento,
+            })
+
+        df = pd.DataFrame(dados)
+        output = BytesIO()
+        df.to_excel(output, index=False, sheet_name="Faturas")
+        output.seek(0)
+
+        return StreamingResponse(
+            output,
+            media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+            headers={"Content-Disposition": "attachment; filename=relatorio_faturas.xlsx"}
+        )