Update app.py

app.py

@@ -1,3 +1,79 @@
+import gradio as gr
+import matplotlib.pyplot as plt
+import numpy as np
+import os
+import PIL
+import tensorflow as tf
+
+from tensorflow import keras
+from tensorflow.keras import layers
+from tensorflow.keras.models import Sequential
+
+
+from PIL import Image
+import gdown
+import zipfile
+
+import pathlib
+
+
+
+
+
+# Define the Google Drive shareable link
+gdrive_url = 'https://drive.google.com/file/d/1HjHYlQyRz5oWt8kehkt1TiOGRRlKFsv8/view?usp=drive_link'
+
+# Extract the file ID from the URL
+file_id = gdrive_url.split('/d/')[1].split('/view')[0]
+direct_download_url = f'https://drive.google.com/uc?id={file_id}'
+
+# Define the local filename to save the ZIP file
+local_zip_file = 'file.zip'
+
+# Download the ZIP file
+gdown.download(direct_download_url, local_zip_file, quiet=False)
+
+# Directory to extract files
+extracted_path = 'extracted_files'
+
+# Verify if the downloaded file is a ZIP file and extract it
+try:
+    with zipfile.ZipFile(local_zip_file, 'r') as zip_ref:
+        zip_ref.extractall(extracted_path)
+    print("Extraction successful!")
+except zipfile.BadZipFile:
+    print("Error: The downloaded file is not a valid ZIP file.")
+
+# Optionally, you can delete the ZIP file after extraction
+os.remove(local_zip_file)
+
+# Convert the extracted directory path to a pathlib.Path object
+data_dir = pathlib.Path(extracted_path)
+
+# Print the directory structure to debug
+for root, dirs, files in os.walk(extracted_path):
+    level = root.replace(extracted_path, '').count(os.sep)
+    indent = ' ' * 4 * (level)
+    print(f"{indent}{os.path.basename(root)}/")
+    subindent = ' ' * 4 * (level + 1)
+    for f in files:
+        print(f"{subindent}{f}")
+
+import pathlib
+# Path to the dataset directory
+data_dir = pathlib.Path('extracted_files/Pest_Dataset')
+data_dir = pathlib.Path(data_dir)
+
+
+bees = list(data_dir.glob('bees/*'))
+print(bees[0])
+PIL.Image.open(str(bees[0]))
+
+
+bees = list(data_dir.glob('bees/*'))
+print(bees[0])
+PIL.Image.open(str(bees[0]))
+
 batch_size = 32
 img_height = 180
 img_width = 180