Skip to content

Commit

Permalink
ui changes
Browse files Browse the repository at this point in the history
  • Loading branch information
@nisalrenuja
nisalrenuja committed Nov 4, 2024
1 parent 4928903 commit ebac098
Show file tree
Hide file tree
Showing 3 changed files with 168 additions and 157 deletions.
105 changes: 54 additions & 51 deletions src/app/coco-ssd/page.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ import ImageUploader from "@/components/pages/coco-ssd/ImageUploader";
import DetectedObjectsList from "@/components/pages/coco-ssd/DetectedObjectsList";
import Header from "@/components/common/Header";
import Footer from "@/components/common/Footer";
import { Github } from "lucide-react";

export default function ObjectDetection() {
const canvasRef = useRef(null);
Expand Down Expand Up @@ -97,61 +96,65 @@ export default function ObjectDetection() {
return (
<div className="flex flex-col min-h-screen bg-white dark:bg-gray-900">
<Header />
<Card className="w-full max-w-3xl mx-auto">
<CardHeader>
<CardTitle className="text-2xl font-bold text-center">
COCO-SSD Model
</CardTitle>
</CardHeader>
<CardContent className="space-y-4">
{loadingState.loading && loadingState.progress < 100 && (
<div className="space-y-2">
<p className="text-sm text-muted-foreground">Loading model...</p>
<Progress value={loadingState.progress} className="w-full" />
</div>
)}
<main className="flex-grow container mx-auto px-4 py-8">
<Card className="w-full max-w-3xl mx-auto">
<CardHeader>
<CardTitle className="text-2xl font-bold text-center">
COCO-SSD Model
</CardTitle>
</CardHeader>
<CardContent className="space-y-4">
{loadingState.loading && loadingState.progress < 100 && (
<div className="space-y-2">
<p className="text-sm text-muted-foreground">
Loading model...
</p>
<Progress value={loadingState.progress} className="w-full" />
</div>
)}

{error && (
<Alert variant="destructive">
<AlertTitle>Error</AlertTitle>
<AlertDescription>{error}</AlertDescription>
</Alert>
)}
{error && (
<Alert variant="destructive">
<AlertTitle>Error</AlertTitle>
<AlertDescription>{error}</AlertDescription>
</Alert>
)}

<div className="grid gap-4 md:grid-cols-2">
<ImageUploader
handleFileChange={handleFileChange}
startDetection={startDetection}
isDisabled={!uploadedImage || loadingState.loading}
/>
<div className="grid gap-4 md:grid-cols-2">
<ImageUploader
handleFileChange={handleFileChange}
startDetection={startDetection}
isDisabled={!uploadedImage || loadingState.loading}
/>

<div className="relative aspect-square">
{uploadedImage && (
<>
<Image
ref={imageRef}
src={uploadedImage}
alt="Uploaded"
height={400}
width={400}
className="object-cover rounded-lg"
/>
<canvas
ref={canvasRef}
width={400}
height={400}
className="absolute top-0 left-0 w-full h-full"
/>
</>
)}
<div className="relative aspect-square">
{uploadedImage && (
<>
<Image
ref={imageRef}
src={uploadedImage}
alt="Uploaded"
height={400}
width={400}
className="object-cover rounded-lg"
/>
<canvas
ref={canvasRef}
width={400}
height={400}
className="absolute top-0 left-0 w-full h-full"
/>
</>
)}
</div>
</div>
</div>

{detectedObjects.length > 0 && (
<DetectedObjectsList detectedObjects={detectedObjects} />
)}
</CardContent>
</Card>
{detectedObjects.length > 0 && (
<DetectedObjectsList detectedObjects={detectedObjects} />
)}
</CardContent>
</Card>
</main>
<Footer />
</div>
);
Expand Down
178 changes: 92 additions & 86 deletions src/app/mobilenet/page.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@ import { useState, useEffect, useMemo } from "react";
import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input";
import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";
import { Alert, AlertDescription, AlertTitle } from "@/components/ui/alert";
import ImageUploader from "@/components/pages/coco-ssd/ImageUploader";
import { Progress } from "@/components/ui/progress";
import { Upload, Github } from "lucide-react";
import { loadModel, loadImage } from "@/utils/imageProcessing";
Expand All @@ -18,15 +20,22 @@ export default function MobileNet() {
const [selectedFile, setSelectedFile] = useState(null);
const [imagePreview, setImagePreview] = useState(null);
const [error, setError] = useState(null);
const [loadingState, setLoadingState] = useState({
loading: true,
progress: 0,
});

useEffect(() => {
const loadTensorFlowModel = async () => {
try {
setLoadingState({ loading: true, progress: 10 });
const loadedModel = await loadModel();
setModel(loadedModel);
setLoadingState({ loading: false, progress: 100 });
} catch (error) {
console.error("Error loading the model:", error);
setError("Failed to load the model. Please try again later.");
setLoadingState({ loading: false, progress: 100 });
}
};
loadTensorFlowModel();
Expand Down Expand Up @@ -98,96 +107,93 @@ export default function MobileNet() {
<div className="flex flex-col min-h-screen bg-white dark:bg-gray-900">
<Header />
<main className="flex-grow container mx-auto px-4 py-8">
<Card className="max-w-md mx-auto sm:max-w-full md:max-w-md lg:max-w-lg">
<Card className="w-full max-w-3xl mx-auto">
<CardHeader>
<CardTitle>MobileNet Model</CardTitle>
<CardTitle className="text-2xl font-bold text-center">
MobileNet Model
</CardTitle>
</CardHeader>
<CardContent>
<div className="space-y-4">
<div className="flex flex-col sm:flex-row sm:items-center space-y-4 sm:space-y-0 sm:space-x-2">
<Input
type="file"
id="image-upload"
onChange={handleFileChange}
className="flex-grow"
/>
<Button
onClick={handleAnalyzeClick}
disabled={isAnalyzing}
className="w-full sm:w-auto"
>
{isAnalyzing ? (
<div className="flex items-center">
<span className="spinner-border animate-spin inline-block w-4 h-4 border-4 rounded-full"></span>
<span className="ml-2">Analyzing...</span>
</div>
) : (
<>
<Upload className="w-4 h-4 mr-2" />
Analyze
</>
)}
</Button>
<CardContent className="space-y-4">
{loadingState.loading && loadingState.progress < 100 && (
<div className="space-y-2">
<p className="text-sm text-muted-foreground">
Loading model...
</p>
<Progress value={loadingState.progress} className="w-full" />
</div>
)}

{error && (
<Alert variant="destructive">
<AlertTitle>Error</AlertTitle>
<AlertDescription>{error}</AlertDescription>
</Alert>
)}

<div className="grid gap-4 md:grid-cols-2">
<ImageUploader
handleFileChange={handleFileChange}
startDetection={handleAnalyzeClick}
isDisabled={!imagePreview || loadingState.loading}
/>

<div className="relative aspect-square">
{imagePreview && (
<>
<Image
src={imagePreview}
alt="Preview"
className="object-cover rounded-lg"
width={400}
height={400}
/>
</>
)}
</div>

{imagePreview && (
<div className="mt-4">
<h3 className="font-semibold mb-2">Image Preview:</h3>
<Image
src={imagePreview}
alt="Preview"
className="max-w-full h-auto rounded-lg"
width={500}
height={500}
/>
</div>
)}

{isAnalyzing && <Progress value={50} className="w-full" />}

{error && <p className="text-red-600 mt-2">{error}</p>}

{predictions.length > 0 && (
<div className="mt-4">
<h3 className="font-semibold mb-2">Results:</h3>
<ul className="space-y-2">
{predictions.map((pred, index) => (
<li
key={index}
className="grid grid-cols-3 items-center w-full gap-1"
>
<div className="font-xs">
{pred.className}
<a
href={`https://en.wikipedia.org/wiki/${pred.className}`}
target="_blank"
rel="noreferrer"
className="text-blue-500 ml-2"
>
(Learn More)
</a>
</div>

<div className="w-full">
<Progress
value={pred.probability * 100}
className="w-full"
/>
</div>

<div className="text-right font-mono">
{(pred.probability * 100).toFixed(2)}%
</div>
</li>
))}
</ul>

<Button onClick={downloadResults} className="mt-4 w-full">
Download Results
</Button>
</div>
)}
</div>

{isAnalyzing && <Progress value={50} className="w-full" />}

{predictions.length > 0 && (
<div className="mt-4">
<h3 className="font-semibold mb-2">Results:</h3>
<ul className="space-y-2">
{predictions.map((pred, index) => (
<li
key={index}
className="grid grid-cols-3 items-center w-full gap-1"
>
<div className="font-xs">
{pred.className}
<a
href={`https://en.wikipedia.org/wiki/${pred.className}`}
target="_blank"
rel="noreferrer"
className="text-blue-500 ml-2"
>
(Learn More)
</a>
</div>

<div className="w-full">
<Progress
value={pred.probability * 100}
className="w-full"
/>
</div>

<div className="text-right font-mono">
{(pred.probability * 100).toFixed(2)}%
</div>
</li>
))}
</ul>

<Button onClick={downloadResults} className="mt-4 w-full">
Download Results
</Button>
</div>
)}
</CardContent>
</Card>
</main>
Expand Down
42 changes: 22 additions & 20 deletions src/components/pages/home/TensorFlowModels.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,28 +12,30 @@ import { ArrowRight, Brain, Camera, Mic, FileText } from "lucide-react";
const models = [
{
title: "MobileNet",
description: "Classify images into predefined categories",
icon: Camera,
description:
"A lightweight, efficient deep learning model designed for mobile and embedded devices, optimized for tasks like image classification and object detection.",
// icon: Camera,
link: "/mobilenet",
},
{
title: "Natural Language Processing",
description: "Process and analyze human language",
icon: FileText,
link: "/models/nlp",
},
{
title: "Speech Recognition",
description: "Convert spoken language into text",
icon: Mic,
link: "/models/speech-recognition",
},
{
title: "Neural Network",
description: "Build and train custom neural networks",
icon: Brain,
link: "/models/neural-network",
title: "COCO-SSD",
description:
"An object detection model trained on the COCO dataset, capable of identifying and localizing multiple objects in images in real-time.",
// icon: FileText,
link: "/coco-ssd",
},
// {
// title: "Speech Recognition",
// description: "Convert spoken language into text",
// icon: Mic,
// link: "/models/speech-recognition",
// },
// {
// title: "Neural Network",
// description: "Build and train custom neural networks",
// icon: Brain,
// link: "/models/neural-network",
// },
];

export default function TensorFlowModels() {
Expand All @@ -43,9 +45,9 @@ export default function TensorFlowModels() {
{models.map((model, index) => (
<Card key={index} className="flex flex-col">
<CardHeader>
<div className="w-12 h-12 bg-primary rounded-full flex items-center justify-center mb-4">
{/* <div className="w-12 h-12 bg-primary rounded-full flex items-center justify-center mb-4">
<model.icon className="w-6 h-6 text-primary-foreground" />
</div>
</div> */}
<CardTitle>{model.title}</CardTitle>
<CardDescription>{model.description}</CardDescription>
</CardHeader>
Expand Down

0 comments on commit ebac098

Please sign in to comment.