Implementa resposta rica com JSON (info de doenças)

doencas.json

@@ -0,0 +1,97 @@
+{
+  "Oidio": {
+    "descricao": "Descrição aqui.",
+    "sintomas": [
+      "Sintoma 1.",
+      "Sintoma 2!",
+      "Sintoma 3?"
+    ],
+    "tratamento": "Tratamento aqui."
+  },
+  "Mancha Parda": {
+    "descricao": "Descrição aqui.",
+    "sintomas": [
+      "Sintoma 1.",
+      "Sintoma 2!",
+      "Sintoma 3?"
+    ],
+    "tratamento": "Tratamento aqui."
+  },
+  "Ferrugem": {
+    "descricao": "Causada pelo fungo Phakopsora pachyrhizi, é uma das doenças mais severas da soja...",
+    "sintomas": [
+      "Pequenas pústulas (bolinhas) na parte de baixo da folha.",
+      "Cor amarelada que evolui para marrom-escuro.",
+      "Desfolha precoce."
+    ],
+    "tratamento": "Aplicação de fungicidas sistêmicos e de contato. Rotacionar culturas e usar variedades resistentes."
+  },
+  "Mosaico Amarelo": {
+    "descricao": "Descrição aqui.",
+    "sintomas": [
+      "Sintoma 1.",
+      "Sintoma 2!",
+      "Sintoma 3?"
+    ],
+    "tratamento": "Tratamento aqui."
+  },
+  "Virus do Mosaico": {
+    "descricao": "Causado pelo Soybean Mosaic Virus (SMV), transmitido por pulgões.",
+    "sintomas": [
+      "Folhas novas ficam enrugadas e com padrões de verde claro e escuro (mosaico).",
+      "Plantas podem ficar atrofiadas."
+    ],
+    "tratamento": "Não há cura. Use sementes sadias e controle os pulgões (vetores) com inseticidas."
+  },
+  "Mancha bacteriana": {
+    "descricao": "Descrição aqui.",
+    "sintomas": [
+      "Sintoma 1.",
+      "Sintoma 2!",
+      "Sintoma 3?"
+    ],
+    "tratamento": "Tratamento aqui."
+  },
+  "Podridão do Sul": {
+    "descricao": "Descrição aqui.",
+    "sintomas": [
+      "Sintoma 1.",
+      "Sintoma 2!",
+      "Sintoma 3?"
+    ],
+    "tratamento": "Tratamento aqui."
+  },
+  "Queima das folhas por cercosporea": {
+    "descricao": "Descrição aqui.",
+    "sintomas": [
+      "Sintoma 1.",
+      "Sintoma 2!",
+      "Sintoma 3?"
+    ],
+    "tratamento": "Tratamento aqui."
+  },
+  "Sindome da Morte Subita": {
+    "descricao": "Descrição aqui.",
+    "sintomas": [
+      "Sintoma 1.",
+      "Sintoma 2!",
+      "Sintoma 3?"
+    ],
+    "tratamento": "Tratamento aqui."
+  },
+  "Saudavel": {
+    "descricao": "A folha não apresenta sinais visíveis de patógenos.",
+    "sintomas": ["Nenhum."],
+    "tratamento": "Continuar monitoramento e boas práticas agrícolas."
+  },
+  
+  "Lagarta": {
+    "descricao": "Diversas espécies de lagartas, como a lagarta-da-soja (Anticarsia gemmatalis) ou a falsa-medideira (Chrysodeixis includens), atacam as lavouras de soja, causando desfolha e redução da produtividade.",
+    "sintomas": [
+      "Folhas comidas ou roídas, muitas vezes começando pelas bordas.",
+      "Presença de fezes (pequenas bolinhas escuras) nas folhas ou no solo.",
+      "Em ataques severos, pode ocorrer desfolha completa, restando apenas os talos."
+    ],
+    "tratamento": "Monitoramento constante (batida de pano) para identificar o nível de infestação. O controle é feito com inseticidas biológicos (como o Bt - Bacillus thuringiensis) ou químicos, dependendo da espécie da lagarta e do estágio da cultura."
+  }
+}

main.py

@@ -1,6 +1,7 @@
 import os
 import shutil
 import joblib
+import json
 import numpy as np
 from fastapi import FastAPI, File, UploadFile, Depends, HTTPException, status
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
@@ -29,7 +30,7 @@ async def verify_token(credentials: HTTPAuthorizationCredentials = Depends(secur
         )
     return credentials.credentials
 
-#2. CARREGAMENTO DOS MODELOS DE CLASSIFICAÇÃO (ML)
+#2. CARREGAMENTO DOS MODELOS E DADOS
 #Carrega os 4 arquivos .pkl UMA VEZ quando a API inicia.
 print("Carregando modelos de classificação (.pkl)...")
 try:
@@ -42,8 +43,17 @@ except FileNotFoundError:
     print("ERRO: Arquivos .pkl do modelo não encontrados. Certifique-se de que 'scaler.pkl', 'umap_reducer.pkl', 'svm_model.pkl', e 'encoder.pkl' estão no repositório.")
     #Em um cenário real, poderíamos impedir a API de iniciar aqui
 
-#3. ENDPOINTS DA API
+print("Carregando banco de dados de informações (doencas.json)...")
+try:
+    with open('doencas.json', 'r', encoding='utf-8') as f:
+        DB_INFO = json.load(f)  #Carrega o JSON para a memória
+    print("Banco de dados de informações carregado.")
+except FileNotFoundError:
+    print("ERRO: 'doencas.json' não encontrado.")
+    DB_INFO = {} #Inicia vazio para a API não quebrar
 
+
+#3. ENDPOINTS DA API
 @app.post("/classify/")
 async def classify_image(file: UploadFile = File(...), token: str = Depends(verify_token)):
     """
@@ -80,7 +90,20 @@ async def classify_image(file: UploadFile = File(...), token: str = Depends(veri
         #Converte a predição numérica (ex: 2) de volta para o nome da classe (ex: "Ferrugem")
         classe_predita = ENCODER.inverse_transform(pred_numerica)[0]
 
-        return {"diagnostico": classe_predita}
+        #BUSCA AS INFORMAÇÕES ADICIONAIS (json)
+        #Usa .get() para buscar a chave. Se não encontrar, retorna um 
+        #dicionário vazio ou uma mensagem de erro, com segurança
+        info_adicional = DB_INFO.get(classe_predita, {
+            "descricao": "Informações não disponíveis.",
+            "sintomas": [],
+            "tratamento": ""
+        })
+
+        #Retorna a resposta "rica"
+        return {
+            "diagnostico": classe_predita,
+            "info": info_adicional
+        }
 
     except Exception as e:
         #Pega qualquer erro que acontecer durante a extração ou classificação