Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
 import requests
 import datetime
+import os
 
 API_URL = "https://agent-shield-chbxh2hkhxgucgax.eastasia-01.azurewebsites.net/v1/check"
+API_KEY = os.environ.get("AGENT_SHIELD_API_KEY", "")
 
 css = """
 body { background-color: #000000 !important; }
@@ -46,10 +48,13 @@ footer { display: none !important; }
 
 def check_prompt(prompt):
     try:
-        response = requests.post(API_URL, json={"prompt": prompt})
+        response = requests.post(
+            API_URL,
+            json={"prompt": prompt},
+            headers={"X-API-Key": API_KEY}
+        )
         result = response.json()
         
-        # Format the output for the UI
         verdict = result.get("verdict", "UNKNOWN")
         layer = result.get("layer_hit", "N/A")
         conf = result.get("confidence", 0)
@@ -62,7 +67,6 @@ def check_prompt(prompt):
                    f"--- RAW METADATA ---\n"
                    f"{result.get('details', 'No details available')}")
         
-        # LOGGING: Append every test to a local file for your portfolio
         with open("security_audit.log", "a") as f:
             f.write(f"[{datetime.datetime.now()}] Input: {prompt} | Verdict: {verdict} | Layer: {layer}\n")