Update src/services/ai_service.py

src/services/ai_service.py

@@ -22,18 +22,9 @@ class AITutorService:
     def initialize_client(self):
         """Initialize Anthropic client with better error handling"""
         try:
-            # First try getting API key from Streamlit secrets
-            api_key = None
-            if hasattr(st, 'secrets') and 'ANTHROPIC_API_KEY' in st.secrets:
-                api_key = st.secrets['ANTHROPIC_API_KEY']
-            
-            # If not in secrets, try environment variable
-            if not api_key:
-                api_key = os.getenv('ANTHROPIC_API_KEY')
-            
+            api_key = st.secrets.get("ANTHROPIC_API_KEY") or os.getenv('ANTHROPIC_API_KEY')
             if not api_key:
-                st.error("Anthropic API key not found. Please set ANTHROPIC_API_KEY in .streamlit/secrets.toml or environment variables.")
-                st.info("For testing, you can set this up locally by creating a .streamlit/secrets.toml file with:\nANTHROPIC_API_KEY='your-key-here'")
+                st.error("ANTHROPIC_API_KEY not found in secrets or environment variables")
                 return
 
             self._client = anthropic.Anthropic(api_key=api_key)
@@ -144,13 +135,4 @@ class AITutorService:
             'timestamp': datetime.now().isoformat(),
             'sentiment_score': min(1.0, len(user_input) / 100.0),
             'interaction_length': len(user_input)
-        })
-
-def get_tutor_context() -> dict:
-    """Get or initialize tutor context"""
-    if 'tutor_context' not in st.session_state:
-        st.session_state.tutor_context = {
-            'current_topic': None,
-            'chat_history': []
-        }
-    return st.session_state.tutor_context
+        })