Update app.py

app.py

@@ -1,51 +1,56 @@
 import gradio as gr
 
 CLASSES = {
-    "learn" : "Sage", "grow": "Druid", "build": "Artificer",
-    "lead" : "Paladin", "create" : "Bard", "fight" : "Warrior",
-    "heal" : "Cleric", "explore": "Ranger"
+    "learn": "Sage", "grow": "Druid", "build": "Artificer",
+    "lead": "Paladin", "create": "Bard", "fight": "Warrior",
+    "heal": "Cleric", "explore": "Ranger",
 }
 
 QUESTS = {
-    "fear": "Conquer the Shadow keep of inner doubt",
-    "time": "Recover the lost Hoourglass of priorities",
-    "money": "reclaim the treasury of the forgotten kingdom",
-    "confidence": "Drink from the wellspring of the Unshakable Resolve",
-    "focus": " destroy the fog wraith scattered across the realm",
-    "motivation": "relight the ancient flame of purpose",
-    "skills" : "Apprentice under the Grandmaster hidden in the mountains",
+    "fear": "Conquer the Shadow Keep of Inner Doubt",
+    "time": "Recover the Lost Hourglass of Priorities",
+    "money": "Reclaim the Treasury of the Forgotten Kingdom",
+    "confidence": "Drink from the Wellspring of Unshakeable Resolve",
+    "focus": "Destroy the Fog Wraith scattered across the realm",
+    "motivation": "Relight the Ancient Flame of Purpose",
+    "skill": "Apprentice under the Grandmaster hidden in the mountains",
 }
 
 def generate_hero(name, goal, challenge):
     name = name.strip() or "Stranger"
     goal = goal.lower()
     challenge = challenge.lower()
+
     hero_class = next((v for k, v in CLASSES.items() if k in goal), "Adventurer")
-        quest = next((v for k, v in QUESTS.items() for k in challenge),
-                    "unravel the mystery at the end of the world ")
+
+    quest = next((v for k, v in QUESTS.items() if k in challenge),
+                 "Unravel the Mystery at the Edge of the Known World")
+
     intro = (
-        f'In the age when legends were still being written , {name} the {helo_class}'
-        f'rose from the humble origins with one burning desire: to {goal_rstrip(".")}. '
-        f' yet the path was not easy . {name} faced a great darkness , {challenge.rstrip(" . ")}. '
-        f' undettered, they set forth on the quest : to {quest:lower()}. '
-        f' The world would never be the same'
+        f'In the age when legends were still being written, {name} the {hero_class} '
+        f'rose from humble origins with one burning desire: to {goal.rstrip(".")}. '
+        f'Yet the path was not easy. {name} faced a great darkness — {challenge.rstrip(".")}. '
+        f'Undeterred, they set forth on the quest: to {quest.lower()}. '
+        f'The world would never be the same.'
     )
 
-    return hero_class, quest , intro 
+    return hero_class, quest, intro
 
-with gr.Blocks(title = "HERO GENERATOR") as app:
-    gr.markdown("This is the hero generator")
+with gr.Blocks(title="Hero Generator") as app:
+    gr.Markdown("# ⚔️ Hero Generator")
 
     with gr.Row():
-        name = gr.Textbox(label = "Name", placeholder = "your name")
-        goal = gr.Textbox(label = "Goal", placeholder = "eg. get a job ")
-    challenge = gr.Textbox(label = "Biggest Challenge", placeholder = "consistency")
+        name  = gr.Textbox(label="Name", placeholder="Your name")
+        goal  = gr.Textbox(label="Goal", placeholder="e.g. learn to code")
+    challenge = gr.Textbox(label="Biggest Challenge", placeholder="e.g. lack of focus")
+
     btn = gr.Button("Generate Hero", variant="primary")
-    hero_class = gr.Textbox(label= "Hero Class")
-    quest = gr.TextBox(label = "Main Quest")
-    story = gr.TextBox(label = "Stroy Introduction", lines = 10)
 
-    btn.click(generate_hero, inputs = [name, goal, challenge], 
-              outputs = [hero_class, quest, story])
+    hero_class = gr.Textbox(label="Hero Class")
+    quest      = gr.Textbox(label="Main Quest")
+    story      = gr.Textbox(label="Story Introduction", lines=5)
+
+    btn.click(generate_hero, inputs=[name, goal, challenge],
+              outputs=[hero_class, quest, story])
 
 app.launch()