Update app.py

Signed-off-by: PROJECT ZERO <[email protected]>

Author: ProjectZeroDays

app.py

@@ -72,19 +72,24 @@
 }
 
 # Configure logging
-logging.basicConfig(level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+
 
 async def random_url(_):
-    try:
-        pet = random.choice(["cat", "dog"])
-        api_url = f"https://api.the{pet}api.com/v1/images/search"
-        async with aiohttp.ClientSession() as session:
-            async with session.get(api_url) as resp:
-                resp.raise_for_status()
-                return (await resp.json())[0]["url"]
-    except aiohttp.ClientError as e:
-        logging.error(f"API request failed: {e}")
-        return None
+    retries = 3
+    for _ in range(retries):
+        try:
+            pet = random.choice(["cat", "dog"])
+            api_url = f"https://api.the{pet}api.com/v1/images/search"
+            async with aiohttp.ClientSession() as session:
+                async with session.get(api_url) as resp:
+                    resp.raise_for_status()
+                    return (await resp.json())[0]["url"]
+        except aiohttp.ClientError as e:
+            logging.error(f"API request failed: {e}")
+        except Exception as e:
+            logging.error(f"Unexpected error: {e}")
+    return None
 
 
 @pn.cache
@@ -97,14 +102,18 @@ def load_processor_model(
 
 
 async def open_image_url(image_url: str) -> Image:
-    try:
-        async with aiohttp.ClientSession() as session:
-            async with session.get(image_url) as resp:
-                resp.raise_for_status()
-                return Image.open(io.BytesIO(await resp.read()))
-    except aiohttp.ClientError as e:
-        logging.error(f"HTTP request failed: {e}")
-        return None
+    retries = 3
+    for _ in range(retries):
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.get(image_url) as resp:
+                    resp.raise_for_status()
+                    return Image.open(io.BytesIO(await resp.read()))
+        except aiohttp.ClientError as e:
+            logging.error(f"HTTP request failed: {e}")
+        except Exception as e:
+            logging.error(f"Unexpected error: {e}")
+    return None
 
 
 def get_similarity_scores(class_items: List[str], image: Image) -> List[float]: