|
|
import os |
|
|
import requests |
|
|
from PIL import Image |
|
|
from io import BytesIO |
|
|
|
|
|
def download_example_images(): |
|
|
""" |
|
|
Downloads example images for the application. |
|
|
""" |
|
|
|
|
|
os.makedirs('example_images', exist_ok=True) |
|
|
|
|
|
|
|
|
example_images = { |
|
|
'tomato_early_blight.jpg': 'https://www.gardeningknowhow.com/wp-content/uploads/2019/05/tomato-early-blight.jpg', |
|
|
'apple_scab.jpg': 'https://www.planetnatural.com/wp-content/uploads/2012/12/apple-scab-1.jpg', |
|
|
'corn_rust.jpg': 'https://extension.umn.edu/sites/extension.umn.edu/files/styles/large/public/corn-rust-JosieMontgomery.jpg' |
|
|
} |
|
|
|
|
|
|
|
|
for filename, url in example_images.items(): |
|
|
try: |
|
|
response = requests.get(url) |
|
|
if response.status_code == 200: |
|
|
img = Image.open(BytesIO(response.content)) |
|
|
img.save(os.path.join('example_images', filename)) |
|
|
print(f"Downloaded {filename}") |
|
|
else: |
|
|
print(f"Failed to download {filename}: HTTP {response.status_code}") |
|
|
except Exception as e: |
|
|
print(f"Error downloading {filename}: {e}") |
|
|
|
|
|
print("Example images download complete") |
|
|
|
|
|
if __name__ == "__main__": |
|
|
download_example_images() |
|
|
|