Spaces:
Runtime error
Runtime error
Alejadro Sanchez-Giraldo
commited on
Commit
·
796b25f
1
Parent(s):
2ffcbc1
added Injuries query
Browse files- fpl_client.py +8 -1
- nlp_utils.py +5 -0
fpl_client.py
CHANGED
@@ -89,4 +89,11 @@ class FPLClient:
|
|
89 |
logging.debug(f"Players found for team {team_name}: {players}")
|
90 |
return players
|
91 |
logging.warning(f"Team not found: {team_name}")
|
92 |
-
return "Team not found"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
89 |
logging.debug(f"Players found for team {team_name}: {players}")
|
90 |
return players
|
91 |
logging.warning(f"Team not found: {team_name}")
|
92 |
+
return "Team not found"
|
93 |
+
|
94 |
+
def list_injuries(self):
|
95 |
+
logging.info("Fetching injuries list")
|
96 |
+
players = self.data['elements']
|
97 |
+
injuries = [(p['web_name'], p['news']) for p in players if p['news'] and 'injury' in p['news'].lower()]
|
98 |
+
logging.debug(f"Injured players: {injuries}")
|
99 |
+
return injuries
|
nlp_utils.py
CHANGED
@@ -59,5 +59,10 @@ def process_query(query, fpl_client):
|
|
59 |
|
60 |
return "It sound like you need the latest EPL team list: " + "\n" + "\n".join(enumerated_teams)
|
61 |
|
|
|
|
|
|
|
|
|
|
|
62 |
|
63 |
return "I'm not sure how to help with that."
|
|
|
59 |
|
60 |
return "It sound like you need the latest EPL team list: " + "\n" + "\n".join(enumerated_teams)
|
61 |
|
62 |
+
if "injuries" in query.lower():
|
63 |
+
injuries = fpl_client.list_injuries()
|
64 |
+
formatted_injuries = "\n\n".join([f"Player: {name} Injury: {news}" for name, news in injuries])
|
65 |
+
injury_count = len(injuries)
|
66 |
+
return f"There are {injury_count} players with injuries:\n\n{formatted_injuries}"
|
67 |
|
68 |
return "I'm not sure how to help with that."
|