Update static/js/script.js

static/js/script.js

@@ -4,11 +4,16 @@ $(document).ready(function() {
     const capturedImage = document.getElementById('capturedImage');
     const hfResult = document.getElementById('hfResult');
     const cameraSelect = document.getElementById('cameraSelect');
+    const actionButtons = document.getElementById('actionButtons');
     let stream = null;
+    let currentImageUrl = null;
 
     // Open camera
     $('#openCamera').click(function() {
         $('#cameraContainer').show();
+        actionButtons.style.display = 'none';
+        capturedImage.style.display = 'none';
+        hfResult.textContent = '';
         startCamera(cameraSelect.value);
     });
 
@@ -37,6 +42,8 @@ $(document).ready(function() {
                     capturedImage.src = response.image_url;
                     capturedImage.style.display = 'block';
                     hfResult.textContent = JSON.stringify(response.hf_result, null, 2);
+                    actionButtons.style.display = 'block';
+                    currentImageUrl = response.image_url;
                     stopCamera();
                     $('#cameraContainer').hide();
                 } else {
@@ -49,6 +56,41 @@ $(document).ready(function() {
         });
     });
 
+    // Retake button
+    $('#retakeButton').click(function() {
+        $('#cameraContainer').show();
+        actionButtons.style.display = 'none';
+        capturedImage.style.display = 'none';
+        hfResult.textContent = '';
+        startCamera(cameraSelect.value);
+    });
+
+    // Upload to Instagram
+    $('#uploadButton').click(function() {
+        if (!currentImageUrl) {
+            alert('No image to upload.');
+            return;
+        }
+        $.ajax({
+            type: 'POST',
+            url: '/upload_instagram',
+            data: { image_url: currentImageUrl },
+            success: function(response) {
+                if (response.status === 'success') {
+                    alert('Image uploaded to Instagram successfully!');
+                    actionButtons.style.display = 'none';
+                    capturedImage.style.display = 'none';
+                    hfResult.textContent = '';
+                } else {
+                    alert('Error uploading to Instagram: ' + response.message);
+                }
+            },
+            error: function() {
+                alert('Failed to upload image to Instagram.');
+            }
+        });
+    });
+
     function startCamera(facingMode) {
         const constraints = {
             video: { facingMode: facingMode }