Update app.py

app.py

@@ -190,8 +190,6 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Generated Names"),
-        gr.Button("Flag Output", elem_id="flag_button"),
-        gr.HTML(display_disclaimer())
     ],
     live=True,
     title='Dungen Dev - Name Generator',
@@ -205,4 +203,20 @@ demo = gr.Interface(
     ]
 )
 
+flag_ui = gr.Interface(
+    fn=flag_output,
+    inputs=[
+        gr.Textbox(label="Prompt", interactive=False),
+        gr.Textbox(label="Generated Names", interactive=False),
+        gr.Number(label="Next Words", interactive=False),
+        gr.Number(label="Temperature", interactive=False),
+    ],
+    outputs=[
+        gr.Textbox(label="Flag Status")
+    ],
+    title="Flag Output"
+)
+
+disclaimer_ui = gr.HTML(display_disclaimer())
+
 demo.launch()