Feat:Added Plant disease prediction

This commit is contained in:
2025-02-23 10:52:31 +05:30
parent dbbc0ce5f2
commit 0048e9f462
+16 -15
View File
@@ -1,42 +1,43 @@
import React, { useState } from "react"; import React, { useState } from "react";
import Loader from "../../components/Loader";
const Ai = () => { const Ai = () => {
const [selectedFile, setSelectedFile] = useState(null); const [selectedFile, setSelectedFile] = useState(null);
const [result, setResult] = useState(""); const [prediction, setPrediction] = useState("");
const [loading, setLoading] = useState(false); const [loading, setLoading] = useState(false);
const [error, setError] = useState(""); const [error, setError] = useState("");
const handleFileChange = (e) => { const handleFileChange = (e) => {
if (e.target.files && e.target.files[0]) {
setSelectedFile(e.target.files[0]); setSelectedFile(e.target.files[0]);
}
}; };
const handleSubmit = async (e) => { const handleSubmit = async (e) => {
e.preventDefault(); e.preventDefault();
if (!selectedFile) return; if (!selectedFile) return;
setLoading(true); setLoading(true);
setResult(""); setPrediction("");
setError(""); setError("");
try { try {
const formData = new FormData(); const formData = new FormData();
formData.append("image", selectedFile); formData.append("image", selectedFile);
// Update the URL if needed. const response = await fetch("http://localhost:3000/predict", {
const response = await fetch("http://localhost:8000/api/v1/vegetable", {
method: "POST", method: "POST",
body: formData, body: formData,
}); });
if (!response.ok) { if (!response.ok) {
throw new Error("Failed to classify image"); throw new Error("Prediction request failed");
} }
const data = await response.json(); const data = await response.json();
// Assume the API returns an object with an "output" field. // Assuming the API returns a JSON object with an "output" field.
setResult(data.output || "No result provided"); setPrediction(data.output || "No prediction returned");
} catch (err) { } catch (err) {
console.error("Error:", err); console.error("Error during prediction:", err);
setError(err.message); setError(err.message);
} finally { } finally {
setLoading(false); setLoading(false);
@@ -46,7 +47,7 @@ const Ai = () => {
return ( return (
<div className="max-w-md mx-auto p-6 bg-white shadow rounded-lg"> <div className="max-w-md mx-auto p-6 bg-white shadow rounded-lg">
<h2 className="text-2xl font-bold mb-4 text-center"> <h2 className="text-2xl font-bold mb-4 text-center">
Vegetable Classifier Plant disease prediction
</h2> </h2>
<form onSubmit={handleSubmit} className="space-y-4"> <form onSubmit={handleSubmit} className="space-y-4">
<input <input
@@ -60,14 +61,14 @@ const Ai = () => {
disabled={loading || !selectedFile} disabled={loading || !selectedFile}
className="w-full bg-blue-600 hover:bg-blue-700 text-white py-2 px-4 rounded focus:outline-none focus:ring-2 focus:ring-blue-500" className="w-full bg-blue-600 hover:bg-blue-700 text-white py-2 px-4 rounded focus:outline-none focus:ring-2 focus:ring-blue-500"
> >
{loading ? <Loader> </Loader> : "Upload and Classify"} {loading ? "Predicting..." : "Predict"}
</button> </button>
</form> </form>
{error && <p className="mt-4 text-red-600 text-center">{error}</p>} {error && <p className="mt-4 text-center text-red-600">{error}</p>}
{result && ( {prediction && (
<div className="mt-4 p-4 bg-green-100 border border-green-300 rounded"> <div className="mt-4 p-4 bg-green-100 border border-green-300 rounded">
<h3 className="text-lg font-semibold">Result:</h3> <h3 className="text-lg font-semibold">Prediction:</h3>
<p>{result}</p> <p>{prediction}</p>
</div> </div>
)} )}
</div> </div>