Amitabhab commited on
Commit
8663a63
·
1 Parent(s): 0ad0723

Enhance parsing logic

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -54,11 +54,15 @@ def run(origin, destination, age, trip_duration, children, budget):
54
  json_addresses = addresses.json_dict
55
  if not json_addresses:
56
  try:
57
- json_addresses = json.loads(addresses.raw[8:-4])
58
  except json.JSONDecodeError as e:
59
- logger.error("Error loading Crew Output for addresses")
60
- logger.info(addresses.raw)
61
- return (result, None)
 
 
 
 
62
  fig = filter_map(json_addresses["name"], json_addresses["lat"], json_addresses["lon"])
63
  return (result, fig)
64
 
 
54
  json_addresses = addresses.json_dict
55
  if not json_addresses:
56
  try:
57
+ json_addresses = json.loads(addresses.raw)
58
  except json.JSONDecodeError as e:
59
+ # Try with different format of result data generated with ```json and ending with ```
60
+ try:
61
+ json_addresses = json.loads(addresses.raw[8:-4])
62
+ except json.JSONDecodeError as e:
63
+ logger.error("Error loading Crew Output for addresses")
64
+ logger.info(addresses.raw)
65
+ return (result, None)
66
  fig = filter_map(json_addresses["name"], json_addresses["lat"], json_addresses["lon"])
67
  return (result, fig)
68