Spaces:
Sleeping
Sleeping
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() |