Update app.py

app.py

@@ -138,7 +138,7 @@ def generate_text(seed_text, next_words=30, temperature=0.5):
         gr.Warning("Flagged potentially harmful output.")
         decoded = 'Flagged Output'
 
-    return decoded
+    return {"text": decoded, "flag_button": True}
 
 flagged_outputs = []
 
@@ -177,11 +177,10 @@ demo = gr.Interface(
         gr.Slider(0.1, 1, value=0.5, label='Temperature', info='Controls randomness of generation, higher values = more creative, lower values = more probabilistic')
     ],
     outputs=[
-        gr.Textbox(label="Generated Names"),
-        gr.Button(value="Flag", elem_id="flag_button"),
+        gr.JSON(label="Generated Names with Flag Button")
     ],
     title='Dungen Dev - Name Generator',
-    description='A prompt-based name generator for game developers. Dungen Dev is an experimental model, and may produce outputs that are inappropriate, biased, or potentially harmful and inaccurate. Caution is advised.',
+    description=None,
     examples=[
         ["a male character name", 30, 0.5],
         ["a futuristic city name", 30, 0.5],