Update app.py

app.py

@@ -15,28 +15,32 @@ def name_generator(mom: str, dad: str) -> list:
     """
     Generate baby names based on parental preferences.
 
-    Args:
-        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.
+    :param mom: The mother's preferences for the name (e.g., "classic", "modern").
+    :type mom: str
+    :param dad: The father's preferences for the name (e.g., "hero names", "short names").
+    :type dad: str
+    :return: A list of baby names that match the given preferences
+    :rtype: list
     """
     prompt = f"""
     You are an AI that suggests baby names based on parental preferences.
     The mother prefers: {mom}.
     The father prefers: {dad}.
-    
+
     Suggest as many baby names as possible.
     Just return the names, separated by commas, nothing else.
     """
 
+    # Ton code pour générer la réponse via 'model'
     response = model.generate(prompt)
-    baby_names = [name.strip() for name in response.strip().split(",") if name.strip()]
+    baby_names = response.strip().split(",")
+    baby_names = [name.strip() for name in baby_names 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.