feat: add tools to app
Browse files
app.py
CHANGED
|
@@ -2,6 +2,12 @@ from typing import NamedTuple
|
|
| 2 |
|
| 3 |
import gradio as gr
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
from tdagent.tools.get_url_content import gr_get_url_http_content
|
| 6 |
from tdagent.tools.internal_company_user_search import gr_internal_company
|
| 7 |
from tdagent.tools.lookup_company_cloud_account_information import (
|
|
@@ -28,6 +34,10 @@ TOOLS = (
|
|
| 28 |
ToolInfo("Query AbuseIPDB", gr_query_abuseipdb),
|
| 29 |
ToolInfo("Query WHOIS", gr_query_whois),
|
| 30 |
ToolInfo("Virus Total URL info", gr_virus_total_url_info),
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
## Fake tools
|
| 32 |
ToolInfo("Fake company directory", gr_internal_company),
|
| 33 |
ToolInfo(
|
|
|
|
| 2 |
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
+
from tdagent.tools.get_domain_information import (
|
| 6 |
+
dns_enumeration_tool,
|
| 7 |
+
extractor_of_ioc_from_threatfox_tool,
|
| 8 |
+
geo_location_tool,
|
| 9 |
+
scrap_subdomains_tool,
|
| 10 |
+
)
|
| 11 |
from tdagent.tools.get_url_content import gr_get_url_http_content
|
| 12 |
from tdagent.tools.internal_company_user_search import gr_internal_company
|
| 13 |
from tdagent.tools.lookup_company_cloud_account_information import (
|
|
|
|
| 34 |
ToolInfo("Query AbuseIPDB", gr_query_abuseipdb),
|
| 35 |
ToolInfo("Query WHOIS", gr_query_whois),
|
| 36 |
ToolInfo("Virus Total URL info", gr_virus_total_url_info),
|
| 37 |
+
ToolInfo("Get IP's Location", geo_location_tool),
|
| 38 |
+
ToolInfo("DNS Enumerator", dns_enumeration_tool),
|
| 39 |
+
ToolInfo("Subdomain Retriever", scrap_subdomains_tool),
|
| 40 |
+
ToolInfo("Extractor of IoCs", extractor_of_ioc_from_threatfox_tool),
|
| 41 |
## Fake tools
|
| 42 |
ToolInfo("Fake company directory", gr_internal_company),
|
| 43 |
ToolInfo(
|