Update app.py

app.py

@@ -1,3 +1,4 @@
+from huggingface_hub import HfApi
 import gradio as gr
 from urllib.parse import urlparse
 import requests
@@ -51,7 +52,29 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
     headers = {'Content-Type': 'application/json'}
 
     payload = {"input": {}}
+
+    # TODO: extract the Bearer access token from the request
+    if not request:
+        raise gr.Error("The submission failed!")
+
+    print("Request headers dictionary:", request.headers)
+
+    try:
+        authorization = request.headers["Authorization"]
+    except KeyError:
+        raise gr.Error("Missing authorization in the headers")
     
+    # Extract the token part from the authorization
+    try:
+        bearer, token = authorization.split(" ")
+    except ValueError:
+        raise gr.Error("Invalid format for Authorization header. It should be 'Bearer <token>'")
+
+    try:
+        hf_api = HfApi(token=token)
+        userInfo = hf_api.whoami(token)
+        if not userInfo:
+            raise gr.Error("The provider API key is invalid!")
     
     base_url = "http://0.0.0.0:7860"
     for i, key in enumerate(names):
@@ -62,7 +85,6 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
             payload["input"][key] = value
 
     response = requests.post("http://0.0.0.0:5000/predictions", headers=headers, json=payload)
-
     
     if response.status_code == 201:
         follow_up_url = response.json()["urls"]["get"]