RodDoSanz commited on
Commit
35aff7c
·
1 Parent(s): 1fbf466

feat: add tools to app

Browse files
Files changed (1) hide show
  1. app.py +10 -0
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(