Update function.py

function.py

@@ -14,6 +14,10 @@ huggingfacehub_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 config = {'max_new_tokens': 512, 'temperature': 0.6}
 
 def GetLLMResponse(selected_topic_level, selected_topic, num_quizzes):
+    # Ensure that the Hugging Face Hub API token is available
+    if huggingfacehub_api_token is None:
+        raise ValueError("HUGGINGFACEHUB_API_TOKEN environment variable is not set. Set the API token and try again.")
+
     # Initialize Hugging Face Hub with API token
     llm = HuggingFaceHub(
         repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
@@ -49,4 +53,5 @@ def GetLLMResponse(selected_topic_level, selected_topic, num_quizzes):
 
     # Return the response
     return response
+