Update app.py

app.py

@@ -6,6 +6,7 @@ import asyncio
 from typing import Dict
 import os
 import mimetypes
+import uuid
 
 app = FastAPI()
 
@@ -634,7 +635,7 @@ HTML_CONTENT = """
             for (let i = 0; i < items.length; i++) {
                 if (items[i].kind === 'file') {
                     const file = items[i].getAsFile();
-                    const newFileName = generateUniqueFileName(file.name, pastedFiles);
+                    const newFileName = `pasted_image_${Date.now()}.png`;
                     const renamedFile = new File([file], newFileName, { type: file.type });
                     pastedFiles.push(renamedFile);
                 }
@@ -645,20 +646,6 @@ HTML_CONTENT = """
             }
         });
 
-        function generateUniqueFileName(originalName, fileList) {
-            const extension = originalName.split('.').pop();
-            const baseName = originalName.split('.').slice(0, -1).join('.');
-            let counter = 1;
-            let newName = originalName;
-
-            while (fileList.some(file => file.name === newName)) {
-                newName = `${baseName}${counter}.${extension}`;
-                counter++;
-            }
-
-            return newName;
-        }
-
         function updateFileNameDisplay() {
             const fileNames = pastedFiles.map(file => file.name).join(', ');
             fileName.textContent = fileNames;
@@ -723,42 +710,26 @@ HTML_CONTENT = """
                 const formData = new FormData();
                 formData.append('file', file);
 
-                while (true) {
-                    try {
-                        const xhr = new XMLHttpRequest();
-                        xhr.open('POST', '/upload', true);
-                        xhr.upload.onprogress = (event) => updateProgress(event, progressBar.querySelector('.progress'));
-                        
-                        xhr.onload = function() {
-                            if (xhr.status === 200) {
-                                const response = JSON.parse(xhr.responseText);
-                                if (response.url) {
-                                    addResultLink(response.url, file.name, response.originalExtension);
-                                    saveToHistory(file.name, response.url, response.originalExtension);
-                                } else {
-                                    throw new Error('Upload failed: ' + response.error);
-                                }
-                            } else {
-                                throw new Error(`HTTP error! status: ${xhr.status}`);
-                            }
-                        };
-
-                        xhr.onerror = function() {
-                            throw new Error('Network error occurred');
-                        };
-
-                        xhr.send(formData);
-
-                        await new Promise((resolve, reject) => {
-                            xhr.onloadend = resolve;
-                            xhr.onerror = reject;
-                        });
-
-                        break;
-                    } catch (error) {
-                        console.error('Upload error:', error);
-                        await new Promise(resolve => setTimeout(resolve, 1000));
+                try {
+                    const response = await fetch('/upload', {
+                        method: 'POST',
+                        body: formData
+                    });
+
+                    if (response.ok) {
+                        const result = await response.json();
+                        if (result.url) {
+                            addResultLink(result.url, file.name, result.originalExtension);
+                            saveToHistory(file.name, result.url, result.originalExtension);
+                        } else {
+                            throw new Error('Upload failed: ' + result.error);
+                        }
+                    } else {
+                        throw new Error(`HTTP error! status: ${response.status}`);
                     }
+                } catch (error) {
+                    console.error('Upload error:', error);
+                    alert(`Failed to upload ${file.name}: ${error.message}`);
                 }
             }
 
@@ -781,13 +752,6 @@ HTML_CONTENT = """
             return container;
         }
 
-        function updateProgress(event, progressBar) {
-            if (event.lengthComputable) {
-                const percentComplete = (event.loaded / event.total) * 100;
-                progressBar.style.width = percentComplete + '%';
-            }
-        }
-
         function resetUploadState() {
             fileInput.value = '';
             fileName.textContent = '';
@@ -873,7 +837,7 @@ HTML_CONTENT = """
                 const historyItem = document.createElement('div');
                 historyItem.className = 'history-item';
                 
-                                const itemName = document.createElement('span');
+                const itemName = document.createElement('span');
                 itemName.className = 'history-item-name';
                 itemName.textContent = item.fileName;
                 historyItem.appendChild(itemName);
@@ -899,7 +863,7 @@ HTML_CONTENT = """
                 };
                 actionsContainer.appendChild(openBtn);
 
-                if (item.originalExtension.toLowerCase() === 'mp4') {
+                                if (item.originalExtension.toLowerCase() === 'mp4') {
                     const embedBtn = document.createElement('button');
                     embedBtn.textContent = 'Embed';
                     embedBtn.className = 'small-btn';
@@ -956,7 +920,7 @@ HTML_CONTENT = """
                 quickOpenContent.appendChild(link);
             }
 
-            quickOpenModal.style.display = "block";
+                        quickOpenModal.style.display = "block";
         }
     </script>
 </body>