Feat:Added Plant disease prediction
This commit is contained in:
@@ -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) => {
|
||||||
setSelectedFile(e.target.files[0]);
|
if (e.target.files && 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>
|
||||||
|
|||||||
Reference in New Issue
Block a user