Update image_fetcher.py

image_fetcher.py

@@ -10,6 +10,7 @@ import time
 # Setup /tmp directory for Hugging Face
 IMAGE_DIR = "/tmp/images"
 os.makedirs(IMAGE_DIR, exist_ok=True)
+image_nlst=[]
 
 # Headers
 def get_headers():
@@ -55,6 +56,7 @@ async def search_and_download(session, prompt, sem):
                 url = item.get("image")
                 result = await fetch_image(session, url, name)
                 if "Saved" in result:
+                    image_nlst.append(result.lstrip("Saved:"))
                     return result
             return f"No valid image for: {prompt}"
         except Exception as e:
@@ -68,4 +70,5 @@ async def main(prompts):
         results = await asyncio.gather(*tasks)
         for res in results:
             print(res)
+    return image_nlst