Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI, File, UploadFile, Request
 from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse
 import requests
 import time
+import asyncio
 from typing import Dict
 
 app = FastAPI()
@@ -462,7 +463,7 @@ HTML_CONTENT = """
             }
         }
 
-        function uploadFile(file) {
+        async function uploadFile(file) {
             progressContainer.innerHTML = '';
             progressContainer.style.display = 'block';
             loadingSpinner.style.display = 'block';
@@ -476,27 +477,41 @@ HTML_CONTENT = """
             const formData = new FormData();
             formData.append('file', file);
 
-            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);
+            let retryCount = 0;
+            const maxRetries = 5;
+            const retryDelay = 1000; // 1 second
+
+            while (retryCount < maxRetries) {
+                try {
+                    const response = await fetch('/upload', {
+                        method: 'POST',
+                        body: formData
+                    });
+
+                    if (!response.ok) {
+                        throw new Error(`HTTP error! status: ${response.status}`);
+                    }
+
+                    const result = await response.json();
+                    if (result.url) {
+                        addResultLink(result.url, file.name);
+                    } else {
+                        alert('Upload failed: ' + result.error);
+                    }
+                    break; // Success, exit the loop
+                } catch (error) {
+                    console.error('Upload error:', error);
+                    retryCount++;
+                    if (retryCount < maxRetries) {
+                        alert(`Network error. Retrying in ${retryDelay / 1000} seconds... (Attempt ${retryCount} of ${maxRetries})`);
+                        await new Promise(resolve => setTimeout(resolve, retryDelay));
                     } else {
-                        alert('Upload failed: ' + response.error);
+                        alert('Upload failed after multiple attempts. Please check your internet connection and try again.');
                     }
-                } else {
-                    alert('Upload failed: ' + xhr.statusText);
                 }
-                resetUploadState();
-            };
-            xhr.onerror = function() {
-                alert('Upload failed: Network error');
-                resetUploadState();
-            };
-            xhr.send(formData);
+            }
+
+            resetUploadState();
         }
 
         function createProgressBar(fileName) {
@@ -718,10 +733,17 @@ async def upload_file(upload_url: str, file_content: bytes, content_type: str) -
 
 async def retry_upload(upload_url: str, file_content: bytes, content_type: str, max_retries: int = 5, delay: int = 1) -> bool:
     for attempt in range(1, max_retries + 1):
-        success = await upload_file(upload_url, file_content, content_type)
-        if success:
-            return True
+        try:
+            success = await upload_file(upload_url, file_content, content_type)
+            if success:
+                return True
+            print(f"Upload attempt {attempt} failed. Retrying...")
+        except Exception as e:
+            print(f"Error during upload attempt {attempt}: {e}")
+        
         if attempt < max_retries:
-            time.sleep(delay)
+            await asyncio.sleep(delay)
             delay *= 2  # Exponential backoff
+    
+    print("Upload failed after all retry attempts")
     return False