DHEIVER's picture
Update app.py
37089f2 verified
raw
history blame
3.42 kB
import gradio as gr
import numpy as np
from analisador_iridologico import (
pre_processar_imagem, detectar_esclera, detectar_iris_pupila,
analisar_textura_setorial, validar_metricas, AnalisadorIridologicoNLP
)
def criar_interface():
theme = gr.themes.Soft(
primary_hue="teal",
secondary_hue="green",
).set(
body_text_color="#2A9D8F",
block_title_text_color="#264653",
block_label_text_color="#2A9D8F",
input_background_fill="#E9F5F3",
button_primary_background_fill="#2A9D8F",
button_primary_background_fill_dark="#264653",
)
def processar_imagem(imagem):
try:
imagem_processada = pre_processar_imagem(imagem)
mask_esclera = detectar_esclera(imagem_processada)
iris_info, pupil_info = detectar_iris_pupila(imagem_processada, mask_esclera)
if iris_info is None or pupil_info is None:
return imagem, "Não foi possível detectar íris ou pupila corretamente."
analise_setorial = analisar_textura_setorial(imagem_processada, iris_info, pupil_info)
# Avaliando circularidade
metricas = {
'pupila': {
'raio': pupil_info[2],
'circularidade': avaliar_circularidade(pupil_info)
},
'iris': {
'densidade_media': np.mean([dados['contraste'] for dados in analise_setorial.values()]),
'homogeneidade': np.mean([dados['homogeneidade'] for dados in analise_setorial.values()])
},
}
metricas = validar_metricas(metricas)
analisador = AnalisadorIridologicoNLP()
interpretacao_nlp = analisador.gerar_interpretacao(metricas)
relatorio = "ANÁLISE IRIDOLÓGICA DETALHADA\n\n"
relatorio += "1. MEDIDAS ESTRUTURAIS\n"
relatorio += f"Pupila: Centro ({pupil_info[0]}, {pupil_info[1]}), Raio {pupil_info[2]}px\n"
relatorio += "2. ANÁLISE SETORIAL\n"
for setor, dados in analise_setorial.items():
relatorio += f"Setor {setor}: Contraste {dados['contraste']}, Homogeneidade {dados['homogeneidade']}\n"
relatorio += f"3. INTERPRETAÇÃO NLP: {interpretacao_nlp}\n"
return imagem, relatorio
except Exception as e:
return imagem, f"Erro durante o processamento: {str(e)}"
with gr.Blocks(theme=theme, title="Análise Iridológica Avançada") as interface:
gr.Markdown("""
# Sistema Avançado de Análise Iridológica
### Detecção precisa de esclera, íris e pupila com análise setorial e interpretação em linguagem natural
""")
with gr.Row():
input_image = gr.Image(label="Imagem do Olho", type="numpy")
output_image = gr.Image(label="Análise Visual")
analysis_btn = gr.Button("Analisar Olho", variant="primary")
output_text = gr.Textbox(label="Relatório de Análise", lines=20)
analysis_btn.click(
fn=processar_imagem,
inputs=[input_image],
outputs=[output_image, output_text]
)
return interface
def main():
interface = criar_interface()
interface.launch(share=True)
if __name__ == "__main__":
main()