Update app.py

app.py

@@ -17,10 +17,10 @@ def generate_text(seed_text, next_words=30, temperature=0.5):
     hate_speech = detect_hate_speech(seed_text)
     profanity = detect_profanity([seed_text], language='All')
 
-    if profanity:  # Simplified check
+    if profanity:
         gr.Warning("Profanity detected in the prompt, using the default prompt.")
         seed_text = 'game name | '
-    elif hate_speech and hate_speech[0] in ['Hate Speech', 'Offensive Speech']: # Simplified check
+    elif hate_speech and hate_speech[0] in ['Hate Speech', 'Offensive Speech']:
         gr.Warning('Harmful speech detected in the seed text, using default prompt.')
         seed_text = 'game name | '
 
@@ -30,9 +30,8 @@ def generate_text(seed_text, next_words=30, temperature=0.5):
         token_list = pad_sequences([token_list], maxlen=max_seq_len - 1, padding='pre')
         predicted = model.predict(token_list, verbose=0)[0]
 
-        # Apply temperature with numerical stability fix
         predicted = np.asarray(predicted).astype("float64")
-        predicted = np.log(predicted + 1e-8) / temperature # Add small epsilon to prevent log(0)
+        predicted = np.log(predicted + 1e-8) / temperature
         exp_preds = np.exp(predicted)
         predicted = exp_preds / np.sum(exp_preds)
 
@@ -44,12 +43,16 @@ def generate_text(seed_text, next_words=30, temperature=0.5):
             break
 
     decoded = sp.decode_pieces(sp.encode_as_pieces(generated_text))
-    decoded = decoded.replace("</s>", "").replace("<unk>", "").strip() # Combined replace and strip
-    
+    decoded = decoded.replace("</s>", "").replace("<unk>", "").strip()
+
+    # Remove the prompt from the generated text
+    if '|' in decoded:
+        decoded = decoded.split('|', 1)[1].strip() #Split at the first occurence of '|' and take the second part
+
     hate_speech2 = detect_hate_speech(decoded)
     profanity2 = detect_profanity([decoded], language='All')
-    
-    if profanity2 or (hate_speech2 and hate_speech2[0] in ['Hate Speech', 'Offensive Speech']): #Combined check
+
+    if profanity2 or (hate_speech2 and hate_speech2[0] in ['Hate Speech', 'Offensive Speech']):
         gr.Warning("Flagged potentially harmful output.")
         decoded = 'Flagged Output'
 
@@ -59,12 +62,19 @@ demo = gr.Interface(
     fn=generate_text,
     inputs=[
         gr.Textbox(label="Prompt", value="a female character name", max_lines=1),
-        gr.Slider(1, 100, step=1, label='Next Words', value=30),
+        gr.Slider(1, 50, step=1, label='Next Words', value=30),
         gr.Slider(0.1, 1, value=0.5, label='Temperature', info='Controls randomness of generation, higher values = more creative, lower values = more probalistic')
     ],
-    outputs=gr.Textbox(label="Generated Names"), #Simplified output
+    outputs=gr.Textbox(label="Generated Names"),
     title='Dungen Dev - Name Generator',
-    description='A prompt-based name generator for game developers.'
+    description='A prompt-based name generator for game developers.',
+    examples=[
+        ["a male elf name"],
+        ["a futuristic city name"],
+        ["a powerful magic item name"],
+        ["a dark and mysterious forest name"],
+        ["a female character name"]
+    ]
 )
 
 demo.launch()