Update app.py

app.py

@@ -7,97 +7,40 @@ model = joblib.load('mushroom_classifier.pkl')
 mappings = joblib.load('mappings.pkl')
 
 feature_options = {
-    'cap-shape': ['bell', 'conical', 'convex', 'flat', 'knobbed', 'sunken'],
-    'cap-surface': ['fibrous', 'grooves', 'scaly', 'smooth'],
-    'cap-color': ['brown', 'buff', 'cinnamon', 'gray', 'green', 'pink', 'purple', 'red', 'white', 'yellow'],
-    'bruises': ['bruises', 'no'],
-    'odor': ['almond', 'anise', 'creosote', 'fishy', 'foul', 'musty', 'none', 'pungent', 'spicy'],
-    'gill-attachment': ['attached', 'descending', 'free', 'notched'],
-    'gill-spacing': ['close', 'crowded', 'distant'],
-    'gill-size': ['broad', 'narrow'],
-    'gill-color': ['black', 'brown', 'buff', 'chocolate', 'gray', 'green', 'orange', 'pink', 'purple', 'red', 'white', 'yellow'],
-    'stalk-shape': ['enlarging', 'tapering'],
-    'stalk-root': ['bulbous', 'club', 'cup', 'equal', 'rhizomorphs', 'rooted', 'missing'],
-    'stalk-surface-above-ring': ['fibrous', 'scaly', 'silky', 'smooth'],
-    'stalk-surface-below-ring': ['fibrous', 'scaly', 'silky', 'smooth'],
-    'stalk-color-above-ring': ['brown', 'buff', 'cinnamon', 'gray', 'orange', 'pink', 'red', 'white', 'yellow'],
-    'stalk-color-below-ring': ['brown', 'buff', 'cinnamon', 'gray', 'orange', 'pink', 'red', 'white', 'yellow'],
-    'veil-type': ['partial', 'universal'],
-    'veil-color': ['brown', 'orange', 'white', 'yellow'],
-    'ring-number': ['none', 'one', 'two'],
-    'ring-type': ['cobwebby', 'evanescent', 'flaring', 'large', 'none', 'pendant', 'sheathing', 'zone'],
-    'spore-print-color': ['black', 'brown', 'buff', 'chocolate', 'green', 'orange', 'purple', 'white', 'yellow'],
-    'population': ['abundant', 'clustered', 'numerous', 'scattered', 'several', 'solitary'],
-    'habitat': ['grasses', 'leaves', 'meadows', 'paths', 'urban', 'waste', 'woods']
+    'cap-shape': {'bell': 'b', 'conical': 'c', 'convex': 'x', 'flat': 'f', 'knobbed': 'k', 'sunken': 's'},
+    'cap-surface': {'fibrous': 'f', 'grooves': 'g', 'scaly': 'y', 'smooth': 's'},
+    'cap-color': {'brown': 'n', 'buff': 'b', 'cinnamon': 'c', 'gray': 'g', 'green': 'r', 'pink': 'p', 'purple': 'u', 'red': 'e', 'white': 'w', 'yellow': 'y'},
+    'bruises': {'bruises': 't', 'no': 'f'},
+    'odor': {'almond': 'a', 'anise': 'l', 'creosote': 'c', 'fishy': 'y', 'foul': 'f', 'musty': 'm', 'none': 'n', 'pungent': 'p', 'spicy': 's'},
+    'gill-attachment': {'attached': 'a', 'descending': 'd', 'free': 'f', 'notched': 'n'},
+    'gill-spacing': {'close': 'c', 'crowded': 'w', 'distant': 'd'},
+    'gill-size': {'broad': 'b', 'narrow': 'n'},
+    'gill-color': {'black': 'k', 'brown': 'n', 'buff': 'b', 'chocolate': 'h', 'gray': 'g', 'green': 'r', 'orange': 'o', 'pink': 'p', 'purple': 'u', 'red': 'e', 'white': 'w', 'yellow': 'y'},
+    'stalk-shape': {'enlarging': 'e', 'tapering': 't'},
+    'stalk-root': {'bulbous': 'b', 'club': 'c', 'cup': 'u', 'equal': 'e', 'rhizomorphs': 'z', 'rooted': 'r', 'missing': '?'},
+    'stalk-surface-above-ring': {'fibrous': 'f', 'scaly': 'y', 'silky': 'k', 'smooth': 's'},
+    'stalk-surface-below-ring': {'fibrous': 'f', 'scaly': 'y', 'silky': 'k', 'smooth': 's'},
+    'stalk-color-above-ring': {'brown': 'n', 'buff': 'b', 'cinnamon': 'c', 'gray': 'g', 'orange': 'o', 'pink': 'p', 'red': 'e', 'white': 'w', 'yellow': 'y'},
+    'stalk-color-below-ring': {'brown': 'n', 'buff': 'b', 'cinnamon': 'c', 'gray': 'g', 'orange': 'o', 'pink': 'p', 'red': 'e', 'white': 'w', 'yellow': 'y'},
+    'veil-type': {'partial': 'p', 'universal': 'u'},
+    'veil-color': {'brown': 'n', 'orange': 'o', 'white': 'w', 'yellow': 'y'},
+    'ring-number': {'none': 'n', 'one': 'o', 'two': 't'},
+    'ring-type': {'cobwebby': 'c', 'evanescent': 'e', 'flaring': 'f', 'large': 'l', 'none': 'n', 'pendant': 'p', 'sheathing': 's', 'zone': 'z'},
+    'spore-print-color': {'black': 'k', 'brown': 'n', 'buff': 'b', 'chocolate': 'h', 'green': 'r', 'orange': 'o', 'purple': 'u', 'white': 'w', 'yellow': 'y'},
+    'population': {'abundant': 'a', 'clustered': 'c', 'numerous': 'n', 'scattered': 's', 'several': 'v', 'solitary': 'y'},
+    'habitat': {'grasses': 'g', 'leaves': 'l', 'meadows': 'm', 'paths': 'p', 'urban': 'u', 'waste': 'w', 'woods': 'd'}
 }
 
 def predict_mushroom(features):
-    numerical_features = {}
-    for feature, value in features.items():
-        for key, val in mappings[feature].items():
-            if val == value:
-                numerical_features[feature] = key
-                break
+    numerical_features = {feature: feature_options[feature][value] for feature, value in features.items()}
     input_df = pd.DataFrame([numerical_features])
     prediction = model.predict(input_df)
     return 'Poisonous' if prediction[0] == 1 else 'Edible'
 
-examples = [
-    {  # Death Cap
-        'cap-shape': 'convex',
-        'cap-surface': 'smooth',
-        'cap-color': 'green',
-        'bruises': 'no',
-        'odor': 'foul',
-        'gill-attachment': 'free',
-        'gill-spacing': 'crowded',
-        'gill-size': 'narrow',
-        'gill-color': 'white',
-        'stalk-shape': 'tapering',
-        'stalk-root': 'bulbous',
-        'stalk-surface-above-ring': 'smooth',
-        'stalk-surface-below-ring': 'smooth',
-        'stalk-color-above-ring': 'white',
-        'stalk-color-below-ring': 'white',
-        'veil-type': 'partial',
-        'veil-color': 'white',
-        'ring-number': 'one',
-        'ring-type': 'pendant',
-        'spore-print-color': 'white',
-        'population': 'several',
-        'habitat': 'woods'
-    },
-    {  # Edible Brown Mushroom
-        'cap-shape': 'convex',
-        'cap-surface': 'smooth',
-        'cap-color': 'brown',
-        'bruises': 'no',
-        'odor': 'almond',
-        'gill-attachment': 'free',
-        'gill-spacing': 'crowded',
-        'gill-size': 'broad',
-        'gill-color': 'brown',
-        'stalk-shape': 'enlarging',
-        'stalk-root': 'equal',
-        'stalk-surface-above-ring': 'smooth',
-        'stalk-surface-below-ring': 'smooth',
-        'stalk-color-above-ring': 'white',
-        'stalk-color-below-ring': 'white',
-        'veil-type': 'partial',
-        'veil-color': 'white',
-        'ring-number': 'one',
-        'ring-type': 'pendant',
-        'spore-print-color': 'brown',
-        'population': 'numerous',
-        'habitat': 'grasses'
-    }
-]
-
 demo = gr.Interface(
     fn=predict_mushroom,
-    inputs=[gr.Dropdown(choices=options, label=feature) for feature, options in feature_options.items()],
+    inputs=[gr.Dropdown(choices=list(options.keys()), label=feature) for feature, options in feature_options.items()],
     outputs="text",
-    examples=[[ex[feature] for feature in feature_options] for ex in examples],
     title="MycoNom - Mushroom Edibility Classifier",
     description="Select the mushroom features to determine if it's edible or poisonous.<br><br>**Disclaimer:** This model is for **educational purposes only** and should not be used for real-life mushroom classification or any decision-making processes related to the consumption of mushrooms. While the model performs well on the provided dataset, it has not been thoroughly validated for real-world scenarios and may not accurately detect poisonous mushrooms in all conditions. Always consult an expert or use trusted resources when identifying mushrooms."
 )