Upload index.html

templates/index.html

@@ -19,281 +19,151 @@
             position: relative;
         }
 
-        /* Animated background particles */
-        .particles {
-            position: absolute;
-            width: 100%;
-            height: 100%;
-            overflow: hidden;
-            z-index: 0;
+        /* --- NEW: Tab Styles --- */
+        .tabs {
+            display: flex;
+            justify-content: center;
+            margin-bottom: 2rem;
+            animation: slideDown 1s ease-out;
         }
 
-        .particle {
-            position: absolute;
-            width: 2px;
-            height: 2px;
+        .tab-button {
+            background: transparent;
+            border: 2px solid #00d4ff;
+            color: #00d4ff;
+            padding: 10px 25px;
+            margin: 0 10px;
+            font-size: 1rem;
+            font-weight: 600;
+            border-radius: 50px;
+            cursor: pointer;
+            transition: all 0.3s ease;
+        }
+        
+        .tab-button:hover {
+            background: rgba(0, 212, 255, 0.2);
+            transform: translateY(-2px);
+        }
+
+        .tab-button.active {
             background: #00d4ff;
-            border-radius: 50%;
-            animation: float 6s ease-in-out infinite;
-            opacity: 0.6;
+            color: #1a1a2e;
+            box-shadow: 0 0 20px rgba(0, 212, 255, 0.5);
         }
+        /* --- End Tab Styles --- */
 
+        .particles {
+            position: absolute; width: 100%; height: 100%; overflow: hidden; z-index: 0;
+        }
+        .particle {
+            position: absolute; width: 2px; height: 2px; background: #00d4ff; border-radius: 50%; animation: float 6s ease-in-out infinite; opacity: 0.6;
+        }
         @keyframes float {
             0%, 100% { transform: translateY(0px) rotate(0deg); }
             50% { transform: translateY(-20px) rotate(180deg); }
         }
-
         .container {
-            position: relative;
-            z-index: 1;
-            max-width: 800px;
-            margin: 0 auto;
-            padding: 2rem;
-            min-height: 100vh;
-            display: flex;
-            flex-direction: column;
-            justify-content: center;
-            align-items: center;
+            position: relative; z-index: 1; max-width: 800px; margin: 0 auto; padding: 2rem; min-height: 100vh; display: flex; flex-direction: column; justify-content: center; align-items: center;
         }
-
         .header {
-            text-align: center;
-            margin-bottom: 3rem;
-            animation: slideDown 1s ease-out;
+            text-align: center; margin-bottom: 2rem; animation: slideDown 1s ease-out;
         }
-
         .title {
-            font-size: 3.5rem;
-            font-weight: 700;
-            background: linear-gradient(45deg, #00d4ff, #ff00ff, #00ff88);
-            background-size: 200% 200%;
-            -webkit-background-clip: text;
-            -webkit-text-fill-color: transparent;
-            background-clip: text;
-            animation: gradientShift 3s ease-in-out infinite;
-            margin-bottom: 1rem;
-            text-shadow: 0 0 30px rgba(0, 212, 255, 0.5);
+            font-size: 3.5rem; font-weight: 700; background: linear-gradient(45deg, #00d4ff, #ff00ff, #00ff88); background-size: 200% 200%; -webkit-background-clip: text; -webkit-text-fill-color: transparent; background-clip: text; animation: gradientShift 3s ease-in-out infinite; margin-bottom: 1rem; text-shadow: 0 0 30px rgba(0, 212, 255, 0.5);
         }
-
         .subtitle {
-            font-size: 1.2rem;
-            color: #a0a0a0;
-            font-weight: 300;
+            font-size: 1.2rem; color: #a0a0a0; font-weight: 300;
         }
-
         .upload-area {
-            width: 100%;
-            max-width: 500px;
-            min-height: 300px;
-            border: 2px dashed #00d4ff;
-            border-radius: 20px;
-            background: rgba(0, 212, 255, 0.05);
-            backdrop-filter: blur(10px);
-            display: flex;
-            flex-direction: column;
-            justify-content: center;
-            align-items: center;
-            cursor: pointer;
-            transition: all 0.3s ease;
-            position: relative;
-            overflow: hidden;
-            animation: slideUp 1s ease-out 0.3s both;
+            width: 100%; max-width: 500px; min-height: 300px; border: 2px dashed #00d4ff; border-radius: 20px; background: rgba(0, 212, 255, 0.05); backdrop-filter: blur(10px); display: flex; flex-direction: column; justify-content: center; align-items: center; cursor: pointer; transition: all 0.3s ease; position: relative; overflow: hidden; animation: slideUp 1s ease-out 0.3s both;
         }
-
         .upload-area:hover {
-            border-color: #ff00ff;
-            background: rgba(255, 0, 255, 0.05);
-            transform: translateY(-5px);
-            box-shadow: 0 20px 40px rgba(0, 212, 255, 0.2);
+            border-color: #ff00ff; background: rgba(255, 0, 255, 0.05); transform: translateY(-5px); box-shadow: 0 20px 40px rgba(0, 212, 255, 0.2);
         }
-
         .upload-area.dragover {
-            border-color: #00ff88;
-            background: rgba(0, 255, 136, 0.1);
-            transform: scale(1.02);
+            border-color: #00ff88; background: rgba(0, 255, 136, 0.1); transform: scale(1.02);
         }
-
         .upload-icon {
-            font-size: 4rem;
-            color: #00d4ff;
-            margin-bottom: 1rem;
-            transition: all 0.3s ease;
+            font-size: 4rem; color: #00d4ff; margin-bottom: 1rem; transition: all 0.3s ease;
         }
-
         .upload-area:hover .upload-icon {
-            color: #ff00ff;
-            transform: scale(1.1);
+            color: #ff00ff; transform: scale(1.1);
         }
-
         .upload-text {
-            color: #ffffff;
-            font-size: 1.1rem;
-            margin-bottom: 0.5rem;
-            font-weight: 500;
+            color: #ffffff; font-size: 1.1rem; margin-bottom: 0.5rem; font-weight: 500;
         }
-
         .upload-subtext {
-            color: #a0a0a0;
-            font-size: 0.9rem;
+            color: #a0a0a0; font-size: 0.9rem;
         }
-
         .file-input {
             display: none;
         }
-
         .preview-container {
-            margin-top: 2rem;
-            text-align: center;
-            animation: fadeIn 0.5s ease-out;
+            margin-top: 2rem; text-align: center; animation: fadeIn 0.5s ease-out;
         }
-
         .preview-image {
-            max-width: 100%;
-            max-height: 300px;
-            border-radius: 15px;
-            box-shadow: 0 10px 30px rgba(0, 0, 0, 0.5);
-            border: 2px solid #00d4ff;
+            max-width: 100%; max-height: 300px; border-radius: 15px; box-shadow: 0 10px 30px rgba(0, 0, 0, 0.5); border: 2px solid #00d4ff;
         }
-
         .predict-button {
-            background: linear-gradient(45deg, #00d4ff, #0099cc);
-            border: none;
-            color: white;
-            padding: 15px 40px;
-            font-size: 1.1rem;
-            font-weight: 600;
-            border-radius: 50px;
-            cursor: pointer;
-            margin-top: 1.5rem;
-            transition: all 0.3s ease;
-            text-transform: uppercase;
-            letter-spacing: 1px;
-            position: relative;
-            overflow: hidden;
+            background: linear-gradient(45deg, #00d4ff, #0099cc); border: none; color: white; padding: 15px 40px; font-size: 1.1rem; font-weight: 600; border-radius: 50px; cursor: pointer; margin-top: 1.5rem; transition: all 0.3s ease; text-transform: uppercase; letter-spacing: 1px; position: relative; overflow: hidden;
         }
-
         .predict-button:hover {
-            transform: translateY(-2px);
-            box-shadow: 0 10px 25px rgba(0, 212, 255, 0.4);
-            background: linear-gradient(45deg, #ff00ff, #cc0099);
+            transform: translateY(-2px); box-shadow: 0 10px 25px rgba(0, 212, 255, 0.4); background: linear-gradient(45deg, #ff00ff, #cc0099);
         }
-
         .predict-button:disabled {
-            opacity: 0.6;
-            cursor: not-allowed;
-            transform: none;
+            opacity: 0.6; cursor: not-allowed; transform: none;
         }
-
         .loading {
-            display: none;
-            margin-top: 1rem;
+            display: none; margin-top: 1rem;
         }
-
         .spinner {
-            width: 40px;
-            height: 40px;
-            border: 4px solid rgba(0, 212, 255, 0.3);
-            border-top: 4px solid #00d4ff;
-            border-radius: 50%;
-            animation: spin 1s linear infinite;
-            margin: 0 auto;
+            width: 40px; height: 40px; border: 4px solid rgba(0, 212, 255, 0.3); border-top: 4px solid #00d4ff; border-radius: 50%; animation: spin 1s linear infinite; margin: 0 auto;
         }
-
-        .result-container {
+        
+        /* --- NEW: Result Area Styles --- */
+        .results-wrapper {
             margin-top: 2rem;
-            padding: 2rem;
-            background: rgba(255, 255, 255, 0.05);
-            backdrop-filter: blur(15px);
-            border-radius: 20px;
-            border: 1px solid rgba(0, 212, 255, 0.3);
-            animation: slideUp 0.5s ease-out;
-            text-align: center;
+            width: 100%;
+            max-width: 550px;
         }
-
-        .result-class {
-            font-size: 2rem;
-            font-weight: 700;
-            color: #00ff88;
-            margin-bottom: 1rem;
+        .result-container {
+            padding: 2rem; background: rgba(255, 255, 255, 0.05); backdrop-filter: blur(15px); border-radius: 20px; border: 1px solid rgba(0, 212, 255, 0.3); animation: slideUp 0.5s ease-out; text-align: center;
+        }
+        .result-title {
+            font-size: 1.1rem;
+            color: #00d4ff;
+            font-weight: 400;
             text-transform: uppercase;
-            letter-spacing: 2px;
+            letter-spacing: 1px;
+            margin-bottom: 1.5rem;
+            border-bottom: 1px solid rgba(0, 212, 255, 0.2);
+            padding-bottom: 0.8rem;
+        }
+        .result-class {
+            font-size: 2rem; font-weight: 700; color: #00ff88; margin-bottom: 1rem; text-transform: uppercase; letter-spacing: 2px;
         }
-
         .result-confidence {
-            font-size: 1.2rem;
-            color: #ffffff;
-            margin-bottom: 0.5rem;
+            font-size: 1.2rem; color: #ffffff; margin-bottom: 0.5rem;
         }
-
         .confidence-bar {
-            width: 100%;
-            height: 10px;
-            background: rgba(255, 255, 255, 0.1);
-            border-radius: 5px;
-            overflow: hidden;
-            margin-top: 1rem;
+            width: 100%; height: 10px; background: rgba(255, 255, 255, 0.1); border-radius: 5px; overflow: hidden; margin-top: 1rem;
         }
-
         .confidence-fill {
-            height: 100%;
-            background: linear-gradient(90deg, #00d4ff, #00ff88);
-            border-radius: 5px;
-            transition: width 1s ease-out;
-            animation: pulse 2s ease-in-out infinite;
+            height: 100%; background: linear-gradient(90deg, #00d4ff, #00ff88); border-radius: 5px; transition: width 1s ease-out; animation: pulse 2s ease-in-out infinite;
         }
+        /* --- End Result Area Styles --- */
 
         .error {
-            color: #ff4444;
-            background: rgba(255, 68, 68, 0.1);
-            padding: 1rem;
-            border-radius: 10px;
-            border: 1px solid #ff4444;
-            margin-top: 1rem;
-        }
-
-        @keyframes slideDown {
-            from { opacity: 0; transform: translateY(-50px); }
-            to { opacity: 1; transform: translateY(0); }
-        }
-
-        @keyframes slideUp {
-            from { opacity: 0; transform: translateY(50px); }
-            to { opacity: 1; transform: translateY(0); }
-        }
-
-        @keyframes fadeIn {
-            from { opacity: 0; }
-            to { opacity: 1; }
-        }
-
-        @keyframes spin {
-            0% { transform: rotate(0deg); }
-            100% { transform: rotate(360deg); }
-        }
-
-        @keyframes gradientShift {
-            0%, 100% { background-position: 0% 50%; }
-            50% { background-position: 100% 50%; }
-        }
-
-        @keyframes pulse {
-            0%, 100% { opacity: 1; }
-            50% { opacity: 0.7; }
-        }
-
-        /* Responsive design */
+            color: #ff4444; background: rgba(255, 68, 68, 0.1); padding: 1rem; border-radius: 10px; border: 1px solid #ff4444; margin-top: 1rem;
+        }
+        @keyframes slideDown { from { opacity: 0; transform: translateY(-50px); } to { opacity: 1; transform: translateY(0); } }
+        @keyframes slideUp { from { opacity: 0; transform: translateY(50px); } to { opacity: 1; transform: translateY(0); } }
+        @keyframes fadeIn { from { opacity: 0; } to { opacity: 1; } }
+        @keyframes spin { 0% { transform: rotate(0deg); } 100% { transform: rotate(360deg); } }
+        @keyframes gradientShift { 0%, 100% { background-position: 0% 50%; } 50% { background-position: 100% 50%; } }
+        @keyframes pulse { 0%, 100% { opacity: 1; } 50% { opacity: 0.7; } }
         @media (max-width: 768px) {
-            .title {
-                font-size: 2.5rem;
-            }
-            
-            .container {
-                padding: 1rem;
-            }
-            
-            .upload-area {
-                min-height: 250px;
-            }
+            .title { font-size: 2.5rem; } .container { padding: 1rem; } .upload-area { min-height: 250px; } .tabs { flex-wrap: wrap; } .tab-button { margin-bottom: 10px;}
         }
     </style>
 </head>
@@ -302,10 +172,17 @@
     
     <div class="container">
         <div class="header">
-            <h1 class="title">YOLO12 Vision AI</h1>
+            <h1 class="title">YOLO Vision AI</h1>
             <p class="subtitle">Advanced Image Classification with Neural Networks</p>
         </div>
 
+        <!-- NEW: Tabs for model selection -->
+        <div class="tabs">
+            <button class="tab-button active" data-mode="model1">Model 1</button>
+            <button class="tab-button" data-mode="model2">Model 2 (Tyre/Alloy)</button>
+            <button class="tab-button" data-mode="combined">Combined</button>
+        </div>
+
         <div class="upload-area" id="uploadArea">
             <div class="upload-icon">🔮</div>
             <div class="upload-text">Drop your image here or click to upload</div>
@@ -323,13 +200,30 @@
             <p style="color: #00d4ff; margin-top: 1rem;">Processing your image...</p>
         </div>
 
-        <div class="result-container" id="resultContainer" style="display: none;">
-            <div class="result-class" id="resultClass"></div>
-            <div class="result-confidence">
-                Confidence: <span id="confidencePercentage"></span>
+        <!-- NEW: Wrapper for all result blocks -->
+        <div class="results-wrapper" id="resultsWrapper" style="display: none;">
+            <!-- Block for Model 1 / Single Result -->
+            <div class="result-container" id="resultBlock1" style="display: none;">
+                <h3 class="result-title" id="resultTitle1">Prediction</h3>
+                <div class="result-class" id="resultClass1"></div>
+                <div class="result-confidence">
+                    Confidence: <span id="confidencePercentage1"></span>
+                </div>
+                <div class="confidence-bar">
+                    <div class="confidence-fill" id="confidenceFill1"></div>
+                </div>
             </div>
-            <div class="confidence-bar">
-                <div class="confidence-fill" id="confidenceFill"></div>
+            
+            <!-- Block for Model 2 (in combined mode) -->
+            <div class="result-container" id="resultBlock2" style="display: none; margin-top: 1.5rem;">
+                <h3 class="result-title">Model 2 Prediction</h3>
+                <div class="result-class" id="resultClass2"></div>
+                <div class="result-confidence">
+                    Confidence: <span id="confidencePercentage2"></span>
+                </div>
+                <div class="confidence-bar">
+                    <div class="confidence-fill" id="confidenceFill2"></div>
+                </div>
             </div>
         </div>
 
@@ -339,21 +233,17 @@
     <script>
         // Create animated particles
         function createParticles() {
-            const particlesContainer = document.getElementById('particles');
-            const particleCount = 50;
-
-            for (let i = 0; i < particleCount; i++) {
+            const container = document.getElementById('particles');
+            for (let i = 0; i < 50; i++) {
                 const particle = document.createElement('div');
                 particle.className = 'particle';
                 particle.style.left = Math.random() * 100 + '%';
                 particle.style.top = Math.random() * 100 + '%';
                 particle.style.animationDelay = Math.random() * 6 + 's';
                 particle.style.animationDuration = (3 + Math.random() * 3) + 's';
-                particlesContainer.appendChild(particle);
+                container.appendChild(particle);
             }
         }
-
-        // Initialize particles
         createParticles();
 
         // DOM elements
@@ -363,74 +253,75 @@
         const previewImage = document.getElementById('previewImage');
         const predictButton = document.getElementById('predictButton');
         const loading = document.getElementById('loading');
-        const resultContainer = document.getElementById('resultContainer');
         const errorContainer = document.getElementById('errorContainer');
-        const resultClass = document.getElementById('resultClass');
-        const confidencePercentage = document.getElementById('confidencePercentage');
-        const confidenceFill = document.getElementById('confidenceFill');
+        const resultsWrapper = document.getElementById('resultsWrapper');
+
+        // Result Block 1 Elements
+        const resultBlock1 = document.getElementById('resultBlock1');
+        const resultTitle1 = document.getElementById('resultTitle1');
+        const resultClass1 = document.getElementById('resultClass1');
+        const confidencePercentage1 = document.getElementById('confidencePercentage1');
+        const confidenceFill1 = document.getElementById('confidenceFill1');
+        
+        // Result Block 2 Elements
+        const resultBlock2 = document.getElementById('resultBlock2');
+        const resultClass2 = document.getElementById('resultClass2');
+        const confidencePercentage2 = document.getElementById('confidencePercentage2');
+        const confidenceFill2 = document.getElementById('confidenceFill2');
 
         let selectedFile = null;
-
-        // Upload area click handler
-        uploadArea.addEventListener('click', () => {
-            fileInput.click();
+        let currentMode = 'model1'; // NEW: state for current model
+
+        // --- NEW: Tab switching logic ---
+        document.querySelectorAll('.tab-button').forEach(button => {
+            button.addEventListener('click', (e) => {
+                // Update active button
+                document.querySelector('.tab-button.active').classList.remove('active');
+                e.target.classList.add('active');
+                
+                // Update mode
+                currentMode = e.target.dataset.mode;
+
+                // Reset UI
+                resetUI();
+            });
         });
 
-        // File input change handler
+        // Event Listeners
+        uploadArea.addEventListener('click', () => fileInput.click());
         fileInput.addEventListener('change', handleFileSelect);
-
-        // Drag and drop handlers
-        uploadArea.addEventListener('dragover', (e) => {
-            e.preventDefault();
-            uploadArea.classList.add('dragover');
-        });
-
-        uploadArea.addEventListener('dragleave', () => {
-            uploadArea.classList.remove('dragover');
-        });
-
+        uploadArea.addEventListener('dragover', (e) => { e.preventDefault(); uploadArea.classList.add('dragover'); });
+        uploadArea.addEventListener('dragleave', () => uploadArea.classList.remove('dragover'));
         uploadArea.addEventListener('drop', (e) => {
             e.preventDefault();
             uploadArea.classList.remove('dragover');
             const files = e.dataTransfer.files;
-            if (files.length > 0) {
-                handleFile(files[0]);
-            }
+            if (files.length > 0) handleFile(files[0]);
         });
-
-        // Predict button handler
         predictButton.addEventListener('click', predictImage);
 
         function handleFileSelect(e) {
-            const file = e.target.files[0];
-            if (file) {
-                handleFile(file);
-            }
+            if (e.target.files.length > 0) handleFile(e.target.files[0]);
         }
 
         function handleFile(file) {
-            // Validate file type
             const allowedTypes = ['image/png', 'image/jpeg', 'image/jpg'];
             if (!allowedTypes.includes(file.type)) {
                 showError('Please select a valid image file (PNG, JPG, JPEG)');
                 return;
             }
-
-            // Validate file size (max 10MB)
             if (file.size > 10 * 1024 * 1024) {
                 showError('File size too large. Please select a file smaller than 10MB.');
                 return;
             }
 
             selectedFile = file;
-            
-            // Show preview
             const reader = new FileReader();
             reader.onload = (e) => {
                 previewImage.src = e.target.result;
                 previewContainer.style.display = 'block';
                 hideError();
-                hideResult();
+                hideResults();
             };
             reader.readAsDataURL(file);
         }
@@ -441,21 +332,17 @@
                 return;
             }
 
-            // Show loading
             loading.style.display = 'block';
             predictButton.disabled = true;
             hideError();
-            hideResult();
+            hideResults();
 
             try {
                 const formData = new FormData();
                 formData.append('file', selectedFile);
+                formData.append('model_type', currentMode); // NEW: Send current mode to backend
 
-                const response = await fetch('/predict', {
-                    method: 'POST',
-                    body: formData
-                });
-
+                const response = await fetch('/predict', { method: 'POST', body: formData });
                 const data = await response.json();
 
                 if (response.ok) {
@@ -473,16 +360,37 @@
         }
 
         function showResult(data) {
-            resultClass.textContent = data.class;
+            resultsWrapper.style.display = 'block';
+
+            if (currentMode === 'combined') {
+                // Display two results
+                if (data.model1_result) {
+                    displaySingleResult(data.model1_result, 'Model 1 Prediction', resultBlock1, resultTitle1, resultClass1, confidencePercentage1, confidenceFill1);
+                }
+                if (data.model2_result) {
+                    displaySingleResult(data.model2_result, 'Model 2 Prediction', resultBlock2, null, resultClass2, confidencePercentage2, confidenceFill2);
+                }
+            } else {
+                // Display single result
+                const title = currentMode === 'model1' ? 'Model 1 Prediction' : 'Model 2 Prediction';
+                displaySingleResult(data, title, resultBlock1, resultTitle1, resultClass1, confidencePercentage1, confidenceFill1);
+            }
+        }
+        
+        // NEW: Helper function to populate a result block
+        function displaySingleResult(data, title, block, titleEl, classEl, percentEl, fillEl) {
+            if (!data) return;
+            if (titleEl) titleEl.textContent = title;
+            classEl.textContent = data.class;
             const confidence = Math.round(data.confidence * 100);
-            confidencePercentage.textContent = confidence + '%';
-            confidenceFill.style.width = confidence + '%';
+            percentEl.textContent = confidence + '%';
             
-            resultContainer.style.display = 'block';
+            block.style.display = 'block';
             
-            // Add some visual flair
+            // Animate confidence bar
+            fillEl.style.width = '0%';
             setTimeout(() => {
-                confidenceFill.style.width = confidence + '%';
+                fillEl.style.width = confidence + '%';
             }, 100);
         }
 
@@ -495,23 +403,21 @@
             errorContainer.style.display = 'none';
         }
 
-        function hideResult() {
-            resultContainer.style.display = 'none';
+        function hideResults() {
+            resultsWrapper.style.display = 'none';
+            resultBlock1.style.display = 'none';
+            resultBlock2.style.display = 'none';
+        }
+        
+        // NEW: Function to reset the entire UI state
+        function resetUI() {
+            hideResults();
+            hideError();
+            loading.style.display = 'none';
+            previewContainer.style.display = 'none';
+            fileInput.value = ''; // Clear file input
+            selectedFile = null;
         }
-
-        // Add some interactive effects
-        document.addEventListener('mousemove', (e) => {
-            const particles = document.querySelectorAll('.particle');
-            const x = e.clientX / window.innerWidth;
-            const y = e.clientY / window.innerHeight;
-            
-            particles.forEach((particle, index) => {
-                const speed = (index % 3 + 1) * 0.5;
-                const newX = x * speed;
-                const newY = y * speed;
-                particle.style.transform = `translate(${newX}px, ${newY}px)`;
-            });
-        });
     </script>
 </body>
 </html>