Update app.py

app.py

@@ -1,7 +1,10 @@
+# Importing the necessary libraries
 import streamlit as st
 import pandas as pd
 import pickle
 
+
+# Setting up the page configuration for Streamlit App
 st.set_page_config(
     page_title=" :mushroom: Mushroom App",
     page_icon="🍄",
@@ -10,7 +13,9 @@ st.set_page_config(
 )
 
 
+# Function for user input features
 def user_input_features():
+    # Creating sliders and select boxes for user input in the sidebar
     cap_diameter = st.sidebar.slider('Cap Diameter',
                             min_value=0.0,
                             max_value=2000.0,
@@ -24,7 +29,7 @@ def user_input_features():
                                     'flat',
                                     'sunken',
                                     'spherical',
-                                    'other')
+                                    'other',)
     )
     gill_attachment = st.sidebar.selectbox('Gill Attachment',
                             options=('adnate',
@@ -33,7 +38,7 @@ def user_input_features():
                                     'free',
                                     'sinuate',
                                     'pores',
-                                    'none')
+                                    'none',)
     )
     gill_color = st.sidebar.selectbox('Gill Color',
                             options=('brown',
@@ -47,7 +52,7 @@ def user_input_features():
                                     'yellow',
                                     'blue',
                                     'orange',
-                                    'black')
+                                    'black',)
     )
     stem_height = st.sidebar.slider('Stem Height',
                             min_value=0.0,
@@ -73,16 +78,17 @@ def user_input_features():
                                     'yellow',
                                     'blue',
                                     'orange',
-                                    'black')
+                                    'black',)
     )
     season = st.sidebar.selectbox('Season',
                             options=('spring',
                                     'summer',
                                     'autumn',
-                                    'winter')                                    
+                                    'winter',)
     )
-    
-    
+
+
+    # Function to get the color code
     def get_color(color_name):
         color_dict = {
             'brown': 0,
@@ -96,11 +102,12 @@ def user_input_features():
             'yellow': 8,
             'blue': 9,
             'orange': 10,
-            'black': 11
+            'black': 11,
         }
         return color_dict.get(color_name.lower(), "not found")
-    
 
+
+    # Function to get the cap shape code
     def get_cap_shape(cap_shape):
         shape_dict = {
             'bell': 0,
@@ -109,11 +116,12 @@ def user_input_features():
             'flat': 3,
             'sunken': 4,
             'spherical': 5,
-            'other': 6
+            'other': 6,
         }
         return shape_dict.get(cap_shape.lower(), "not found")
-    
 
+
+    # Function to get gill attachment code
     def get_gill_attachment(gill_attachment):
         gill_attachment_dict = {
             'adnate': 0,
@@ -122,20 +130,22 @@ def user_input_features():
             'free': 3,
             'sinuate': 4,
             'pores': 5,
-            'none': 6
+            'none': 6,
         }
         return gill_attachment_dict.get(gill_attachment.lower(), "not found")
 
 
+    # Function to get season code
     def get_season(season):
         season_dict = {
             'spring': 0,
             'summer': 1,
             'autumn': 2,
-            'winter': 3
+            'winter': 3,
         }
         return season_dict.get(season.lower(), "not found")
 
+    # Creating a data dictionary to store the user input data
     data = {'cap-diameter': cap_diameter,
             'cap-shape': get_cap_shape(cap_shape),
             'gill-attachment': get_gill_attachment(gill_attachment),
@@ -143,40 +153,47 @@ def user_input_features():
             'stem-height': stem_height,
             'stem-width': stem_width,
             'stem-color': get_color(stem_color),
-            'season': get_season(season)                               
+            'season': get_season(season),
     }
-    
+
+    # Creating a DataFrame from the data dictionary
     features = pd.DataFrame(data, index=[0])
     return features
 
 
+# Function to load the prediction model
 #@st.cache_data()
 def get_model():
-    model = pickle.load(open("models/rfc_model.pkl", "rb"))    
+    model = pickle.load(open("models/rfc_model.pkl", "rb"))
     return model
 
 
+# Function to make prediction using the model and input data
 def make_prediction(data):
     model = get_model()
     return model.predict(data)
 
+
+# Main function
 def main():
     st.write("""# :mushroom: Mushroom App""")
     st.sidebar.image("img/dataset-cover.jpg")
     user_data = user_input_features()
 
-    
+    # Creating a session state button for prediction
     if 'btn_predict' not in st.session_state:
         st.session_state['btn_predict'] = False
 
     st.session_state['btn_predict'] = st.button("Predict")
-    
-    if st.session_state['btn_predict'] == True:        
+
+    # Making prediction and showing result
+    if st.session_state['btn_predict'] == True:
         if make_prediction(user_data) == 1:
             st.error("# Result: Poisonous :skull_and_crossbones: ")
         else:
             st.success("# Result: Edible :mushroom: ")
-        
-    
 
-main()
+
+# Running the main function
+if __name__ == "__main__":
+    main()