Update app.py

app.py

@@ -2,77 +2,104 @@ import gradio as gr
 import joblib
 import pandas as pd
 
-# Load the trained model and mappings
+# Load model and mappings
 model = joblib.load('mushroom_classifier.pkl')
 mappings = joblib.load('mappings.pkl')
 
 feature_options = {
-    'cap-shape': {'b': 'bell', 'c': 'conical', 'x': 'convex', 'f': 'flat', 'k': 'knobbed', 's': 'sunken'},
-    'cap-surface': {'f': 'fibrous', 'g': 'grooves', 'y': 'scaly', 's': 'smooth'},
-    'cap-color': {'n': 'brown', 'b': 'buff', 'c': 'cinnamon', 'g': 'gray', 'r': 'green', 'p': 'pink', 'u': 'purple', 'e': 'red', 'w': 'white', 'y': 'yellow'},
-    'bruises': {'t': 'bruises', 'f': 'no'},
-    'odor': {'a': 'almond', 'l': 'anise', 'c': 'creosote', 'y': 'fishy', 'f': 'foul', 'm': 'musty', 'n': 'none', 'p': 'pungent', 's': 'spicy'},
-    'gill-attachment': {'a': 'attached', 'd': 'descending', 'f': 'free', 'n': 'notched'},
-    'gill-spacing': {'c': 'close', 'w': 'crowded', 'd': 'distant'},
-    'gill-size': {'b': 'broad', 'n': 'narrow'},
-    'gill-color': {'k': 'black', 'n': 'brown', 'b': 'buff', 'h': 'chocolate', 'g': 'gray', 'r': 'green', 'o': 'orange', 'p': 'pink', 'u': 'purple', 'e': 'red', 'w': 'white', 'y': 'yellow'},
-    'stalk-shape': {'e': 'enlarging', 't': 'tapering'},
-    'stalk-root': {'b': 'bulbous', 'c': 'club', 'u': 'cup', 'e': 'equal', 'z': 'rhizomorphs', 'r': 'rooted', '?': 'missing'},
-    'veil-type': {'p': 'partial', 'u': 'universal'},
-    'veil-color': {'n': 'brown', 'o': 'orange', 'w': 'white', 'y': 'yellow'},
-    'ring-number': {'n': 'none', 'o': 'one', 't': 'two'},
-    'ring-type': {'c': 'cobwebby', 'e': 'evanescent', 'f': 'flaring', 'l': 'large', 'n': 'none', 'p': 'pendant', 's': 'sheathing', 'z': 'zone'},
-    'spore-print-color': {'k': 'black', 'n': 'brown', 'b': 'buff', 'h': 'chocolate', 'r': 'green', 'o': 'orange', 'u': 'purple', 'w': 'white', 'y': 'yellow'},
-    'population': {'a': 'abundant', 'c': 'clustered', 'n': 'numerous', 's': 'scattered', 'v': 'several', 'y': 'solitary'},
-    'habitat': {'g': 'grasses', 'l': 'leaves', 'm': 'meadows', 'p': 'paths', 'u': 'urban', 'w': 'waste', 'd': 'woods'}
+    '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']
 }
 
 def predict_mushroom(features):
     numerical_features = {}
     for feature, value in features.items():
-        if feature in mappings and value in mappings[feature]:
-            numerical_features[feature] = mappings[feature][value]
-        else:
-            return "Invalid input. Please check feature values."
-    
+        for key, val in mappings[feature].items():
+            if val == value:
+                numerical_features[feature] = key
+                break
     input_df = pd.DataFrame([numerical_features])
     prediction = model.predict(input_df)
-    return "Edible" if prediction[0] == 0 else "Poisonous"
+    return 'Poisonous' if prediction[0] == 1 else 'Edible'
 
 examples = [
-    {  # Death Cap (Poisonous)
-        'cap-shape': 'x', 'cap-surface': 's', 'cap-color': 'r', 'bruises': 'f', 'odor': 'p',
-        'gill-attachment': 'f', 'gill-spacing': 'w', 'gill-size': 'n', 'gill-color': 'w',
-        'stalk-shape': 'e', 'stalk-root': 'b', 'veil-type': 'p', 'veil-color': 'w',
-        'ring-number': 'o', 'ring-type': 'p', 'spore-print-color': 'w', 'population': 'v', 'habitat': 'd'
+    {  # 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'
     },
-    {  # Common Brown Mushroom (Edible)
-        'cap-shape': 'x', 'cap-surface': 's', 'cap-color': 'n', 'bruises': 't', 'odor': 'a',
-        'gill-attachment': 'f', 'gill-spacing': 'w', 'gill-size': 'b', 'gill-color': 'h',
-        'stalk-shape': 'e', 'stalk-root': 'c', 'veil-type': 'p', 'veil-color': 'w',
-        'ring-number': 'o', 'ring-type': 'p', 'spore-print-color': 'n', 'population': 'n', 'habitat': 'm'
-    },
-    {  # Fly Agaric (Poisonous)
-        'cap-shape': 'x', 'cap-surface': 'y', 'cap-color': 'e', 'bruises': 'f', 'odor': 'n',
-        'gill-attachment': 'f', 'gill-spacing': 'w', 'gill-size': 'b', 'gill-color': 'w',
-        'stalk-shape': 'e', 'stalk-root': 'b', 'veil-type': 'p', 'veil-color': 'w',
-        'ring-number': 'o', 'ring-type': 'p', 'spore-print-color': 'w', 'population': 'v', 'habitat': 'd'
-    },
-    {  # Oyster Mushroom (Edible)
-        'cap-shape': 'x', 'cap-surface': 's', 'cap-color': 'w', 'bruises': 'f', 'odor': 'l',
-        'gill-attachment': 'd', 'gill-spacing': 'c', 'gill-size': 'b', 'gill-color': 'w',
-        'stalk-shape': 't', 'stalk-root': '?', 'veil-type': 'p', 'veil-color': 'w',
-        'ring-number': 'n', 'ring-type': 'n', 'spore-print-color': 'u', 'population': 'c', 'habitat': 'd'
+    {  # 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=list(options.keys()), label=key) for key, options in feature_options.items()],
+    inputs=[gr.Dropdown(choices=options, label=feature) for feature, options in feature_options.items()],
     outputs="text",
-    examples=[[v for v in example.values()] for example in examples],
-    title="MycoNom - Mushroom Classifier",
-    description="Select mushroom characteristics to predict if it's edible or poisonous. **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."
+    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."
 )
 
 demo.launch()