Update app.py

app.py

@@ -71,26 +71,24 @@ def initialize_api_keys():
 initialize_api_keys()
 
 
-def is_openai_api_key_valid(openai_api_key: str) -> bool:
+def is_openai_api_key_valid():
     """
-    Return True if the given OpenAI API key is valid.
+    Validate the OpenAI API key from Hugging Face secrets.
     """
-
-    headers = {
-        "Authorization": f"Bearer {openai_api_key}",
-    }
-    response = requests.get(
-        "https://api.openai.com/v1/models", headers=headers
-    )
-
+    openai_api_key = os.environ.get("OPENAI_API_KEY")
+    if not openai_api_key:
+        return False
+    headers = {"Authorization": f"Bearer {openai_api_key}"}
+    response = requests.get("https://api.openai.com/v1/models", headers=headers)
     return response.status_code == 200
 
 
-def is_bing_subscription_key_valid(bing_subscription_key: str) -> bool:
+
+def is_bing_subscription_key_valid():
     """
-    Return True if the given Bing subscription key is valid.
+    Validate the Bing Subscription key from Hugging Face secrets.
     """
-
+    bing_subscription_key = os.environ.get("BING_SUBSCRIPTION_KEY")
     if not bing_subscription_key:
         return False
     try:
@@ -99,11 +97,11 @@ def is_bing_subscription_key_valid(bing_subscription_key: str) -> bool:
             bing_search_url="https://api.bing.microsoft.com/v7.0/search",
             k=1
         )
-        bing_search.run("Where can I get a Bing subscription key?")
-    except:
+        bing_search.run("Test Query")
+    except Exception:
         return False
-    else:
-        return True
+    return True
+
 
 
 def check_api_keys() -> None: