Update web_ui.py

web_ui.py

@@ -20,7 +20,7 @@ def analyze_cve_handler(cve_input, progress=gr.Progress()):
     if not cve_input.strip():
         return "Please enter a CVE ID", ""
     
-    # Step 1: Input processing
+    # Input processing
     progress(0.1, desc="Processing CVE ID format...")
     time.sleep(0.5)  # Small delay to show progress
     
@@ -30,7 +30,7 @@ def analyze_cve_handler(cve_input, progress=gr.Progress()):
         if '-' in cve_id and len(cve_id.split('-')) == 2:
             cve_id = f"CVE-{cve_id}"
     
-    # Step 2: Format validation
+    # Format validation
     progress(0.2, desc="Validating CVE format...")
     time.sleep(0.3)
     
@@ -38,7 +38,7 @@ def analyze_cve_handler(cve_input, progress=gr.Progress()):
         return f"Invalid CVE format: {cve_id}", ""
     
     try:
-        # Step 3: Check cache
+        # Check cache
         progress(0.3, desc="Checking local cache...")
         time.sleep(0.5)
         
@@ -65,7 +65,7 @@ def analyze_cve_handler(cve_input, progress=gr.Progress()):
             progress(0.8, desc="Saving analysis results...")
             time.sleep(0.5)
         
-        # Step 4: Format results
+        # Format results
         progress(0.9, desc="Formatting results...")
         time.sleep(0.3)
         
@@ -142,7 +142,7 @@ def search_handler(keyword, progress=gr.Progress()):
     except Exception as e:
         return f"Search failed: {str(e)}"
 
-# Main interface with Gradio default styling and progress support
+# Main interface 
 with gr.Blocks(title="CVE Analysis Agent", theme=gr.themes.Default(primary_hue="violet")) as interface:
     
     with gr.Column():
@@ -194,7 +194,7 @@ with gr.Blocks(title="CVE Analysis Agent", theme=gr.themes.Default(primary_hue="
             # AI Analysis below
             ai_output = gr.Markdown(
                 label="AI Security Analysis",
-                value="*AI-generated security analysis will appear here...*"
+                value="*AI-generated security analysis will appear here*"
             )
         
         # Search section