'adsf'

app.py

@@ -65,7 +65,7 @@ worthlessInputs = ["Name", "Ticket", "Cabin", "Title"]
 categoricalInputs = ["Sex", "Embarked", "Pclass"]
 
 columnHeaders = ["Pclass", "Sex", "Age", "SibSp",
-                 "Parch", "Fare", "Embarked", "Title"] # Todo: remove title
+                 "Parch", "Fare", "Embarked", "Title"]  # Todo: remove title
 
 
 def titanic(Pclass, Sex, Age, SibSp, Parch, Fare, Embarked):
@@ -86,23 +86,35 @@ def titanic(Pclass, Sex, Age, SibSp, Parch, Fare, Embarked):
 
     intLabelToText = {0: "Died", 1: "Survived"}
 
-    age = input_list[0]
-    gender = input_list[1]
     survived = res[0]
-    survivedText = intLabelToText[survived]
 
     # Temp:
     age = 20
     gender = "female"
 
     # Todo: survivor, "https://fakeface.rest/face/json?maximum_age=50&gender=female&minimum_age=49"
-    generate_survivor_url = f'https://fakeface.rest/face/json?maximum_age={age}&gender={gender}&minimum_age={age}'
+    generate_survivor_url = f'https://fakeface.rest/face/json?maximum_age={Age}&gender={Sex}&minimum_age={age}'
     randomized_face_url = requests.get(
         generate_survivor_url).json()["image_url"]
 
     survivor_url = randomized_face_url
     img = Image.open(requests.get(survivor_url, stream=True).raw)
-    return img
+
+    #
+    red_cross_url = "https://www.iconsdb.com/icons/preview/red/x-mark-xxl.png"
+    green_check_mark_url = "https://www.iconsdb.com/icons/preview/green/checkmark-xxl.png"
+
+    label_to_url = {
+        0: red_cross_url,
+        1: green_check_mark_url
+    }
+
+    url = label_to_url.get(survived)
+
+    # Save the image of the person
+    img2 = Image.open(requests.get(url, stream=True).raw)
+
+    return img, img2
 
 
 catToInput = {