Firewall CLI config

app.py

@@ -56,13 +56,23 @@ def predict(message, history, system_prompt, temperature, max_tokens):
         outputs.append(text)
         yield "".join(outputs)
 
-
+"""
+        examples=[
+            ["How can you move a policy by policy ID?"],
+            ["What is the command to enable security profiles in a firewall policy?"],
+            ["How do you configure a service group in the GUI?"],
+            ["How can you configure the firewall policy change summary in the CLI?"],
+            ["How do you disable hardware acceleration for an IPv4 firewall policy in the CLI?"],
+            ["How can you enable WAN optimization in a firewall policy using the CLI?"],
+            ["What are services in FortiOS and how are they used in firewall policies?"],
+        ],
+        """
 
 if __name__ == "__main__":
     args = parse_args()
-    tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios_cli")
+    tokenizer = AutoTokenizer.from_pretrained("lliu01/fortios")
     model = AutoModelForCausalLM.from_pretrained(
-        "lliu01/fortios_cli",
+        "lliu01/fortios",
         torch_dtype=torch.bfloat16,
         low_cpu_mem_usage=True
     )
@@ -79,18 +89,18 @@ if __name__ == "__main__":
         undo_btn="Delete Previous",
         clear_btn="Clear",
         additional_inputs=[
-            gr.Textbox("FortiOS firewall policy configuration.", label="System Prompt"),
+            gr.Textbox("FortiOS CLI firewall policy configuration.", label="System Prompt"),
             gr.Slider(0, 1, 0.5, label="Temperature"),
             gr.Slider(100, 2048, 1024, label="Max Tokens"),
         ],
         examples=[
-            ["How can you move a policy by policy ID?"],
-            ["What is the command to enable security profiles in a firewall policy?"],
-            ["How do you configure a service group in the GUI?"],
-            ["How can you configure the firewall policy change summary in the CLI?"],
-            ["How do you disable hardware acceleration for an IPv4 firewall policy in the CLI?"],
-            ["How can you enable WAN optimization in a firewall policy using the CLI?"],
-            ["What are services in FortiOS and how are they used in firewall policies?"],
+            ["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."],
+            ["Configure a firewall policy to allow users 'dina' and '15947' to access 'DR-Exchange-Servers' and 'HQ-Exchange-Servers' using RDP protocol from the 'SSL-VPN-IT-Pool' address range, incoming from the 'ssl.FG-Traffic' interface and outgoing to the 'FG-PA-Inside' interface. The policy should have Antivirus scanning enabled with profile 'ABE_AV' and log all traffic. The policy should be always active and currently disabled for testing or maintenance purposes."],
+            ["Configure a firewall policy named 'ZoomAccess' that allows traffic from the 'IP_10.96.54.149' and 'HighCourt_Zoom' addresses coming in through the 'VLAN51' interface to access the 'Zoom_access' destination through the 'npu0_vlink1' interface, at any time, with all services allowed, using proxy-based inspection and SSL certificate inspection."],
+            ["I want to enable packet capture for firewall policy 22, so that packets matching this policy are captured for troubleshooting and analysis purposes."],
+            ["I want to configure the firewall policy to enable TCP NPU session delay, ensuring that the packet order of the 3-way handshake is guaranteed."],
+            ["I want to configure a firewall policy with the inspection mode set to 'flow' to enable flow-based inspection, allowing the firewall to track the state of network flows and make decisions based on that state."],
+            ["I want to enable the Security Group Tags (SGT) check in firewall policy 52."],
         ],
         additional_inputs_accordion_name="Parameters",
     ).queue().launch()