Update variables.py

variables.py

@@ -1,15 +1,21 @@
 import json
 import os
 
-# Load JSON data from environment variable and parse it
-json_data = os.getenv('PROMPT_TEMPLATES', '{}')
-prompt_data = json.loads(json_data)
+# Load templates from environment variable with a safe default
+templates_json = os.getenv('PROMPT_TEMPLATES', '{}')
 
-# Create dictionaries from the JSON data
+try:
+    # Parse JSON data with error handling
+    prompt_data = json.loads(templates_json)
+except json.JSONDecodeError:
+    # Fallback to empty dict if JSON is invalid
+    prompt_data = {}
+
+# Create explanations dictionary with safe access
 metaprompt_explanations = {
-    key: data["description"] 
+    key: data.get("description", "No description available")
     for key, data in prompt_data.items()
-}
+} if prompt_data else {}
 
 # Generate markdown explanation
 explanation_markdown = "".join([
@@ -19,7 +25,6 @@ explanation_markdown = "".join([
 
 # Define models list
 models = [
-    # Meta-Llama models (all support system)
     "meta-llama/Meta-Llama-3-70B-Instruct",
     "meta-llama/Meta-Llama-3-8B-Instruct",
     "meta-llama/Llama-3.1-70B-Instruct",
@@ -28,25 +33,21 @@ models = [
     "meta-llama/Llama-3.2-1B-Instruct",
     "meta-llama/Llama-2-13b-chat-hf",
     "meta-llama/Llama-2-7b-chat-hf",
-    
-    # HuggingFaceH4 models (support system)
     "HuggingFaceH4/zephyr-7b-beta",
     "HuggingFaceH4/zephyr-7b-alpha",
-    
-    # Qwen models (support system)
     "Qwen/Qwen2.5-72B-Instruct",
     "Qwen/Qwen2.5-1.5B",
-
     "microsoft/Phi-3.5-mini-instruct"
 ]
 
-# Check for API token
+# Get API token with error handling
 api_token = os.getenv('HF_API_TOKEN')
 if not api_token:
     raise ValueError("HF_API_TOKEN not found in environment variables")
 
-# Store templates in a dictionary
+# Create meta_prompts dictionary with safe access
 meta_prompts = {
-    key: data["template"]
+    key: data.get("template", "No template available")
     for key, data in prompt_data.items()
-}
+} if prompt_data else {}
+