fixes to the team find theme

.gitignore

@@ -2,4 +2,6 @@ fplenv/
 .DS_Store
 .env
 
-__pycache__/
+__pycache__/
+
+flagged

app.py

@@ -2,6 +2,15 @@ import gradio as gr
 from fpl_client import FPLClient
 from nlp_utils import process_query
 
+# Theme builder
+# gr.themes.builder()
+
+theme = gr.themes.Soft(
+    primary_hue="sky",
+    neutral_hue="slate",
+)
+
+
 # Initialize the FPL client
 fpl_client = FPLClient()
 
@@ -11,5 +20,12 @@ def chatbot_response(query):
     return response
 
 # Set up the Gradio interface
-iface = gr.Interface(fn=chatbot_response, inputs="text", outputs="text", title="FPL Chatbot")
-iface.launch()
+iface = gr.Interface(
+    fn=chatbot_response,
+    inputs=gr.Textbox(label="Ask our FPL Expert"),
+    outputs=gr.Textbox(label="Hope it helps!"),
+    theme=theme,
+    title="Devops1 FPL Chatbot"
+)
+if __name__ == "__main__":
+    iface.launch()

nlp_utils.py

@@ -10,12 +10,14 @@ teams = fpl_client.get_teams()
 
 def extract_player_name(query):
     # Simple regex example to extract player names (can be enhanced)
-    match = re.search(r"(\b[A-Z][a-z]+(?:\s[A-Z][a-z]+)?)", query)
+    # Updated regex to include accented characters and names with periods
+    match = re.search(r"(\b[A-Z][a-zA-Z]*[a-zA-Z.]*\b(?:\s[A-Z][a-zA-Z]*[a-zA-Z.]*\b)?)", query, re.UNICODE)
+    logging.info(f"Extracted player name: {match}")
     return match.group(1) if match else None
 
 def extract_team_name(query):
     logging.info(f"Extracting team name from query: {query}")
-    logging.info(f"Teams: {teams}")
+    logging.debug(f"Teams: {teams}")
     # Add your logic to extract team names, could use a list of teams
     for team in teams:
         if team.lower() in query.lower():
@@ -30,11 +32,22 @@ def process_query(query, fpl_client):
             return stats
         return "Player name not found in the query."
     
-    if "team" in query.lower():
+    if "team" in query.lower() or "players" in query.lower():
+        # Extract the team name from the query
         team_name = extract_team_name(query)
+        
+        # Check if a valid team name was found
         if team_name:
+            # Get the list of players for the specified team
             players = fpl_client.get_team_players(team_name)
-            return f"Players from {team_name}: {', '.join(players)}"
-        return "Team name not found in the query, please use from the list: " + ", ".join(teams)
+                      
+            # Return the formatted list of players from the specified team
+            return f"Players from {team_name} are:\n" + "\n".join(players)
+        
+        # If no valid team name was found, enumerate the teams and format them as a numbered list
+        enumerated_teams = [f"{index + 1}. {team}" for index, team in enumerate(teams)]
+        
+        return "It sound like you need the latest EPL team list: " + "\n" + "\n".join(enumerated_teams)
     
+       
     return "I'm not sure how to help with that."

requirements.txt

@@ -1,4 +1,5 @@
 huggingface_hub==0.22.2
 gradio
 requests
-transformers
+transformers
+minijinja