Update app.py

app.py

@@ -5,15 +5,23 @@ from transformers import pipeline
 # Load the NLP model from Hugging Face
 nlp_model = pipeline('text-classification', model='distilbert-base-uncased')
 
-# Enhanced tree dataset with trees found in Pakistan
+# Enhanced tree dataset with more trees, climates, and soil types
 data = {
-    'Tree': ['Neem', 'Peepal', 'Mango', 'Bamboo', 'Chili', 'Guava', 'Coconut', 'Papaya'],
-    'Climate': ['Tropical', 'Tropical', 'Tropical', 'Tropical', 'Tropical', 'Tropical', 'Tropical', 'Tropical'],
-    'Soil': ['Loamy', 'Sandy', 'Loamy', 'Loamy', 'Sandy', 'Loamy', 'Sandy', 'Loamy'],
-    'Use': ['Shade', 'Shade', 'Fruit', 'Construction', 'Spice', 'Fruit', 'Fruit', 'Fruit'],
-    'Height': ['Medium', 'Medium', 'Tall', 'Tall', 'Short', 'Medium', 'Tall', 'Medium'],
-    'Leaf Color': ['Green', 'Green', 'Green', 'Green', 'Green', 'Green', 'Green', 'Green'],
-    'Growth Rate': ['Fast', 'Fast', 'Fast', 'Fast', 'Medium', 'Fast', 'Fast', 'Fast']
+    'Tree': ['Neem', 'Peepal', 'Mango', 'Bamboo', 'Chili', 'Guava', 'Coconut', 'Papaya', 
+             'Pine', 'Oak', 'Acacia', 'Eucalyptus', 'Teak', 'Date Palm', 'Banana', 'Apple'],
+    'Climate': ['Tropical', 'Tropical', 'Tropical', 'Tropical', 'Tropical', 'Tropical', 
+                'Tropical', 'Tropical', 'Temperate', 'Temperate', 'Arid', 'Temperate', 
+                'Tropical', 'Arid', 'Tropical', 'Temperate'],
+    'Soil': ['Loamy', 'Sandy', 'Loamy', 'Loamy', 'Sandy', 'Loamy', 'Sandy', 'Loamy', 
+             'Loamy', 'Clay', 'Sandy', 'Sandy', 'Loamy', 'Sandy', 'Loamy', 'Loamy'],
+    'Use': ['Shade', 'Shade', 'Fruit', 'Construction', 'Spice', 'Fruit', 'Fruit', 'Fruit', 
+            'Timber', 'Timber', 'Shade', 'Timber', 'Timber', 'Fruit', 'Fruit', 'Fruit'],
+    'Height': ['Medium', 'Medium', 'Tall', 'Tall', 'Short', 'Medium', 'Tall', 'Medium', 
+               'Tall', 'Tall', 'Medium', 'Tall', 'Tall', 'Tall', 'Medium', 'Medium'],
+    'Leaf Color': ['Green', 'Green', 'Green', 'Green', 'Green', 'Green', 'Green', 'Green', 
+                   'Green', 'Green', 'Green', 'Green', 'Green', 'Green', 'Green', 'Green'],
+    'Growth Rate': ['Fast', 'Fast', 'Fast', 'Fast', 'Medium', 'Fast', 'Fast', 'Fast', 
+                    'Slow', 'Slow', 'Medium', 'Fast', 'Medium', 'Slow', 'Fast', 'Medium']
 }
 
 # Convert to DataFrame
@@ -76,7 +84,15 @@ tree_images = {
     'Chili': 'https://upload.wikimedia.org/wikipedia/commons/f/fd/Chili_pepper_1.jpg',
     'Guava': 'https://upload.wikimedia.org/wikipedia/commons/7/7e/Guava_Fruit.jpg',
     'Coconut': 'https://upload.wikimedia.org/wikipedia/commons/7/7b/Coconut_%28fruit%29.jpg',
-    'Papaya': 'https://upload.wikimedia.org/wikipedia/commons/4/4e/Papaya_fruit.jpg'
+    'Papaya': 'https://upload.wikimedia.org/wikipedia/commons/4/4e/Papaya_fruit.jpg',
+    'Pine': 'https://upload.wikimedia.org/wikipedia/commons/5/55/Pine_forest.JPG',
+    'Oak': 'https://upload.wikimedia.org/wikipedia/commons/5/58/Oak_tree.jpg',
+    'Acacia': 'https://upload.wikimedia.org/wikipedia/commons/8/8d/Acacia_tree.jpg',
+    'Eucalyptus': 'https://upload.wikimedia.org/wikipedia/commons/3/38/Eucalyptus_tree.JPG',
+    'Teak': 'https://upload.wikimedia.org/wikipedia/commons/1/1a/Teak_Tree.jpg',
+    'Date Palm': 'https://upload.wikimedia.org/wikipedia/commons/a/ab/Date_Palm.JPG',
+    'Banana': 'https://upload.wikimedia.org/wikipedia/commons/4/42/Banana_Tree.JPG',
+    'Apple': 'https://upload.wikimedia.org/wikipedia/commons/1/15/Apple_Tree.jpg'
 }
 
 # Dynamic sidebar content