Update src/components/ai_tutor.py

src/components/ai_tutor.py

@@ -1,15 +1,13 @@
 import streamlit as st
 from src.services.ai_service import AITutorService
 from src.utils.session import get_tutor_context
-import openai
-import os  # Import the os module
 
 class AITutor:
     def __init__(self):
         self.service = AITutorService()
 
     def display_chat_interface(self):
-        """Display the enhanced chat interface with voice output and tutor representation"""
+        """Display the enhanced chat interface with voice output"""
         st.header("AI Tutor")
 
         # Voice controls
@@ -30,6 +28,9 @@ class AITutor:
                     </svg>
                     Voice Off
                 """, unsafe_allow_html=True)
+        with col2:
+            if self.service.tts_mode:
+                st.info(f"Using {self.service.tts_mode.upper()} TTS")
 
         # Topic selection
         topics = [None, 'Physics', 'Mathematics', 'Computer Science', 'Artificial Intelligence']
@@ -47,15 +48,12 @@ class AITutor:
         # Display chat container
         chat_container = st.container()
         with chat_container:
-            # Display chat history with voice output and tutor representation
-            for i, message in enumerate(context['chat_history']):
-                if message["role"] == "assistant":
-                    # Show tutor image for assistant messages
-                    st.image("path/to/your/tutor_image.png", width=100)  # Replace with your image path
+            # Display chat history with voice output
+            for message in context['chat_history']:
                 with st.chat_message(message["role"]):
                     st.write(message["content"])
                     if message["role"] == "assistant" and voice_active:
-                        self.speak(message["content"])
+                        self.service.speak(message["content"])
 
         # Chat input
         if prompt := st.text_input("Ask your question...", key="chat_input"):
@@ -78,22 +76,7 @@ class AITutor:
         context['chat_history'].append({
             "role": "assistant",
             "content": response,
+            "speak": True
         })
 
-        # No need to rerun here, Streamlit will update automatically
-
-    def speak(self, text):
-        """Uses OpenAI's TTS to speak the given text."""
-        try:
-            # Get the OpenAI API key from the environment variable
-            openai.api_key = os.getenv("OPENAI_API_KEY")  
-            response = openai.Audio.create(
-                input=text,
-                model="tts-1",
-                voice="alloy" 
-            )
-
-            # Play the audio
-            st.audio(response['audio_content'], format="audio/mpeg") 
-        except Exception as e:
-            st.error(f"Error in voice output: {e}")
+        # No need to rerun here, Streamlit will update automatically