Update app.py

app.py

@@ -25,7 +25,7 @@ textarea, input {
 }
 button { 
     background-color: #111111 !important; 
-    color #ffffff !important;
+    color: #ffffff !important;
     border: 1px solid #444444 !important;
     font-family: 'Courier New', monospace !important;
 }
@@ -44,7 +44,6 @@ label, p, span {
 }
 footer { display: none !important; }
 .examples { display: none !important; }
-.center-img { display: flex; justify-content: center; margin-bottom: 20px; }
 """
 
 def check_prompt(prompt):
@@ -55,57 +54,44 @@ def check_prompt(prompt):
             headers={"X-API-Key": API_KEY}
         )
         result = response.json()
-
+        
         verdict = result.get("verdict", "UNKNOWN")
         layer = result.get("layer_hit", "N/A")
         conf = result.get("confidence", 0)
         lat = result.get("latency_ms", 0)
-
+        
         display = (f"VERDICT: {verdict}\n"
                    f"LAYER  : {layer}\n"
                    f"CONF   : {conf:.2f}\n"
                    f"LATENCY: {lat:.1f}ms")
-
+        
         with open("security_audit.log", "a") as f:
             f.write(f"[{datetime.datetime.now()}] Input: {prompt} | Verdict: {verdict} | Layer: {layer}\n")
-
+            
         return display
     except Exception as e:
         return f"[SYSTEM ERROR]\n{str(e)}"
 
-
 with gr.Blocks(css=css, title="Agent Shield") as demo:
 
-    # Centered banner image
-    with gr.Row(elem_classes="center-img"):
-        with gr.Column(scale=1):
-            pass
-        with gr.Column(scale=2):
-            gr.Image(
-                value="banner.png",
-                show_label=False,
-                container=False
-            )
-        with gr.Column(scale=1):
-            pass
-
-    # Input / Output boxes
-    with gr.Row():
-        with gr.Column():
-            prompt_input = gr.Textbox(
-                lines=4,
-                placeholder="$ enter prompt...",
-                label="INPUT"
-            )
-            check_btn = gr.Button("[ SCAN ]", variant="primary")
-
-        with gr.Column():
-            output_box = gr.Textbox(
-                label="OUTPUT",
-                lines=5,
-                interactive=False
-            )
+    gr.Image(
+        value="banner.png",
+        show_label=False,
+        container=False
+    )
 
-    check_btn.click(fn=check_prompt, inputs=prompt_input, outputs=output_box)
+    gr.Interface(
+        fn=check_prompt,
+        inputs=gr.Textbox(
+            lines=4,
+            placeholder="$ enter prompt...",
+            label="INPUT"
+        ),
+        outputs=gr.Textbox(
+            label="OUTPUT",
+            lines=5
+        ),
+        description="Protects your AI",
+    )
 
 demo.launch()