import gradio as gr
from server import predict

iface = gr.Interface(
    fn=lambda img: (lambda pred: f"Predicted Disease: {pred.get('Disease', 'Error')}\nRecommended Treatment: {pred.get('Treatment', 'N/A')}")(predict(img)),
    inputs=gr.Image(type="numpy"),
    outputs="text",
    title="🌿 Plant Disease Detector",
    description="Upload an image to identify plant diseases.",
)

if __name__ == "__main__":
    iface.launch(server_name="0.0.0.0", server_port=7861, share=True)