Update app.py

app.py

@@ -16,13 +16,12 @@ def name_generator(mom: str, dad: str) -> list:
     Generate baby names based on parental preferences.
 
     Args:
-        mom (string): The mother's preferences for the name (e.g., "classic", "modern", "French names").
-        dad (string): The father's preferences for the name (e.g., "hero names", "Corsican names", "short names").
+        mom (str): The mother's preferences for the name (e.g., "classic", "modern", "French names").
+        dad (str): The father's preferences for the name (e.g., "hero names", "Corsican names", "short names").
 
     Returns:
         list: A list of baby names that match the given preferences.
     """
-    # Création du prompt pour générer plusieurs prénoms
     prompt = f"""
     You are an AI that suggests baby names based on parental preferences.
     The mother prefers: {mom}.
@@ -32,17 +31,12 @@ def name_generator(mom: str, dad: str) -> list:
     Just return the names, separated by commas, nothing else.
     """
 
-    # Envoi de la requête au modèle
-    # Assure-toi que la méthode 'generate' existe bien pour l'objet model
     response = model.generate(prompt)
-
-    # Nettoyage et transformation en liste
-    baby_names = response.strip().split(",")  # Sépare les prénoms par virgule
-    baby_names = [name.strip() for name in baby_names if name.strip()]
-
+    baby_names = [name.strip() for name in response.strip().split(",") if name.strip()]
     return baby_names if baby_names else ["No names generated. Try different preferences."]
 
 
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.