import gradio as gr
from pathlib import Path

from model_comparison import ModelComparison, discover_example_images

comparison = ModelComparison()
PROJECT_DIR = Path(__file__).resolve().parent
example_images = discover_example_images(str(PROJECT_DIR / "example_images"))


def classify_image(image_path: str):
    return comparison.classify_all(image_path)


iface = gr.Interface(
    fn=classify_image,
    inputs=gr.Image(type="filepath", label="Upload bean leaf image"),
    outputs=gr.JSON(label="Model Comparison"),
    title="Bean Disease Classification Comparison",
    description=(
        "Compare predictions for bean leaf disease classification. "
        "Upload a bean leaf image to get predictions from three models: "
        "custom ViT (fine-tuned), CLIP (zero-shot), and OpenAI Vision (reasoning)."
    ),
    examples=example_images if example_images else None,
)


if __name__ == "__main__":
    iface.launch()