Update app.py

app.py

@@ -52,7 +52,7 @@ def clean_name(text, title_case=True, max_repeats=2):
     text = unicodedata.normalize("NFC", text)
     text = re.sub(r'(.)\1{2,}', lambda m: m.group(1) * max_repeats, text)
     text = re.sub(r"’S|\'S", "'s", text)
-    text = re.sub(r"[^0-9A-Za-zÀ-ÖØ-öø-ÿ'’\-\s]", "", text)
+    text = re.sub(r"[^0-9A-Za-zÀ-ɏ'’\-\s]", "", text)
     text = re.sub(r"\s+", " ", text).strip()
     if title_case:
         text = proper_case(text)
@@ -79,6 +79,12 @@ def sample_once(prompt, temperature=1.0, top_k=40, max_new=24):
 
 # === Generation Function ===
 def generate_names(prompt, temperature, top_k, count, retries):
+    prompt = prompt.strip()
+    if not prompt:
+        raise gr.Error("Prompt cannot be empty.")
+    if len(prompt) > 64:
+        raise gr.Error("Prompt is too long. Please keep it under 64 characters.")
+
     results = []
     for _ in range(count):
         for attempt in range(retries):
@@ -92,12 +98,25 @@ def generate_names(prompt, temperature, top_k, count, retries):
     return df, file_path
 
 # === UI ===
-description = """# 🧠 Kaos: A Fantasy Name Generator  
+description = """# 🧠 KaosGen: A Fantasy Name Generator  
 Kaos is a small GPT-style transformer trained from scratch using character-level tokenization.  
 It excels at fantasy and mythic naming conventions.
 
-💡 Give it a prompt like `'a forgotten warrior king'`, `'priestess of the dusk sea'`, or `'demon of frost'`.  
-The model will generate fitting fantasy-style names, ready for your characters, worlds, or gods."""
+Give it a prompt like `'a forgotten warrior king'`, `'priestess of the dusk sea'`, or `'demon of frost'`.  
+It will generate names for characters, gods, factions, or places.
+
+### ⚠️ Disclaimers
+- This model may occasionally produce inaccurate, inappropriate, or nonsensical results.
+- It is a fantasy tool and **not intended for general-purpose language tasks**.
+- The creators are not responsible for any weirdness it spits out. Use responsibly.
+"""
+
+examples = [
+    ["a forgotten warrior king"],
+    ["queen of the shattered realm"],
+    ["blacksmith of shadows"],
+    ["titan of the blazing sky"]
+]
 
 with gr.Blocks() as demo:
     gr.Markdown(description)
@@ -108,11 +127,12 @@ with gr.Blocks() as demo:
             top_k = gr.Slider(10, 100, step=10, value=40, label="Top-K Sampling")
             count = gr.Slider(1, 20, step=1, value=5, label="Names to Generate")
             retries = gr.Slider(1, 5, step=1, value=3, label="Max Retries per Name")
-            generate_btn = gr.Button("🎲 Generate Names")
+            generate_btn = gr.Button("\ud83c\udfb2 Generate Names")
         with gr.Column():
             output = gr.Dataframe(headers=["Generated Name"], datatype="str", label="Generated Names", interactive=False)
-            download = gr.File(label="📥 Export Names as .txt")
+            download = gr.File(label="\ud83d\udcc5 Export Names as .txt")
 
     generate_btn.click(fn=generate_names, inputs=[prompt, temperature, top_k, count, retries], outputs=[output, download])
+    gr.Examples(examples=examples, inputs=prompt)
 
 demo.launch()