Update app.py

app.py

@@ -241,17 +241,21 @@ def search(search_query):
 
     # Iterate over the top 5 results
     for id in best_photo_ids:
-        print(id)
         id, _ = id.split('.')
         url = id2url.get(id, "")
-        print(url)
         if url == "": continue
-
-        r = requests.get(url + "?w=512", stream=True)
-        img = Image.open(r.raw)
-
+        try:
+            r = requests.get(url + "?w=512", stream=True, timeout=10)
+            r.raise_for_status()  # Raise exception for bad status codes
+            r.raw.decode_content = True  # Handle compressed responses
+            img = Image.open(r.raw)
+            imgs.append(img.copy())  # Copy before closing stream
+            r.close()  # Clean up connection
+        except Exception as e:
+            print(f"Failed to load {url}: {e}")
+            continue
+        
         #credits = f'Photo by <a href="https://unsplash.com/@{photo["photographer_username"]}?utm_source=NaturalLanguageImageSearch&utm_medium=referral">{photo["photographer_first_name"]} {photo["photographer_last_name"]}</a> on <a href="https://unsplash.com/?utm_source=NaturalLanguageImageSearch&utm_medium=referral">Unsplash</a>'
-        imgs.append(img)
         #display(HTML(f'Photo by <a href="https://unsplash.com/@{photo["photographer_username"]}?utm_source=NaturalLanguageImageSearch&utm_medium=referral">{photo["photographer_first_name"]} {photo["photographer_last_name"]}</a> on <a href="https://unsplash.com/?utm_source=NaturalLanguageImageSearch&utm_medium=referral">Unsplash</a>'))
 
         if len(imgs) == 5: break