fix: check if the key exists before to access to it in the dict.

app.py

@@ -292,7 +292,8 @@ def generate_answer(text):
     # Extract valid string left dotted
     feature_surfaces = ['SUPERFICIE_BALCON', 'SUPERFICIE_TERRAZA', 'SUPERFICIE_JARDIN', 'SUPERFICIE_TERRENO', 'SUPERFICIE_HABITABLE']
     for feature_name in feature_surfaces:
-        entities_formatted[f"pred_{feature_name}"] = extract_valid_string_left_dotted(entities_formatted[f"pred_{feature_name}"], text, entities_formatted[f"pos_{feature_name}"])
+        if entities_formatted.get(f"pred_{feature_name}", None) != None:
+           entities_formatted[f"pred_{feature_name}"] = extract_valid_string_left_dotted(entities_formatted[f"pred_{feature_name}"], text, entities_formatted[f"pos_{feature_name}"])
 
     # Clean Entities
     entities_names = list({c.replace("pred_", "").replace("prob_", "") for c in list(entities_formatted.keys())})