Improve Anthropic API error handling

src/app.py

@@ -28,6 +28,7 @@ from src.constants import (
     VOTE_FOR_OPTION_TWO
 )
 from src.integrations import (
+    AnthropicError,
     generate_text_with_claude, 
     text_to_speech_with_hume, 
     text_to_speech_with_elevenlabs
@@ -50,23 +51,44 @@ def validate_and_generate_text(prompt: str) -> tuple[Union[str, gr.update], gr.u
             - The generated text or an error message.
             - The updated state of the "Generate" button.
     """
+    # Local prompt validation
     try:
-        validate_prompt_length(prompt, PROMPT_MAX_LENGTH, PROMPT_MIN_LENGTH)  # Raises error if invalid
+        validate_prompt_length(prompt, PROMPT_MAX_LENGTH, PROMPT_MIN_LENGTH)
     except ValueError as ve:
         logger.warning(f'Validation error: {ve}')
-        return str(ve), gr.update(interactive=True)  # Show error, re-enable button
+        # Show validation error to user, re-enable the "Generate" button
+        return str(ve), gr.update(interactive=True)
 
+    # Call the Anthropic API to generate text
     try:
         generated_text = generate_text_with_claude(prompt)
-        if not generated_text:
-            raise ValueError("Claude API returned an empty response.")
+
+        # Optionally handle empty or unusual responses
+        if not generated_text.strip():
+            logger.warning("Anthropic API returned an empty response.")
+            return "Error: Anthropic API returned an empty response.", gr.update(interactive=True)
 
         logger.info(f'Generated text ({len(generated_text)} characters).')
-        return gr.update(value=generated_text), gr.update(interactive=False)  # Keep button disabled
+        return gr.update(value=generated_text), gr.update(interactive=False)
+
+    # Handle Anthropic-specific errors
+    except AnthropicError as ae:
+        # You can log the internal error details
+        logger.error(f'AnthropicError while generating text: {str(ae)}')
+        # Return an error message about Anthropic API failing to generate text, re-enable the "Generate" button
+        return (
+            "Error: There was an issue communicating with the Anthropic API. Please try again later.",
+            gr.update(interactive=True),
+        )
 
+    # Catch any other unexpected exceptions
     except Exception as e:
-        logger.error(f'Error while generating text with Claude API: {e}')
-        return "Error: Failed to generate text. Please try again.", gr.update(interactive=True)  # Re-enable button
+        logger.error(f'Unexpected error while generating text: {e}')
+        # Return a generic catch-all error message, re-enable the "Generate" button
+        return (
+            "Error: Failed to generate text. Please try again.",
+            gr.update(interactive=True),
+        )
 
 
 

src/integrations/__init__.py

@@ -1,3 +1,3 @@
-from .anthropic_api import generate_text_with_claude
-from .hume_api import text_to_speech_with_hume
-from .elevenlabs_api import text_to_speech_with_elevenlabs
+from .anthropic_api import generate_text_with_claude, AnthropicError
+from .elevenlabs_api import text_to_speech_with_elevenlabs
+from .hume_api import text_to_speech_with_hume

src/integrations/anthropic_api.py

@@ -97,10 +97,11 @@ anthropic_config = AnthropicConfig()
     wait=wait_fixed(2),
     before=before_log(logger, logging.DEBUG),
     after=after_log(logger, logging.DEBUG),
+    reraise=True
 )
 def generate_text_with_claude(prompt: str) -> str:
     """
-    Generates text using Claude via the Anthropic SDK.
+    Generates text using Claude (Anthropic LLM) via the Anthropic SDK.
 
     Args:
         prompt (str): The input prompt for Claude.
@@ -109,18 +110,11 @@ def generate_text_with_claude(prompt: str) -> str:
         str: The generated text.
 
     Raises:
-        ValueError: If the prompt exceeds the maximum allowed length.
         AnthropicError: If there is an error communicating with the Anthropic API.
-    
-    Example:
-        >>> generate_text_with_claude("Write a haiku about nature.")
-        "Gentle waves crashing, / Whispering secrets softly, / Infinite blue skies."
-
-        >>> generate_text_with_claude("")
-        "The prompt exceeds the maximum allowed length of 500 characters. Your prompt contains 512 characters."
     """
     logger.debug(f'Generating text with Claude. Prompt length: {len(prompt)} characters.')
 
+    response = None
     try:
         response: Message = anthropic_config.client.messages.create(
             model=anthropic_config.model,
@@ -137,7 +131,6 @@ def generate_text_with_claude(prompt: str) -> str:
 
         # Process response content
         blocks: Union[List[TextBlock], TextBlock, None] = response.content
-
         if isinstance(blocks, list):
             result = '\n\n'.join(block.text for block in blocks if isinstance(block, TextBlock))
             logger.debug(f'Processed response from list: {truncate_text(result)}')
@@ -148,12 +141,12 @@ def generate_text_with_claude(prompt: str) -> str:
 
         logger.warning(f'Unexpected response type: {type(blocks)}')
         return str(blocks or 'No content generated.')
-        
+
     except Exception as e:
-        logger.exception(f'Error generating text with Claude: {e}')
+        logger.exception(f'Error generating text with Anthropic: {e}')
         raise AnthropicError(
             message=(
-                f'Error generating text with Claude: {e}. '
+                f'Error generating text with Anthropic: {e}. '
                 f'HTTP Status: {getattr(response, "status", "N/A")}. '
                 f'Prompt (truncated): {truncate_text(prompt)}. '
                 f'Model: {anthropic_config.model}, Max tokens: {anthropic_config.max_tokens}'