Update app.py

app.py

@@ -169,17 +169,11 @@ def flag_output(prompt, generated_text, next_words, temperature):
 
     return "Output flagged successfully."
 
-def push_flagged_to_dataset():
-    import datasets
-    dataset = datasets.load_dataset("InfinitodeLTD/DungenDev-FlaggedOutputs", split="train")
-    dataset = dataset.add_item(flagged_outputs)
-    dataset.push_to_hub()
-    flagged_outputs.clear()
-    return "Flagged outputs pushed to public dataset."
-
 def display_disclaimer():
-    return ('Dungen Dev is an experimental model and may produce outputs that are inappropriate, biased, or potentially harmful and inaccurate. '
-            'Caution is advised.')
+    return ('<div style="padding:10px; border:1px solid #ccc; border-radius:5px; background-color:#f9f9f9;">
+            Dungen Dev is an experimental model and may produce outputs that are inappropriate, biased, or potentially harmful and inaccurate. 
+            Caution is advised.
+            </div>')
 
 demo = gr.Interface(
     fn=generate_text,
@@ -190,8 +184,8 @@ demo = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Generated Names"),
+        gr.Button(value="Flag", elem_id="flag_button"),
     ],
-    live=True,
     title='Dungen Dev - Name Generator',
     description=None,
     examples=[
@@ -203,20 +197,6 @@ 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())
+disclaimer = gr.HTML(display_disclaimer(), elem_id="disclaimer_div")
 
 demo.launch()