Improve API key handling and add test file

README.md

@@ -20,22 +20,37 @@ A Streamlit application that helps users practice difficult conversations based
 - Customizable conversation scenarios
 - AI-powered dialogue using Claude
 
-## Setup
-1. Install dependencies:
-```bash
-pip install -r requirements.txt
-```
+## API Key Setup
 
-2. Set up environment variables:
-Create a `.env` file with your Anthropic API key:
+### Local Development
+1. Create a `.env` file in the root directory
+2. Add your Anthropic API key:
 ```
 ANTHROPIC_API_KEY=your_api_key_here
 ```
 
-3. Run the app:
+### Hugging Face Spaces
+1. Go to your Space's Settings
+2. Under "Repository secrets", add:
+   - Name: `ANTHROPIC_API_KEY`
+   - Value: Your Anthropic API key
+
+### Testing API Key
+To verify your API key is working:
+1. Run the test app: `streamlit run src/test_api.py`
+2. If successful, you'll see "✅ API key works!"
+3. If there's an error, check your API key configuration
+
+## Running the App
+1. Install dependencies:
+```bash
+pip install -r requirements.txt
+```
+
+2. Run the main app:
 ```bash
 streamlit run src/streamlit_app.py
 ```
 
 ## Deployment
-The app is configured for deployment on Hugging Face Spaces. Make sure to set the `ANTHROPIC_API_KEY` in the Space's secrets.
+The app is configured for deployment on Hugging Face Spaces. Make sure to set the `ANTHROPIC_API_KEY` in the Space's secrets before deploying.

src/streamlit_app.py

@@ -3,27 +3,43 @@ import streamlit as st
 from anthropic import Anthropic
 from dotenv import load_dotenv
 
-# Load environment variables from .env file if it exists (local development)
-if os.path.exists(".env"):
-    load_dotenv()
-
-# Get API key from environment variables (works for both local and Hugging Face Spaces)
-api_key = os.getenv('ANTHROPIC_API_KEY')
-if not api_key:
-    st.error("Please set your ANTHROPIC_API_KEY in the environment variables or .env file")
-    st.stop()
-
-# Configure page
+# Initialize page configuration first
 st.set_page_config(
     page_title="Attachment Style Roleplay Simulator",
     page_icon="🗣️",
     layout="wide"
 )
 
-# Initialize Anthropic client
-c = Anthropic(
-    api_key=api_key,
-)
+# Handle API key
+try:
+    # Load from .env file for local development
+    if os.path.exists(".env"):
+        load_dotenv()
+    
+    # Get API key from environment
+    api_key = os.getenv('ANTHROPIC_API_KEY')
+    if not api_key:
+        st.error("""
+        ⚠️ Anthropic API Key not found!
+        
+        Please make sure you have set the ANTHROPIC_API_KEY:
+        - For local development: in your .env file
+        - For Hugging Face Spaces: in your Space's secrets
+        """)
+        st.stop()
+
+    # Initialize Anthropic client
+    c = Anthropic(api_key=api_key)
+    
+except Exception as e:
+    st.error(f"""
+    ⚠️ Error initializing Anthropic client: {str(e)}
+    
+    Please check:
+    1. Your ANTHROPIC_API_KEY is set correctly
+    2. You're using a valid API key from Anthropic
+    """)
+    st.stop()
 
 # Initialize session state variables
 if 'messages' not in st.session_state:

src/test_api.py

@@ -0,0 +1,29 @@
+import os
+import streamlit as st
+from anthropic import Anthropic
+
+st.title("Anthropic API Key Test")
+
+try:
+    # Get API key from environment
+    api_key = os.getenv('ANTHROPIC_API_KEY')
+    if not api_key:
+        st.error("❌ No API key found in environment variables")
+        st.stop()
+    
+    # Try to initialize client
+    client = Anthropic(api_key=api_key)
+    
+    # Try a simple API call
+    response = client.messages.create(
+        model="claude-3-opus-20240229",
+        max_tokens=10,
+        messages=[{"role": "user", "content": "Say hello"}]
+    )
+    
+    st.success("✅ API key works! Got response from Claude")
+    st.write("Response:", response.content[0].text)
+    
+except Exception as e:
+    st.error(f"❌ Error: {str(e)}")
+    st.write("Please check your API key configuration")