Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
 import datetime
+import os
 
 API_URL = "https://agent-shield-chbxh2hkhxgucgax.eastasia-01.azurewebsites.net/v1/check"
 
@@ -46,7 +47,8 @@ footer { display: none !important; }
 
 def check_prompt(prompt):
     try:
-        response = requests.post(API_URL, json={"prompt": prompt})
+        API_KEY = os.environ.get("AGENT_SHIELD_API_KEY", "")
+        response = requests.post(API_URL, json={"prompt": prompt}, headers={"X-API-Key": API_KEY})
         result = response.json()
         
         # Format the output for the UI