Update app.py

app.py

@@ -18,10 +18,6 @@ feature_options = {
     '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'},
-    'stalk-surface-above-ring': {'f': 'fibrous', 'y': 'scaly', 'k': 'silky', 's': 'smooth'},
-    'stalk-surface-below-ring': {'f': 'fibrous', 'y': 'scaly', 'k': 'silky', 's': 'smooth'},
-    'stalk-color-above-ring': {'n': 'brown', 'b': 'buff', 'c': 'cinnamon', 'g': 'gray', 'o': 'orange', 'p': 'pink', 'e': 'red', 'w': 'white', 'y': 'yellow'},
-    'stalk-color-below-ring': {'n': 'brown', 'b': 'buff', 'c': 'cinnamon', 'g': 'gray', 'o': 'orange', 'p': 'pink', 'e': 'red', 'w': 'white', 'y': 'yellow'},
     'veil-type': {'p': 'partial', 'u': 'universal'},
     'veil-color': {'n': 'brown', 'o': 'orange', 'w': 'white', 'y': 'yellow'},
     'ring-number': {'n': 'none', 'o': 'one', 't': 'two'},
@@ -31,20 +27,52 @@ feature_options = {
     'habitat': {'g': 'grasses', 'l': 'leaves', 'm': 'meadows', 'p': 'paths', 'u': 'urban', 'w': 'waste', 'd': 'woods'}
 }
 
-def predict_mushroom(**features):
+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 detected."
+            return "Invalid input. Please check feature values."
     
     input_df = pd.DataFrame([numerical_features])
     prediction = model.predict(input_df)
     return "Edible" if prediction[0] == 0 else "Poisonous"
 
-dropdowns = [gr.Dropdown(choices=list(options.values()), label=feature.replace('-', ' ').capitalize()) for feature, options in feature_options.items()]
+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'
+    },
+    {  # 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'
+    }
+]
 
-demo = gr.Interface(fn=predict_mushroom, inputs=dropdowns, outputs="text", title="Mushroom Classification", description="Select the characteristics of the mushroom to determine if it's edible or poisonous.")
+demo = gr.Interface(
+    fn=predict_mushroom,
+    inputs=[gr.Dropdown(choices=list(options.keys()), label=key) for key, 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."
+)
 
 demo.launch()