Fix specialist string map keys

backend/app.py

@@ -147,44 +147,44 @@ GENERALIST_MAP = {
 
 # Specialist Map (PlantVillage 38 Classes)
 SPECIALIST_MAP = {
-    "Apple___Apple_scab": "Apple - Scab",
-    "Apple___Black_rot": "Apple - Black Rot",
-    "Apple___Cedar_apple_rust": "Apple - Cedar Rust",
-    "Apple___healthy": "Apple - Healthy",
-    "Blueberry___healthy": "Blueberry - Healthy",
-    "Cherry_(including_sour)___Powdery_mildew": "Cherry - Powdery Mildew",
-    "Cherry_(including_sour)___healthy": "Cherry - Healthy",
-    "Corn_(maize)___Cercospora_leaf_spot Gray_leaf_spot": "Corn - Gray Leaf Spot",
-    "Corn_(maize)___Common_rust_": "Corn - Common Rust",
-    "Corn_(maize)___Northern_Leaf_Blight": "Corn - Leaf Blight",
-    "Corn_(maize)___healthy": "Corn - Healthy",
-    "Grape___Black_rot": "Grape - Black Rot",
-    "Grape___Esca_(Black_Measles)": "Grape - Black Measles",
-    "Grape___Leaf_blight_(Isariopsis_Leaf_Spot)": "Grape - Leaf Blight",
-    "Grape___healthy": "Grape - Healthy",
-    "Orange___Haunglongbing_(Citrus_greening)": "Orange - Huanglongbing (Greening)",
-    "Peach___Bacterial_spot": "Peach - Bacterial Spot",
-    "Peach___healthy": "Peach - Healthy",
-    "Pepper,_bell___Bacterial_spot": "Pepper - Bacterial Spot",
-    "Pepper,_bell___healthy": "Pepper - Healthy",
-    "Potato___Early_blight": "Potato - Early Blight",
-    "Potato___Late_blight": "Potato - Late Blight",
-    "Potato___healthy": "Potato - Healthy",
-    "Raspberry___healthy": "Raspberry - Healthy",
-    "Soybean___healthy": "Soybean - Healthy",
-    "Squash___Powdery_mildew": "Squash - Powdery Mildew",
-    "Strawberry___Leaf_scorch": "Strawberry - Leaf Scorch",
-    "Strawberry___healthy": "Strawberry - Healthy",
-    "Tomato___Bacterial_spot": "Tomato - Bacterial Spot",
-    "Tomato___Early_blight": "Tomato - Early Blight",
-    "Tomato___Late_blight": "Tomato - Late Blight",
-    "Tomato___Leaf_Mold": "Tomato - Leaf Mold",
-    "Tomato___Septoria_leaf_spot": "Tomato - Septoria Spot",
-    "Tomato___Spider_mites Two-spotted_spider_mite": "Tomato - Spider Mite",
-    "Tomato___Target_Spot": "Tomato - Target Spot",
-    "Tomato___Tomato_Yellow_Leaf_Curl_Virus": "Tomato - Yellow Leaf Curl",
-    "Tomato___Tomato_mosaic_virus": "Tomato - Mosaic Virus",
-    "Tomato___healthy": "Tomato - Healthy"
+    "Apple Scab": "Apple - Scab",
+    "Apple with Black Rot": "Apple - Black Rot",
+    "Cedar Apple Rust": "Apple - Rust",
+    "Healthy Apple": "Apple - Healthy",
+    "Healthy Blueberry Plant": "Blueberry - Healthy",
+    "Cherry with Powdery Mildew": "Cherry - Powdery Mildew",
+    "Healthy Cherry Plant": "Cherry - Healthy",
+    "Corn (Maize) with Cercospora and Gray Leaf Spot": "Corn - Gray Leaf Spot",
+    "Corn (Maize) with Common Rust": "Corn - Common Rust",
+    "Corn (Maize) with Northern Leaf Blight": "Corn - Northern Leaf Blight",
+    "Healthy Corn": "Corn - Healthy",
+    "Grape with Black Measles": "Grape - Black Measles",
+    "Grape with Black Rot": "Grape - Black Rot",
+    "Grape with Leaf Blight": "Grape - Leaf Blight",
+    "Healthy Grape": "Grape - Healthy",
+    "Orange with Citrus Greening": "Orange - Citrus Greening",
+    "Peach with Bacterial Spot": "Peach - Bacterial Spot",
+    "Healthy Peach": "Peach - Healthy",
+    "Pepper Bell with Bacterial Spot": "Pepper - Bacterial Spot",
+    "Healthy Pepper Bell": "Pepper - Healthy",
+    "Potato with Early Blight": "Potato - Early Blight",
+    "Potato with Late Blight": "Potato - Late Blight",
+    "Healthy Potato Plant": "Potato - Healthy",
+    "Healthy Raspberry Plant": "Raspberry - Healthy",
+    "Healthy Soybean Plant": "Soybean - Healthy",
+    "Squash with Powdery Mildew": "Squash - Powdery Mildew",
+    "Strawberry with Leaf Scorch": "Strawberry - Leaf Scorch",
+    "Healthy Strawberry Plant": "Strawberry - Healthy",
+    "Tomato with Bacterial Spot": "Tomato - Bacterial Spot",
+    "Tomato with Early Blight": "Tomato - Early Blight",
+    "Tomato with Late Blight": "Tomato - Late Blight",
+    "Tomato with Leaf Mold": "Tomato - Leaf Mold",
+    "Tomato with Septoria Leaf Spot": "Tomato - Septoria Spot",
+    "Tomato with Spider Mites or Two-spotted Spider Mite": "Tomato - Spider Mite",
+    "Tomato with Target Spot": "Tomato - Target Spot",
+    "Tomato Yellow Leaf Curl Virus": "Tomato - Yellow Leaf Curl",
+    "Tomato Mosaic Virus": "Tomato - Mosaic Virus",
+    "Healthy Tomato Plant": "Tomato - Healthy"
 }
 
 @app.get("/")

fix_map.py

@@ -0,0 +1,53 @@
+import re
+
+with open('backend/app.py', 'r', encoding='utf-8') as f:
+    text = f.read()
+
+new_map = """SPECIALIST_MAP = {
+    "Apple Scab": "Apple - Scab",
+    "Apple with Black Rot": "Apple - Black Rot",
+    "Cedar Apple Rust": "Apple - Rust",
+    "Healthy Apple": "Apple - Healthy",
+    "Healthy Blueberry Plant": "Blueberry - Healthy",
+    "Cherry with Powdery Mildew": "Cherry - Powdery Mildew",
+    "Healthy Cherry Plant": "Cherry - Healthy",
+    "Corn (Maize) with Cercospora and Gray Leaf Spot": "Corn - Gray Leaf Spot",
+    "Corn (Maize) with Common Rust": "Corn - Common Rust",
+    "Corn (Maize) with Northern Leaf Blight": "Corn - Northern Leaf Blight",
+    "Healthy Corn": "Corn - Healthy",
+    "Grape with Black Measles": "Grape - Black Measles",
+    "Grape with Black Rot": "Grape - Black Rot",
+    "Grape with Leaf Blight": "Grape - Leaf Blight",
+    "Healthy Grape": "Grape - Healthy",
+    "Orange with Citrus Greening": "Orange - Citrus Greening",
+    "Peach with Bacterial Spot": "Peach - Bacterial Spot",
+    "Healthy Peach": "Peach - Healthy",
+    "Pepper Bell with Bacterial Spot": "Pepper - Bacterial Spot",
+    "Healthy Pepper Bell": "Pepper - Healthy",
+    "Potato with Early Blight": "Potato - Early Blight",
+    "Potato with Late Blight": "Potato - Late Blight",
+    "Healthy Potato Plant": "Potato - Healthy",
+    "Healthy Raspberry Plant": "Raspberry - Healthy",
+    "Healthy Soybean Plant": "Soybean - Healthy",
+    "Squash with Powdery Mildew": "Squash - Powdery Mildew",
+    "Strawberry with Leaf Scorch": "Strawberry - Leaf Scorch",
+    "Healthy Strawberry Plant": "Strawberry - Healthy",
+    "Tomato with Bacterial Spot": "Tomato - Bacterial Spot",
+    "Tomato with Early Blight": "Tomato - Early Blight",
+    "Tomato with Late Blight": "Tomato - Late Blight",
+    "Tomato with Leaf Mold": "Tomato - Leaf Mold",
+    "Tomato with Septoria Leaf Spot": "Tomato - Septoria Spot",
+    "Tomato with Spider Mites or Two-spotted Spider Mite": "Tomato - Spider Mite",
+    "Tomato with Target Spot": "Tomato - Target Spot",
+    "Tomato Yellow Leaf Curl Virus": "Tomato - Yellow Leaf Curl",
+    "Tomato Mosaic Virus": "Tomato - Mosaic Virus",
+    "Healthy Tomato Plant": "Tomato - Healthy"
+}"""
+
+# Replace the specific block in the file
+text = re.sub(r'SPECIALIST_MAP\s*=\s*\{[^}]+\}', new_map, text, count=1)
+
+with open('backend/app.py', 'w', encoding='utf-8') as f:
+    f.write(text)
+
+print("Replaced!")

out.py

@@ -0,0 +1,8 @@
+import urllib.request, json
+url = 'https://huggingface.co/linkanjarad/mobilenet_v2_1.0_224-plant-disease-identification/raw/main/config.json'
+c = json.loads(urllib.request.urlopen(url).read().decode())
+labels = list(c.get('id2label', {}).values())
+print('NEW_MAP = {')
+for x in labels:
+    print(f'    "{x}": "{x}",')
+print('}')