Update app.py

app.py

@@ -12,8 +12,8 @@ from Gradio_UI import GradioUI
 def create_fantasy_story(heros_name: str, number_of_chapter: int) -> str:
     """A tool that generates a fantasy story based on a hero's name and a specified number of chapters.
     Args:
-        arg1: The name of the hero or heroine.
-        arg2: The number of chapters in the story.
+        heros_name: The name of the hero or heroine.
+        number_of_chapter: The number of chapters in the story.
     """
     # Story templates for chapters
     settings = [
@@ -35,8 +35,8 @@ def create_fantasy_story(heros_name: str, number_of_chapter: int) -> str:
         "engaging in an epic battle that became legendary"
     ]
     
-    story = f"The Story of {arg1}:\n\n"
-    for i in range(1, arg2 + 1):
+    story = f"The Story of {heros_name}:\n\n"
+    for i in range(1, number_of_chapter + 1):
         setting = random.choice(settings)
         conflict = random.choice(conflicts)
         resolution = random.choice(resolutions)