diff --git a/static/script.js b/static/script.js
index c104f95..76325ea 100644
--- a/static/script.js
+++ b/static/script.js
@@ -315,7 +315,19 @@ async function testHardcodedImage() {
function displayResults(result, title) {
const resultsSection = document.getElementById('resultsSection');
const resultsContent = document.getElementById('resultsContent');
-
+
+ // Create smart summary message
+ let summaryMessage = '';
+ if (result.success) {
+ if (result.num_detections > 0) {
+ summaryMessage = `
✅ Found ${result.num_detections} memory modules
`;
+ } else {
+ summaryMessage = `❌ No memory modules
`;
+ }
+ } else {
+ summaryMessage = `❌ Error: ${result.error}
`;
+ }
+
let detectionsHtml = '';
if (result.detections && result.detections.length > 0) {
detectionsHtml = result.detections.map((detection, index) => `
@@ -327,14 +339,16 @@ function displayResults(result, title) {
} else {
detectionsHtml = 'No memory modules detected
';
}
-
+
resultsContent.innerHTML = `
-
+
+ ${summaryMessage}
+
${result.num_detections}
@@ -349,22 +363,22 @@ function displayResults(result, title) {
Status
-
+
Detected Memory Modules:
${detectionsHtml}
-
+
${result.annotated_image ? `
Annotated Image:
-
` : ''}
`;
-
+
resultsSection.style.display = 'block';
resultsSection.scrollIntoView({ behavior: 'smooth' });
}
diff --git a/static/style.css b/static/style.css
index 397f46d..7ff40b5 100644
--- a/static/style.css
+++ b/static/style.css
@@ -347,6 +347,33 @@ main {
flex: 1;
}
+.summary-message {
+ padding: 15px;
+ margin: 15px 0;
+ border-radius: 8px;
+ font-weight: 600;
+ font-size: 1.1rem;
+ text-align: center;
+}
+
+.summary-message.success {
+ background: #d4edda;
+ color: #155724;
+ border: 1px solid #c3e6cb;
+}
+
+.summary-message.no-memory {
+ background: #f8d7da;
+ color: #721c24;
+ border: 1px solid #f5c6cb;
+}
+
+.summary-message.error {
+ background: #f8d7da;
+ color: #721c24;
+ border: 1px solid #f5c6cb;
+}
+
.footer {
text-align: center;
margin-top: auto;