Spaces:
Runtime error
Runtime error
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(
|