Update src/components/ai_tutor.py

src/components/ai_tutor.py

@@ -1,11 +1,14 @@
 import streamlit as st
-from src.services.ai_service import AITutorService
-from src.utils.session import get_tutor_context
+from src.services.ai_service import AITutorService  # Assuming this is your AI service
+from src.utils.session import get_tutor_context  # Assuming this provides context
 
 class AITutor:
     def __init__(self):
         self.service = AITutorService()
-        self.last_question = None  # Store the last question asked
+        if "chat_history" not in st.session_state:
+            st.session_state.chat_history = []
+        if "last_question" not in st.session_state:
+            st.session_state.last_question = None
 
     def display_chat_interface(self):
         """Display the enhanced chat interface with voice output"""
@@ -14,8 +17,8 @@ class AITutor:
         # Voice controls
         col1, col2 = st.columns([3, 2])
         with col1:
-            voice_active = st.checkbox("Enable Voice", value=False, key="voice_active")
-            if voice_active:
+            st.session_state.voice_active = st.checkbox("Enable Voice", value=False, key="voice_active")
+            if st.session_state.voice_active:
                 st.markdown("""
                     <svg width="24" height="24" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">
                         <path d="M15 8V16M11 8V16M7 12H5C3.89543 12 3 12.8954 3 14V16C3 17.1046 3.89543 18 5 18H7V12ZM19 12V16C19 17.1046 18.1046 18 17 18H15" stroke="#4A90E2" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
@@ -49,37 +52,30 @@ class AITutor:
         # Display chat container
         chat_container = st.container()
         with chat_container:
-            # Display chat history with voice output
-            for message in context['chat_history']:
+            for message in st.session_state.chat_history:
                 with st.chat_message(message["role"]):
                     st.write(message["content"])
-                    if message["role"] == "assistant" and voice_active:
+                    if message["role"] == "assistant" and st.session_state.voice_active:
                         self.service.speak(message["content"])
 
         # Chat input
-        prompt = st.text_input("Ask your question...", key="chat_input") 
-        if prompt and prompt != self.last_question:  # Check if the question is new
+        prompt = st.text_input("Ask your question...", key="chat_input")
+        if prompt and prompt != st.session_state.last_question:
             self.handle_user_input(prompt)
-            self.last_question = prompt  # Update the last question asked
+            st.session_state.last_question = prompt
 
     def handle_user_input(self, user_input: str):
         """Process user input and generate response"""
         context = get_tutor_context()
 
-        # Add user message
-        context['chat_history'].append({
+        st.session_state.chat_history.append({
             "role": "user",
             "content": user_input
         })
 
-        # Generate and display AI response
         response = self.service.generate_response(user_input, context['current_topic'])
 
-        # Add AI response
-        context['chat_history'].append({
+        st.session_state.chat_history.append({
             "role": "assistant",
             "content": response,
-            "speak": True  # You might want to remove this if not using voice output
         })
-
-        # No need to rerun here, Streamlit will update automatically