fixed Virus total docstring

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 
 from tdagent.tools.get_url_content import gr_get_url_http_content
+from tdagent.tools.internal_company_user_search import gr_internal_company
 from tdagent.tools.letter_counter import gr_letter_counter
 from tdagent.tools.query_abuse_ip_db import gr_query_abuseipdb
 from tdagent.tools.virus_total import gr_virus_total
@@ -11,7 +12,7 @@ gr_app = gr.TabbedInterface(
         gr_letter_counter,
         gr_query_abuseipdb,
         gr_virus_total,
-
+        gr_internal_company,
     ],
 )
 

tdagent/tools/internal_company_user_search.py

@@ -0,0 +1,61 @@
+import gradio as gr
+
+# Fake user database
+users_db = {
+    "jsmith": {
+        "username": "jsmith",
+        "email": "[email protected]",
+        "name": "John Smith",
+        "user_id": "US789456",
+        "jobtitle": "Software Engineer",
+        "department": "Engineering",
+        "country": "United States"
+    },
+    "mhacker": {
+        "username": "mhacker",
+        "email": "[email protected]",
+        "name": "Maria Hacker",
+        "user_id": "US123789",
+        "jobtitle": "Security Specialist",
+        "department": "Pentests",
+        "country": "Germany"
+    }
+}
+
+
+def lookup_user(username):
+    """
+    Function to lookup user information.
+    Company User Lookup System. Enter a username to get user details.
+    Returns a formatted string with user details if found, otherwise returns error message
+    """
+    if username in users_db:
+        user = users_db[username]
+        return f"""User Information:
+Username: {user['username']}
+Email: {user['email']}
+Name: {user['name']}
+User ID: {user['user_id']}
+Job Title: {user['jobtitle']}
+Department: {user['department']}
+Country: {user['country']}"""
+    else:
+        return """User Not Found
+Username: Not found
+Email: N/A
+Name: N/A
+User ID: N/A
+Job Title: N/A
+Department: N/A
+Country: N/A"""
+
+
+# Create Gradio Interface
+gr_internal_company = gr.Interface(
+    fn=lookup_user,
+    inputs=["text"],
+    outputs=["text"],
+    title="Company User Lookup System",
+    description="Company User Lookup System. Enter a username to get user details",
+    theme="default"
+)