Update app.py

app.py

@@ -18,11 +18,34 @@ outputs.append(gr.JSON())
 
 expected_outputs = len(outputs)
 def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
+
+    # 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 provided API key is invalid!")
+
     headers = {'Content-Type': 'application/json'}
 
     payload = {"input": {}}
     
-    
     base_url = "http://0.0.0.0:7860"
     for i, key in enumerate(names):
         value = args[i]