Update app.py

app.py

@@ -1,10 +1,16 @@
 import gradio as gr
 import requests
-import json
 import os
 
-# Configure the endpoint
+# Configure the endpoint and authentication
 ENDPOINT_URL = os.environ.get("ENDPOINT_URL", "https://z3xyxntalfy45pxc.us-east-1.aws.endpoints.huggingface.cloud")
+HF_API_TOKEN = os.environ.get("HF_API_TOKEN")  # Get API token from environment variable
+
+# Check if the API token is configured
+def is_token_configured():
+    if not HF_API_TOKEN:
+        return "⚠️ Warning: HF_API_TOKEN is not configured. The app won't work until you add this secret in your Space settings."
+    return "✅ API token is configured"
 
 # Define the function to call your endpoint
 def check_safety(input_text):
@@ -16,9 +22,10 @@ def check_safety(input_text):
         "inputs": input_text
     }
     
-    # Set headers for JSON content
+    # Set headers with authentication token
     headers = {
-        "Content-Type": "application/json"
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {HF_API_TOKEN}"
     }
     
     try:
@@ -51,6 +58,9 @@ with gr.Blocks(title="Safety Content Classifier", css="footer {display: none !im
     gr.Markdown(f"# Safety Content Classifier")
     gr.Markdown(f"## Connected to external safety model endpoint")
     
+    # Add token status indicator
+    token_status = gr.Markdown(is_token_configured())
+    
     with gr.Accordion("About this demo", open=False):
         gr.Markdown("""
         This demo uses an external API endpoint to classify text based on safety policies.
@@ -95,4 +105,4 @@ with gr.Blocks(title="Safety Content Classifier", css="footer {display: none !im
     )
 
 # Launch the app
-demo.launch()
+demo.launch()