Added Virus total interface

tdagent/tools/virus_total.py

@@ -0,0 +1,54 @@
+import gradio as gr
+import vt
+import os
+from datetime import datetime
+
+# Get API key from environment variable
+API_KEY = os.getenv('VT_API_KEY')
+
+
+def get_url_info(url):
+    try:
+        # Initialize the client
+        client = vt.Client(API_KEY)
+
+        # URL ID is created by computing the base64-encoded SHA-256 of the URL
+        url_id = vt.url_id(url)
+
+        # Get the URL analysis
+        url_analysis = client.get_object(f"/urls/{url_id}")
+
+        # Format the results
+        last_analysis_stats = url_analysis.last_analysis_stats
+        last_analysis_date = datetime.utcfromtimestamp(url_analysis.last_analysis_date).strftime(
+            '%Y-%m-%d %H:%M:%S UTC')
+
+        result = f"""
+URL: {url}
+Last Analysis Date: {last_analysis_date}
+
+Analysis Statistics:
+- Harmless: {last_analysis_stats['harmless']}
+- Malicious: {last_analysis_stats['malicious']}
+- Suspicious: {last_analysis_stats['suspicious']}
+- Undetected: {last_analysis_stats['undetected']}
+- Timeout: {last_analysis_stats['timeout']}
+
+Reputation Score: {url_analysis.reputation}
+Times Submitted: {url_analysis.times_submitted}
+        """
+
+        client.close()
+        return result
+
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+
+gr_query_abuseipdb = gr.Interface(
+    fn=get_url_info,
+    inputs=["text"],
+    outputs="text",
+    title="VirusTotal URL Scanner",
+    description="Get URL Info from VirusTotal URL Scanner. Scan URL is not available",
+)