Add user feedback

app.py

@@ -251,6 +251,14 @@ config firewall policy
 end\n
 """
 
+def submit_correction(original_answer, corrected_answer):
+    # No operation function for the submit button click event
+    return "Correction submitted!"
+    
+correct_answer = gr.Textbox(label="Correct Answer", placeholder="Enter the correct answer if the provided one is wrong")
+submit_btn = gr.Button("Submit Correction")
+submit_btn.click(fn=submit_correction, inputs=[chatbot, correct_answer], outputs="text")
+
 if __name__ == "__main__":
     args = parse_args()
     tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios_one_config")
@@ -275,6 +283,8 @@ if __name__ == "__main__":
             gr.Textbox(sys_prompt, label="System Prompt"),
             gr.Slider(0, 1, 0.5, label="Temperature"),
             gr.Slider(100, 2048, 1024, label="Max Tokens"),
+            correct_answer,
+            submit_btn,
         ],
         examples=[
             ["Allow all traffic from any source IP address and any source interface 'port10' to any destination IP address and any destination interface 'port9'. This policy will be applied at all times (always) and will allow all services. Additionally, this policy will enable UTM features, use proxy-based inspection mode, and use an SSL-SSH profile named 'deep-custom'. Finally, this policy will also enable source NAT."],