Update config

app.py

@@ -38,7 +38,7 @@ def create_agent():
     config = load_agent_config()
     prompts = load_prompts()
     
-    # Configure model a
+    # Configure model
     model_config = config['agent_config']['model']
     model = HfApiModel(
         model_id=model_config['model_id'],
@@ -59,20 +59,13 @@ def create_agent():
     
     return agent, prompts
 
-def process_query(query, analysis_type="general"):
-    """Process a user query"""
+def process_query(message, history):
+    """Process a user query in chat format"""
     agent, prompts = create_agent()
     
-    # Select appropriate template
-    if analysis_type == "vulnerability":
-        template = prompts['vulnerability_analysis']
-        formatted_prompt = template.format(cve_id=query)
-    elif analysis_type == "threat":
-        template = prompts['threat_report']
-        formatted_prompt = template.format(target=query)
-    else:
-        template = prompts['user_prompt']
-        formatted_prompt = template.format(query=query)
+    # Format the prompt
+    template = prompts['user_prompt']
+    formatted_prompt = template.format(query=message)
     
     # Execute agent
     system_prompt = prompts['system_prompt']
@@ -80,32 +73,30 @@ def process_query(query, analysis_type="general"):
     
     return result
 
-# Gradio Interface
+# Gradio Chat Interface
 def create_interface():
-    """Create the Gradio user interface"""
+    """Create the Gradio chat interface"""
     with gr.Blocks(title="Vulnerability Intelligence Agent") as interface:
         gr.Markdown("# Vulnerability Intelligence Agent (VIA)")
         
-        with gr.Row():
-            with gr.Column():
-                query_input = gr.Textbox(
-                    label="Query",
-                    placeholder="Enter your security query..."
-                )
-                analysis_type = gr.Radio(
-                    choices=["general", "vulnerability", "threat"],
-                    label="Analysis Type",
-                    value="general"
-                )
-                submit_btn = gr.Button("Analyze")
-            
-            with gr.Column():
-                output = gr.Markdown(label="Result")
+        chatbot = gr.Chatbot(
+            [],
+            elem_id="chatbot",
+            bubble_full_width=False,
+            avatar_images=(None, "🤖"),
+            height=600,
+        )
+        
+        txt = gr.Textbox(
+            show_label=False,
+            placeholder="Enter your security query...",
+            container=False
+        )
         
-        submit_btn.click(
-            fn=process_query,
-            inputs=[query_input, analysis_type],
-            outputs=output
+        txt.submit(
+            process_query,
+            [txt, chatbot],
+            [chatbot]
         )
     
     return interface

tools/vuln_search.py

@@ -14,11 +14,21 @@ class VulnerabilitySearchTool(Tool):
     name = "vuln_search"
     description = "Search for vulnerabilities in security databases (NVD, MISP if configured)"
     inputs = {
-        'query': {'type': 'str', 'description': 'Search term or CVE ID'},
-        'sources': {'type': 'list', 'description': 'List of sources to query (nvd, misp)', 'default': ['nvd']},
-        'max_results': {'type': 'int', 'description': 'Maximum number of results per source', 'default': 5}
+        'query': {'type': 'string', 'description': 'Search term or CVE ID'},
+        'sources': {
+            'type': 'array', 
+            'description': 'List of sources to query (nvd, misp)', 
+            'default': ['nvd'],
+            'nullable': True
+        },
+        'max_results': {
+            'type': 'integer', 
+            'description': 'Maximum number of results per source', 
+            'default': 5,
+            'nullable': True
+        }
     }
-    output_type = Dict[str, Any]
+    output_type = "object"
 
     def __init__(self):
         """Initialize API connections"""