Update app.py

app.py

@@ -16,11 +16,7 @@ warnings.filterwarnings("ignore", category=FutureWarning)
 
 # Load dataset from Hugging Face
 dataset = load_dataset("hangunwoo07/Naturing_Bird_Data")
-print(dataset)
-
-dataset = dataset['train']['features']  # Access the train split directly
-
-print(dataset)
+dataset = dataset['train']  # Access the train split directly
 
 # Load bird data
 with open('DB/bird_data.json', 'r', encoding='utf-8') as f:
@@ -36,45 +32,16 @@ with open('DB/class_names.json', 'r', encoding='utf-8') as f:
 # Constants
 DSHS_LOCATION = [36.373719, 127.370415]
 
-def load_bird_image_dataset(dataset_path):
-    """
-    Load a bird image dataset stored in a zip file with subfolders for each bird species.
-
-    Args:
-        dataset_path (str): Path to the zip file containing the dataset.
-
-    Returns:
-        dict: A dictionary where the keys are the bird species names and the values are lists of image file paths.
-    """
-    bird_images = {}
-
-    with ZipFile(dataset_path, 'r') as zip_file:
-        for filename in zip_file.namelist():
-            if filename.endswith('.jpg'):
-                # Extract the bird species name from the file path
-                species = filename.split('\\')[1]
-
-                # Initialize the list for the current species if it doesn't exist
-                if species not in bird_images:
-                    bird_images[species] = []
-
-                # Add the file path to the list for the current species
-                bird_images[species].append(filename)
-
-                print(f"filename: {filename}")
-
-    return bird_images
-
 def create_image_popup(bird_name, image_filename):
-    """
-    Create HTML popup content with image from dataset
-    """
-    with ZipFile(IMAGE_DATASET_PATH, 'r') as zip_file:
-        image_bytes = zip_file.read(image_filename)
-    if image_bytes:
-        encoded = base64.b64encode(image_bytes).decode()
+    target_image = '/'.join(bird_name, image_filename.split())
+    row_index = metadata[metadata['image'] == target_image].index[0] - 1
+    
+    try:
+        with open(dataset['image'][row_index], 'rb') as image_file:
+            encoded = base64.b64encode(image_file.read()).decode()
         return f'<img src="data:image/jpeg;base64,{encoded}" width="200px">'
-    return ''
+    except:
+        return ''
 
 def create_map():
     """