Only display those images, which were actually sent to the VLMs

app.py

@@ -126,9 +126,10 @@ def try_scaled_image_url(client: httpx.Client, url: str, marketplace: str, max_r
     
     return url
 
-def download_and_encode_images(image_urls: List[str], marketplace: str) -> List[Dict]:
+def download_and_encode_images(image_urls: List[str], marketplace: str) -> Tuple[List[Dict], List[str]]:
     """Download images and convert them to base64 format for Gemini."""
     encoded_images = []
+    successful_urls = []  # Track URLs that were successfully downloaded
     timeout = httpx.Timeout(10.0, connect=5.0)
     with httpx.Client(timeout=timeout) as client:
         for url in image_urls:
@@ -144,6 +145,7 @@ def download_and_encode_images(image_urls: List[str], marketplace: str) -> List[
                         'mime_type': 'image/jpeg',  # Assuming JPEG format
                         'data': encoded_image
                     })
+                    successful_urls.append(final_url)  # Store the URL that worked (original or scaled)
                     break  # Success, exit retry loop
                 except httpx.TimeoutException:
                     print(f"Timeout downloading image (attempt {attempt + 1}/{max_retries}): {url}")
@@ -153,7 +155,7 @@ def download_and_encode_images(image_urls: List[str], marketplace: str) -> List[
                     print(f"Error downloading image: {type(e).__name__}: {str(e)}")
                     if attempt == max_retries - 1:
                         print(f"Max retries reached, skipping image: {url}")
-    return encoded_images
+    return encoded_images, successful_urls
 
 def get_gemini_response(model_name: str, encoded_images: List[Dict], prompt: str) -> str:
     """Get response from a Gemini model."""
@@ -197,12 +199,15 @@ def process_input(id_product_money: str, prompt: str, progress=gr.Progress()) ->
         
         status_msg = "Downloading and processing images..."
         progress(0.4, desc=status_msg)
-        encoded_images = download_and_encode_images(all_image_urls, marketplace)
+        encoded_images, successful_urls = download_and_encode_images(all_image_urls, marketplace)
         print(f"Number of encoded images: {len(encoded_images)}")
         
         if not encoded_images:
             raise ValueError("No images could be downloaded")
         
+        # Update all_image_urls to only include successfully downloaded URLs
+        all_image_urls = successful_urls
+        
         status_msg = "Getting response from Gemini 1.5 Flash..."
         progress(0.6, desc=status_msg)
         gemini_1_5_response = get_gemini_response("gemini-1.5-flash", encoded_images, prompt)