app.routes

 1"""
 2Módulo routes
 3
 4Define as rotas HTTP da aplicação Flask para processar uploads de arquivos ou textos.
 5Encaminha para o pipeline de processamento e retorna a classificação e resposta sugerida.
 6"""
 7
 8from flask import Blueprint, request, jsonify
 9from app.pipeline import process_uploaded_file, process_raw_text
10
11import logging
12logging.basicConfig(level=logging.INFO)
13
14routes = Blueprint('routes', __name__)
15
16@routes.route("/upload", methods=["POST"])
17def upload():
18    """
19    Rota POST /upload
20
21    Recebe arquivos (TXT ou PDF) e/ou texto bruto via formulário multipart.
22    Processa o conteúdo chamando o pipeline, que realiza extração, pré-processamento,
23    classificação via IA e armazenamento em cache.
24
25    O endpoint aceita:
26    - múltiplos arquivos enviados no campo "file"
27    - texto enviado no campo "text"
28
29    Retorna:
30        JSON:
31        {
32            "results": [
33                {
34                    "type": "file" ou "text",
35                    "file": "nome_do_arquivo.txt" (quando for arquivo),
36                    "categoria": "Produtivo" ou "Improdutivo",
37                    "resposta_sugerida": "..."
38                },
39                ...
40            ]
41        }
42
43    Status Codes:
44        - 200: Sucesso no processamento (mesmo com alguns erros parciais nos arquivos).
45        - 400: Nenhum arquivo ou texto foi enviado.
46    """
47    results = []
48
49    files = request.files.getlist("file")
50    for file in files:
51        if file.filename == '':
52            continue
53        try:
54            classification = process_uploaded_file(file)
55            logging.info(f"--- {file.filename} Classificação --- {classification}")
56            results.append({
57                "type": "file",
58                "file": file.filename,
59                "categoria": classification.get("categoria"),
60                "resposta_sugerida": classification.get("resposta_sugerida")
61            })
62        except ValueError as e:
63            logging.error(f"Erro ao processar {file.filename}: {str(e)}", exc_info=True)
64            results.append({
65                "type": "file",
66                "file": file.filename,
67                "error": str(e)
68            })
69
70    raw_text = request.form.get("text", "").strip()
71    if raw_text:
72        try:
73            classification = process_raw_text(raw_text)
74            logging.info(f"--- Texto inserido Classificação --- {classification}")
75            results.append({
76                "type": "text",
77                "categoria": classification.get("categoria"),
78                "resposta_sugerida": classification.get("resposta_sugerida")
79            })
80        except ValueError as e:
81            logging.error("Erro ao processar texto inserido.", exc_info=True)
82            results.append({
83                "type": "text",
84                "error": str(e)
85            })
86
87    if not results:
88        return jsonify({"error": "Nenhum arquivo ou texto enviado."}), 400
89
90    return jsonify({"results": results})