stkarlos commited on
Commit
6d6b808
·
verified ·
1 Parent(s): 9adbe8d

fix: int arg shoudl fit fairytale length

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -27,8 +27,9 @@ def my_fairytale_tool(arg:int)-> str: #it's import to specify the return type
27
  ]
28
 
29
  # Randomly select fairytales from the list
30
- selected_1 = fairytales[int(arg // len(fairytales))]
31
- fairytales.pop(int(arg // len(fairytales)))
 
32
  selected_2 = fairytales[random.randint(0,len(fairytales)-1)]
33
 
34
  return f"Title to draw: What a crazy idea to see a story of {selected_1} vs {selected_2}!"
 
27
  ]
28
 
29
  # Randomly select fairytales from the list
30
+ normalize_to_fairytales_range = int(arg) // len(fairytales)
31
+ selected_1 = fairytales[normalize_to_fairytales_range]
32
+ fairytales.pop(int(arg) // len(fairytales)
33
  selected_2 = fairytales[random.randint(0,len(fairytales)-1)]
34
 
35
  return f"Title to draw: What a crazy idea to see a story of {selected_1} vs {selected_2}!"