updated the ui

src/ui/app.py

@@ -40,7 +40,10 @@ def analyze_malware(file_path):
         prob_a = result_a["malware_prob"] * 100
         status_a = "Malicious" if result_a["is_malware"] else "Benign"
 
-        text_a = f"Status: {status_a}\nConfidence: {prob_a:.2f}%"
+        text_a = {
+            "Malicious": result_a["malware_prob"],
+            "Benign": 1.0 - result_a["malware_prob"],
+        }
 
         # Build explainability for gr.Label
         explain_dict = {}
@@ -340,14 +343,9 @@ with gr.Blocks() as app:
     gr.Markdown("---")
 
     # Single column layout for maximum horizontal space
-    ensemble_output = gr.Textbox(
-        label="Final Verdict", value="Pending", interactive=False
-    )
-    output_a = gr.Textbox(
+    ensemble_output = gr.Textbox(label="Final Verdict", interactive=False, lines=1)
+    output_a = gr.Label(
         label="Engine A: Structural (EMBER)",
-        value="Awaiting input...",
-        interactive=False,
-        lines=2,
     )
     explain_plot = gr.Label(
         label="Engine A: Explainability (Feature Impact)", num_top_classes=3