Merge pull request #51 from morpheuslord/lammod6

Readme and code update
morpheuslord · Jul 29, 2023 · 0bebcb5 · 0bebcb5
2 parents 740e109 + ce57fca
commit 0bebcb5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -102,6 +102,7 @@ python gpt_vuln.py --target <IP> --attack dns --ai bard / openai <default>
 # Interactive step by step cli interface
 python gpt_vuln.py --menu True
 ```
+
 #### CLI Interface Option
 
 ```bash
@@ -139,10 +140,10 @@ Its same as Openai GPT3.5 but faster. It can generate the same answer but in 2 t
 
 ### OS Supported
 
-| Preview                                                                                                              | Code | Name      | Working Status | OpenAI Status | Bard Status | LLama2 Status |
-| -------------------------------------------------------------------------------------------------------------------- | ---- | --------- | --------- | --------- | --------- | --------- |
-| ![](https://raw.githubusercontent.com/EgoistDeveloper/operating-system-logos/master/src/48x48/LIN.png "LIN (48x48)") | LIN  | GNU/Linux | ✅     | ✅     | ✅     |   ❌ [did not test]  |
-| ![](https://raw.githubusercontent.com/EgoistDeveloper/operating-system-logos/master/src/48x48/WIN.png "WIN (48x48)") | WIN  | Windows   | ✅     | ✅     | ✅     |   ✅   |
+| Preview                                                                                                              | Code | Name      | Working Status | OpenAI Status | Bard Status | LLama2 Status     |
+| -------------------------------------------------------------------------------------------------------------------- | ---- | --------- | -------------- | ------------- | ----------- | ----------------- |
+| ![](https://raw.githubusercontent.com/EgoistDeveloper/operating-system-logos/master/src/48x48/LIN.png "LIN (48x48)") | LIN  | GNU/Linux | ✅             | ✅            | ✅          | ❌ [did not test] |
+| ![](https://raw.githubusercontent.com/EgoistDeveloper/operating-system-logos/master/src/48x48/WIN.png "WIN (48x48)") | WIN  | Windows   | ✅             | ✅            | ✅          | ✅                |
 
 ## Understanding the code
 
@@ -316,6 +317,8 @@ I have to thank **@thisserand** and his [llama2_local](https://github.com/thisse
 The Accuracy of the AI in offline and outside the codes test was great and had equal accuracy to openai or bard but while in code it was facing a few issues may be because of the prompting and all. I will try and fix it.
 The speed depends on your system and the GPU and CPU configs you have. currently, it is using the `TheBloke/Llama-2-7B-Chat-GGML` model and can be changed via the `portscanner` and `dnsrecon` files.
 
+For now the llama code and scans are handeled differently. After few tests I found out llama needs to be trained a little to opparate like how I intended it to work so it needs some time. Any suggestions on how I can do that can be added in the discussions of this repo [Discussions Link](https://github.com/morpheuslord/GPT_Vuln-analyzer/discussions). For now the output wont be a devided list of all the data instead will be an explaination of the vulnerability or issues discovered by the AI.
+
 ### Output
 
 nmap output:

diff --git a/gpt_vuln.py b/gpt_vuln.py
@@ -222,7 +222,7 @@ def nmap_menu() -> None:
             case "5":
                 clearscr()
                 pout: str = p_scanner(t, int(profile_num), akey_set, bkey_set, ai_set)
-                print_output("Nmap", pout)
+                print_output("Nmap", pout, ai_set)
             case "r":
                 clearscr()
                 menu_term()
@@ -290,7 +290,7 @@ def dns_menu() -> None:
             case "4":
                 clearscr()
                 dns_output: str = dnsr(t, akey_set, bkey_set, ai_set)
-                print_output("DNS", dns_output)
+                print_output("DNS", dns_output, ai_set)
             case "r":
                 clearscr()
                 menu_term()