Update app.py

app.py

@@ -7,7 +7,7 @@
 # demo.launch()
 
 
-
+"""
 import gradio as gr
 from llama_cpp import Llama
 
@@ -68,3 +68,85 @@ demo = gr.ChatInterface(
 )
 
 demo.launch()
+"""
+import gradio as gr
+import csv, time
+
+
+# -------------------------------------------------
+# Your chatbot model
+# -------------------------------------------------
+def chat_model(message, history):
+    # Example model: reverse text
+    return message[::-1]
+
+
+# -------------------------------------------------
+# Save flagged conversation
+# -------------------------------------------------
+def flag_conversation(history, reason):
+    # Save the entire chat + reason
+    with open("chat_flags.csv", "a", newline="") as f:
+        writer = csv.writer(f)
+        writer.writerow([
+            time.time(),
+            history,   # full conversation as a Python list
+            reason
+        ])
+    return "Flag recorded!"
+
+
+# -------------------------------------------------
+# UI
+# -------------------------------------------------
+with gr.Blocks() as app:
+
+    gr.Markdown("### Chatbot with Multi-Reason Flagging")
+
+    # Create chat interface
+    chat = gr.ChatInterface(
+        fn=chat_model,
+        chatbot=gr.Chatbot(height=400),
+        textbox=gr.Textbox(placeholder="Say something..."),
+        title="Demo Chatbot",
+        description="Supports multi-reason flagging."
+    )
+
+    # Flag button
+    flag_btn = gr.Button("Flag Conversation", variant="stop")
+
+    # Modal for selecting flag reason
+    with gr.Group(visible=False) as flag_modal:
+        gr.Markdown("### Why are you flagging this conversation?")
+        reason_dd = gr.Dropdown(
+            choices=[
+                "Offensive / Toxic",
+                "Incorrect",
+                "Hallucination",
+                "Safety Issue",
+                "Biased Output",
+                "Other"
+            ],
+            label="Reason"
+        )
+        submit_flag = gr.Button("Submit Flag", variant="primary")
+        cancel_flag = gr.Button("Cancel")
+
+
+    # Show modal
+    flag_btn.click(lambda: gr.update(visible=True), None, flag_modal)
+
+    # Cancel hides modal
+    cancel_flag.click(lambda: gr.update(visible=False), None, flag_modal)
+
+    # Submit flag → save CSV → hide modal → show status
+    flag_status = gr.Textbox(label="Flagging Status", visible=True)
+
+    submit_flag.click(
+        lambda chat_history, reason: flag_conversation(chat_history, reason),
+        inputs=[chat.chatbot, reason_dd],
+        outputs=flag_status
+    ).then(lambda: gr.update(visible=False), None, flag_modal)
+
+
+app.launch()