Update web_ui.py

web_ui.py

@@ -104,93 +104,21 @@ def search_handler(keyword):
     except Exception as e:
         return f"Search failed: {str(e)}"
 
-# Custom CSS for beautiful styling
-custom_css = """
-#main-container {
-    max-width: 1200px;
-    margin: 0 auto;
-    padding: 20px;
-}
+# 移除自定義CSS，讓Gradio主題完全控制樣式
+# 這樣深色模式和淺色模式都會完美適配
 
-#title-header {
-    text-align: center;
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-    color: white;
-    padding: 30px;
-    border-radius: 15px;
-    margin-bottom: 30px;
-    box-shadow: 0 8px 32px rgba(0,0,0,0.1);
-}
-
-#main-input-card {
-    background: white;
-    padding: 25px;
-    border-radius: 12px;
-    box-shadow: 0 4px 20px rgba(0,0,0,0.08);
-    margin-bottom: 20px;
-    border: 1px solid #e1e8ed;
-}
-
-#results-container {
-    background: white;
-    padding: 20px;
-    border-radius: 12px;
-    box-shadow: 0 4px 20px rgba(0,0,0,0.08);
-    margin: 20px 0;
-    border: 1px solid #e1e8ed;
-}
-
-#analyse-btn {
-    background: linear-gradient(45deg, #667eea, #764ba2);
-    border: none;
-    color: white;
-    font-weight: 500;
-    height: 52px;
-    border-radius: 8px;
-    transition: all 0.3s ease;
-    margin-top: 20px;
-}
-
-#analyse-btn:hover {
-    transform: translateY(-2px);
-    box-shadow: 0 4px 15px rgba(102, 126, 234, 0.4);
-}
-
-.example-btn {
-    height: 52px !important;
-    margin-top: 16px !important;
-}
-
-.quick-examples {
-    background: #f8f9fa;
-    border-radius: 10px;
-    padding: 15px;
-    margin: 15px 0;
-    border: 1px solid #e9ecef;
-}
-
-.search-section {
-    background: #fff;
-    border-radius: 10px;
-    padding: 20px;
-    margin: 15px 0;
-    border: 1px solid #e1e8ed;
-    box-shadow: 0 2px 10px rgba(0,0,0,0.05);
-}
-"""
-
-# Main interface with beautiful styling
-with gr.Blocks(css=custom_css, title="CVE Analysis Agent", theme=gr.themes.Soft()) as interface:
+# Main interface with Gradio default styling (支援深色/淺色模式)
+with gr.Blocks(title="CVE Analysis Agent", theme=gr.themes.Default(primary_hue="violet")) as interface:
     
-    with gr.Column(elem_id="main-container"):
-        # Beautiful header - ONLY keep main title
-        with gr.Row(elem_id="title-header"):
+    with gr.Column():
+        # Beautiful header
+        with gr.Row():
             gr.Markdown("""
             # CVE Analysis Agent
             """)
         
         # Main input card
-        with gr.Column(elem_id="main-input-card"):
+        with gr.Column():
             gr.Markdown("### Enter CVE ID for Analysis")
             
             with gr.Row():
@@ -204,12 +132,11 @@ with gr.Blocks(css=custom_css, title="CVE Analysis Agent", theme=gr.themes.Soft(
                 analyse_btn = gr.Button(
                     "Analyse",
                     variant="primary",
-                    scale=1,
-                    elem_id="analyse-btn"
+                    scale=1
                 )
             
             # Quick examples section
-            with gr.Row(elem_classes="quick-examples"):
+            with gr.Row():
                 gr.Markdown("**Quick Examples:**")
                 example_dropdown = gr.Dropdown(
                     choices=[f"{cve} - {desc}" for cve, desc in EXAMPLE_CVES],
@@ -217,10 +144,10 @@ with gr.Blocks(css=custom_css, title="CVE Analysis Agent", theme=gr.themes.Soft(
                     value=None,
                     scale=3
                 )
-                example_btn = gr.Button("Try Example", variant="secondary", scale=1, elem_classes="example-btn")
+                example_btn = gr.Button("Try Example", variant="secondary", scale=1)
         
         # Results container
-        with gr.Column(elem_id="results-container"):
+        with gr.Column():
             gr.Markdown("### Analysis Results")
             
             # CVE Information on top
@@ -237,7 +164,7 @@ with gr.Blocks(css=custom_css, title="CVE Analysis Agent", theme=gr.themes.Soft(
         
         # Search section
         with gr.Accordion("Search Previously Analysed CVEs", open=False):
-            with gr.Column(elem_classes="search-section"):
+            with gr.Column():
                 gr.Markdown("**Search through your analysed vulnerabilities**")
                 with gr.Row():
                     search_input = gr.Textbox(