Update app.py

app.py

@@ -21,9 +21,11 @@ def safe_rerun():
 openai_client = None
 genai_client = None  # Using 'genai_client' to store the configured module/client
 deepseek_api_key = None
-secrets_available = {"openai": False, "gemini": False, "deepseek": False}
+claude_api_key = None
+secrets_available = {"openai": False, "gemini": False, "deepseek": False, "claude": False}
 secret_errors = []  # Store errors for display
 
+# OpenAI API Key
 try:
     openai_api_key = st.secrets.get("OPENAI_API_KEY")
     if openai_api_key:
@@ -36,6 +38,7 @@ except KeyError:
 except Exception as e:
     secret_errors.append(f"Error initializing OpenAI client: {e}")
 
+# Gemini API Key (Google GenAI)
 try:
     gemini_api_key = st.secrets.get("GEMINI_API_KEY")
     if gemini_api_key:
@@ -49,6 +52,7 @@ except KeyError:
 except Exception as e:
     secret_errors.append(f"Error initializing Google GenAI client: {e}")
 
+# DeepSeek API Key
 try:
     deepseek_api_key = st.secrets.get("DEEPSEEK_API_KEY")
     if deepseek_api_key:
@@ -60,12 +64,25 @@ except KeyError:
 except Exception as e:
     secret_errors.append(f"Error reading DeepSeek API key: {e}")
 
+# CLAUDE API Key (Anthropic)
+try:
+    claude_api_key = st.secrets.get("CLAUDE_API_KEY")
+    if claude_api_key:
+        secrets_available["claude"] = True
+    else:
+        secret_errors.append("Streamlit Secret `CLAUDE_API_KEY` not found.")
+except KeyError:
+    secret_errors.append("Streamlit Secret `CLAUDE_API_KEY` not found.")
+except Exception as e:
+    secret_errors.append(f"Error reading CLAUDE API key: {e}")
+
 # Check if at least one API key is loaded
 any_secret_loaded = any(secrets_available.values())
 
 # ---------- Model Configuration ----------
 SUPPORTED_MODELS = {}
 
+# OpenAI Models
 if secrets_available["openai"] and openai_client:
     SUPPORTED_MODELS.update({
         "GPT-4o (OpenAI)": {"id": "gpt-4o", "provider": "openai", "client": openai_client},
@@ -77,6 +94,7 @@ if secrets_available["openai"] and openai_client:
         "ChatGPT o3 (OpenAI)": {"id": "chatgpt-o3", "provider": "openai", "client": openai_client},
     })
 
+# Gemini Models (Google)
 if secrets_available["gemini"] and genai_client:
     SUPPORTED_MODELS.update({
         "Gemini 2.0 (Google)": {"id": "gemini-2.0-latest", "provider": "gemini", "client": genai_client},
@@ -86,12 +104,20 @@ if secrets_available["gemini"] and genai_client:
         "Gemini 1.0 Pro (Google)": {"id": "gemini-1.0-pro", "provider": "gemini", "client": genai_client},
     })
 
+# DeepSeek Models
 if secrets_available["deepseek"] and deepseek_api_key:
     SUPPORTED_MODELS.update({
         "DeepSeek Chat": {"id": "deepseek-chat", "provider": "deepseek", "client": None},
         "DeepSeek Coder": {"id": "deepseek-coder", "provider": "deepseek", "client": None},
     })
 
+# Claude Models (Anthropic) – Using the CLAUDE_API_KEY
+if secrets_available["claude"] and claude_api_key:
+    SUPPORTED_MODELS.update({
+        "Claude 3.7 Sonnet (Anthropic)": {"id": "claude-3-7-sonnet-20250219", "provider": "claude", "client": None},
+        "Claude 3.5 Haiku (Anthropic)": {"id": "claude-3-5-haiku-20241022", "provider": "claude", "client": None},
+    })
+
 DEFAULT_MODEL_PREFERENCE = [
     "GPT-4o Mini (OpenAI)",
     "Gemini 1.5 Flash (Google)",
@@ -102,7 +128,7 @@ DEFAULT_MODEL = next((m for m in DEFAULT_MODEL_PREFERENCE if m in SUPPORTED_MODE
 if not DEFAULT_MODEL and SUPPORTED_MODELS:
     DEFAULT_MODEL = next(iter(SUPPORTED_MODELS))
 
-# ---------- Helper Functions ----------
+# ---------- Helper Functions for Generation ----------
 
 def _generate_with_openai_provider(client, model_id, prompt, max_tokens):
     try:
@@ -177,6 +203,35 @@ def _generate_with_deepseek_provider(api_key, model_id, prompt, max_tokens):
         st.error(f"❌ DeepSeek Error processing response ({model_id}): {e}")
         return f"Error: DeepSeek processing failed for {model_id}. Details: {e}"
 
+def _generate_with_claude_provider(api_key, model_id, prompt, max_tokens):
+    headers = {
+        "Content-Type": "application/json",
+        "x-api-key": api_key
+    }
+    # Anthropic's API expects a prompt starting with "Human:" and the assistant's reply follows after "Assistant:"
+    payload = {
+        "model": model_id,
+        "prompt": f"Human: {prompt}\n\nAssistant:",
+        "max_tokens_to_sample": max_tokens,
+        "stop_sequences": ["\nHuman:"],
+        "temperature": 0.7
+    }
+    try:
+        response = requests.post("https://api.anthropic.com/v1/complete", headers=headers, json=payload, timeout=90)
+        response.raise_for_status()
+        response_data = response.json()
+        if "completion" in response_data:
+            return response_data["completion"]
+        else:
+            st.warning(f"Claude returned unexpected response structure: {response_data}")
+            return f"Error: Claude returned an unexpected structure."
+    except requests.exceptions.RequestException as e:
+        st.error(f"❌ Claude API Request Error ({model_id}): {e}")
+        return f"Error: Claude API request failed for {model_id}. Details: {e}"
+    except Exception as e:
+        st.error(f"❌ Claude Error processing response ({model_id}): {e}")
+        return f"Error: Claude processing failed for {model_id}. Details: {e}"
+
 def generate_with_selected_model(selected_model_name, prompt, max_tokens=2000):
     if not any_secret_loaded or not SUPPORTED_MODELS:
         return "Error: No API keys loaded or models available."
@@ -210,6 +265,11 @@ def generate_with_selected_model(selected_model_name, prompt, max_tokens=2000):
             result = f"Error: DeepSeek API key not available for {selected_model_name}."
         else:
             result = _generate_with_deepseek_provider(deepseek_api_key, model_id, prompt, max_tokens)
+    elif provider == "claude":
+        if not claude_api_key:
+            result = f"Error: CLAUDE_API_KEY not available for {selected_model_name}."
+        else:
+            result = _generate_with_claude_provider(claude_api_key, model_id, prompt, max_tokens)
     else:
         st.error(f"Unknown provider '{provider}' configured for model '{selected_model_name}'.")
         result = f"Error: Unknown provider {provider}."