Update app.py

app.py

@@ -408,8 +408,8 @@ HTML_CONTENT = """
         const uploadForm = document.getElementById('uploadForm');
         const progressContainer = document.getElementById('progressContainer');
         const loadingSpinner = document.getElementById('loadingSpinner');
-                const resultContainer = document.getElementById('resultContainer');
-        const dropZone = document.getElementById('dropZone');
+        const resultContainer = document.getElementById('resultContainer');
+                const dropZone = document.getElementById('dropZone');
         const modal = document.getElementById('embedModal');
         const span = document.getElementsByClassName("close")[0];
         const embedLinkInput = document.getElementById('embedLink');
@@ -486,41 +486,49 @@ HTML_CONTENT = """
             const formData = new FormData();
             formData.append('file', file);
 
-            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);
-                        } else {
-                            throw new Error('Upload failed: ' + response.error);
-                        }
-                    } else {
-                        throw new Error(`HTTP error! status: ${xhr.status}`);
+            let retries = 0;
+            const maxRetries = 5;
+            const retryDelay = 1000;
+
+            while (retries < maxRetries) {
+                try {
+                    const xhr = new XMLHttpRequest();
+                    xhr.open('POST', '/upload', true);
+                    xhr.upload.onprogress = (event) => updateProgress(event, progressBar.querySelector('.progress'));
+                    
+                    const uploadPromise = new Promise((resolve, reject) => {
+                        xhr.onload = function() {
+                            if (xhr.status === 200) {
+                                const response = JSON.parse(xhr.responseText);
+                                if (response.url) {
+                                    resolve(response.url);
+                                } else {
+                                    reject(new Error('Upload failed: ' + response.error));
+                                }
+                            } else {
+                                reject(new Error(`HTTP error! status: ${xhr.status}`));
+                            }
+                        };
+                        xhr.onerror = () => reject(new Error('Network error occurred'));
+                    });
+
+                    xhr.send(formData);
+
+                    const url = await uploadPromise;
+                    addResultLink(url, file.name);
+                    break; // Success, exit the retry loop
+                } catch (error) {
+                    console.error('Upload error:', error);
+                    retries++;
+                    if (retries >= maxRetries) {
+                        alert('Upload failed after multiple attempts: ' + error.message);
+                        break;
                     }
-                };
-
-                xhr.onerror = function() {
-                    throw new Error('Network error occurred');
-                };
-
-                xhr.send(formData);
-
-                await new Promise((resolve, reject) => {
-                    xhr.onloadend = resolve;
-                    xhr.onerror = reject;
-                });
-
-            } catch (error) {
-                console.error('Upload error:', error);
-                alert('Upload failed: ' + error.message);
-            } finally {
-                resetUploadState();
+                    await new Promise(resolve => setTimeout(resolve, retryDelay));
+                }
             }
+
+            resetUploadState();
         }
 
         function createProgressBar(fileName) {