Update app.py

app.py

@@ -1,13 +1,11 @@
 import gradio as gr
-
-# import required libraries
 from ai21 import AI21Client
-# from ai21.logger import set_verbose
 from ai21.models.chat import ChatMessage, DocumentSchema
 import json, requests, re
 from bs4 import BeautifulSoup
+import os
 
-client = AI21Client(api_key=AI21_api_key)
+client = AI21Client(api_key=os.getenv('AI21_api_key'))
 
 cik = "0000091142" 
 headers = {'User-Agent': 'AI21 Labs [email protected]'}