Final changes

app.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from joblib import load
 from PIL import Image
 
@@ -12,7 +13,7 @@ st.image(image)
 st.markdown("## ")
 st.markdown("### Input your passenger data: ")
 
-age = st.number_input('Age', min_value=0, max_value=100)
+age = st.number_input('Age', min_value=1, max_value=80)
 
 sex = st.radio("Sex", ("Male", "Female"))
 
@@ -24,6 +25,10 @@ p_class = st.radio(
     "Passenger Class",
     ('Class 1', 'Class 2', 'Class 3'))
 
+siblings = st.number_input('Number of Siblings/Spouses Aboard', min_value=0, max_value=5)
+
+parents = st.number_input("Number of Parents/Children Aboard", min_value=0, max_value=6)
+
 
 passenger_data = {
     'Age': age,
@@ -34,22 +39,31 @@ passenger_data = {
     'Embarked_C': 1 if embark_port[0] == "C" else 0,
     'Embarked_Q': 1 if embark_port[0] == "Q" else 0,
     'Embarked_S': 1 if embark_port[0] == "S" else 0,
-    'SibSp': 0.0,
-    'Parch': 0.0
+    'SibSp': siblings,
+    'Parch': parents,
 }
 
 passenger_data_df = pd.DataFrame.from_dict([passenger_data])
 
-prediction = model.predict(passenger_data_df)[0]
-prediction_proba = model.predict_proba(passenger_data_df)[:,1][0]
+if st.button("Calculate my chances of survival"):
 
-if prediction == 0:
-    st.markdown("# YOU'RE DEAD :cry:")
-    
-elif sex == "Male":
-    st.markdown("# YOU MADE IT, :man-swimming:")
+    prediction = model.predict(passenger_data_df)[0]
+    prediction_proba = model.predict_proba(passenger_data_df)[:,1][0]
 
-else:
-    st.markdown("# YOU MADE IT, :woman-swimming:")
+    if prediction == 0:
+        st.markdown("# YOU'RE DEAD :skull:")
+        
+    elif sex == "Male":
+        st.markdown("# YOU MADE IT, :man-swimming:")
     
-st.write(f"Probabilty of survival {round(prediction_proba, 2)* 100}%.")
+    else:
+        st.markdown("# YOU MADE IT, :woman-swimming:")
+        
+    
+    st.write(f"Probabilty of survival: {round(prediction_proba, 2)* 100}%.")
+
+    passenger_data_df.insert(0, "date", str(datetime.now()))
+    passenger_data_df["survived"] = prediction
+    passenger_data_df["survival_prob"] = prediction_proba
+
+    passenger_data_df.to_csv('model_log.csv', mode='a', header=False, index=False)

model_log.csv

@@ -0,0 +1,2 @@
+date,Age,is_male,Pclass_class 1,Pclass_class 2, Pclass_class 3, Embarked_C, Embarked_Q, Embarked_S, SibSp, Parch, survived, survival_prob,
+