Update utils.py

utils.py

@@ -17,8 +17,9 @@ from html import escape
 
 # === Env & Clients ===
 load_dotenv()
-openai_api_key = os.getenv("OPENAI_API_KEY")
-anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
+# Strip whitespace/newlines from API keys (common issue with env vars)
+openai_api_key = os.getenv("OPENAI_API_KEY", "").strip()
+anthropic_api_key = os.getenv("ANTHROPIC_API_KEY", "").strip()
 
 # Both clients read keys from env by default; explicit is also fine:
 openai_client = OpenAI(api_key=openai_api_key) if openai_api_key else OpenAI()
@@ -28,6 +29,9 @@ anthropic_client = Anthropic(api_key=anthropic_api_key) if anthropic_api_key els
 def get_response(model: str, prompt: str) -> str:
     """Get response from LLM (OpenAI or Anthropic)."""
     if "claude" in model.lower() or "anthropic" in model.lower():
+        # Check if Anthropic API key is available
+        if not anthropic_api_key:
+            raise ValueError("ANTHROPIC_API_KEY not set. Please add it as a secret in HuggingFace Spaces settings.")
         # Anthropic Claude format
         message = anthropic_client.messages.create(
             model=model,
@@ -36,6 +40,9 @@ def get_response(model: str, prompt: str) -> str:
         )
         return message.content[0].text
     else:
+        # Check if OpenAI API key is available
+        if not openai_api_key:
+            raise ValueError("OPENAI_API_KEY not set. Please add it as a secret in HuggingFace Spaces settings.")
         # Default to OpenAI format for all other models (gpt-4, o3-mini, o1, etc.)
         response = openai_client.responses.create(
             model=model,
@@ -87,6 +94,8 @@ def image_anthropic_call(model_name: str, prompt: str, media_type: str, b64: str
     Call Anthropic Claude (messages.create) with text+image and return *all* text blocks concatenated.
     Adds a system message to enforce strict JSON output.
     """
+    if not anthropic_api_key:
+        raise ValueError("ANTHROPIC_API_KEY not set. Please add it as a secret in HuggingFace Spaces settings.")
     msg = anthropic_client.messages.create(
         model=model_name,
         max_tokens=2000,
@@ -114,6 +123,8 @@ def image_anthropic_call(model_name: str, prompt: str, media_type: str, b64: str
 
 def image_openai_call(model_name: str, prompt: str, media_type: str, b64: str) -> str:
     """Call OpenAI with text+image input."""
+    if not openai_api_key:
+        raise ValueError("OPENAI_API_KEY not set. Please add it as a secret in HuggingFace Spaces settings.")
     data_url = f"data:{media_type};base64,{b64}"
     resp = openai_client.responses.create(
         model=model_name,