Update app.py

app.py

@@ -4,6 +4,11 @@ import openai
 import gradio as gr
 import json
 
+# Set the flagging function
+
+HF_TOKEN = os.getenv('HF_TOKEN')
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "flagged")
+
 # Set the OpenAI API key
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
@@ -129,6 +134,10 @@ iface = gr.Interface(
             on growing your business!""",  # Article text
     thumbnail="https://assets.bigcartel.com/theme_images/101321509/IMG_6002.png",  # Thumbnail image URL
     favicon_path="https://assets.bigcartel.com/theme_images/101321509/IMG_6002.png",  # Favicon image URL
+    allow_flagging="manual",
+    flagging_options=["Inaccurate Information", "Not Applicable", "Not Helpful"],
+    flagging_callback=hf_writer
+
 )
 
 # Launch the interface with authentication