Update app.py

app.py

@@ -31,6 +31,13 @@ ip_access_records = defaultdict(lambda: {
     'last_access_time': None
 })
 
+AUTHORIZED_API_KEY = os.getenv("HF_API_KEY")
+
+def is_authorized(request: gr.Request):
+    """Checks if the request includes the correct API key."""
+    api_key = request.headers.get("API-Key")
+    return api_key == AUTHORIZED_API_KEY
+    
 SESSION_TIMEOUT = timedelta(hours=1)
 
 # Function to log API access attempts
@@ -189,9 +196,13 @@ def analyze_video(video_file, captcha_input, captcha_solution, request: gr.Reque
         }
         return output
 
+    if not is_authorized(request):
+        return {"error": "Unauthorized access. Please provide a valid API key."}
+        
     except Exception as e:
         logging.error(f"Error during analysis: {e}")
         return {"error": "An error occurred during video analysis. Please check your input and try again."}
+        
 
 # Interface with CAPTCHA after video upload
 def main_interface():