Spaces:
Sleeping
Sleeping
import os | |
import streamlit as st | |
from anthropic import Anthropic | |
st.set_page_config(page_title="Claude Test", layout="wide") | |
st.title("Claude API Test") | |
# Debug: Show all environment variables (excluding the actual values) | |
st.write("Available environment variables:", [k for k in os.environ.keys()]) | |
# Try multiple ways to get the API key | |
api_key = None | |
# Try direct environment variable | |
if 'ANTHROPIC_API_KEY' in os.environ: | |
api_key = os.environ['ANTHROPIC_API_KEY'] | |
st.success("Found API key in os.environ['ANTHROPIC_API_KEY']") | |
# Try getenv | |
if not api_key: | |
api_key = os.getenv('ANTHROPIC_API_KEY') | |
if api_key: | |
st.success("Found API key using os.getenv()") | |
# Try Streamlit secrets | |
if not api_key and hasattr(st, 'secrets') and 'ANTHROPIC_API_KEY' in st.secrets: | |
api_key = st.secrets['ANTHROPIC_API_KEY'] | |
st.success("Found API key in Streamlit secrets") | |
if not api_key: | |
st.error(""" | |
⚠️ No API key found. Please make sure: | |
1. You've added the secret in Hugging Face Space settings | |
2. The secret is named exactly 'ANTHROPIC_API_KEY' | |
3. The Space has been rebuilt after adding the secret | |
""") | |
st.stop() | |
try: | |
# Initialize Claude client | |
client = Anthropic(api_key=api_key) | |
st.success("✅ Successfully initialized Anthropic client") | |
# Simple chat interface | |
if "messages" not in st.session_state: | |
st.session_state.messages = [] | |
# Display chat history | |
for message in st.session_state.messages: | |
with st.chat_message(message["role"]): | |
st.write(message["content"]) | |
# Chat input | |
if prompt := st.chat_input("Say something"): | |
# Display user message | |
with st.chat_message("user"): | |
st.write(prompt) | |
# Add user message to history | |
st.session_state.messages.append({"role": "user", "content": prompt}) | |
# Get Claude's response | |
with st.chat_message("assistant"): | |
with st.spinner("Thinking..."): | |
response = client.messages.create( | |
model="claude-3-opus-20240229", | |
max_tokens=1000, | |
messages=[{"role": "user", "content": prompt}] | |
) | |
response_text = response.content[0].text | |
st.write(response_text) | |
# Add assistant response to history | |
st.session_state.messages.append({"role": "assistant", "content": response_text}) | |
except Exception as e: | |
st.error(f"Error: {str(e)}") | |
st.write("Please check your API key configuration") |