Update app.py

app.py

@@ -189,7 +189,14 @@ def CogMapAnalysis(text):
         plot = None  # Use None instead of empty string for non-existent objects
         return formatted_result, plot
 
+
 # Create the GUI using the 'gr' library
+
+def handle_flag(text, description):
+    # Placeholder for handling flags; log to a file or send to an endpoint
+    print(f"Flagged by user: {text}, Reason: {description}")
+    return "Thank you for your feedback!"
+
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown('<div style="text-align: center;"><h1><strong>CogMApp</strong></h1></div> <div style="text-align: center;"><h3>Generate cognitive maps from text with one click!</h3></div>')
@@ -211,20 +218,25 @@ with gr.Blocks() as demo:
     with gr.Row():
         output = gr.Textbox(label="CogMap", lines=1, placeholder=" ")
         cogmap_plot = gr.Plot(label="Visualization")
-
+        flag_description = gr.Textbox(label="Describe the issue", lines=2, placeholder="What's wrong?")
+        flag_button = gr.Button("Flag Issue")
+    
     with gr.Row():
         gr.Markdown("⚠️  Feel free to flag me if you find any errors. :)")
-    
+
     with gr.Row():
         gr.Markdown('<p style="text-align: center; ">Demo made with ❤ by P.K. Ningrum (2024) | Contact: https://ningrumdaud.github.io/ </p>')
 
-    # Function to handle loading examples
+    # Functions for handling inputs and examples
     def load_example(example_index):
         inputs.value = examples[example_index]
 
+    # Link functions to interface elements
     example_dropdown.change(load_example, inputs=[example_dropdown], outputs=[inputs])
     submit_button.click(CogMapAnalysis, inputs=[inputs], outputs=[output, cogmap_plot])
+    flag_button.click(handle_flag, inputs=[inputs, flag_description], outputs=[])
 
 if __name__ == "__main__":
     demo.launch(show_api=False, share=True)
 
+