Update src/components/ai_tutor.py

src/components/ai_tutor.py

@@ -1,13 +1,15 @@
 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"""
+        """Display the enhanced chat interface with voice output and tutor representation"""
         st.header("AI Tutor")
 
         # Voice controls
@@ -28,9 +30,6 @@ 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']
@@ -48,12 +47,15 @@ class AITutor:
         # Display chat container
         chat_container = st.container()
         with chat_container:
-            # Display chat history with voice output
-            for message in context['chat_history']:
+            # 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
                 with st.chat_message(message["role"]):
                     st.write(message["content"])
                     if message["role"] == "assistant" and voice_active:
-                        self.service.speak(message["content"])
+                        self.speak(message["content"])
 
         # Chat input
         if prompt := st.text_input("Ask your question...", key="chat_input"):
@@ -76,7 +78,22 @@ class AITutor:
         context['chat_history'].append({
             "role": "assistant",
             "content": response,
-            "speak": True
         })
 
-        # No need to rerun here, Streamlit will update automatically
+        # 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}")