Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,10 +2,152 @@ import streamlit as st
 import sys
 import os
 
-# Add the path to the 'speech' folder manually
-sys.path.append(os.path.abspath('speech'))
+#-------------------------------------
+import pyttsx3
+import time
+from llama_index.llms.groq import Groq
+
+class TextToSpeechChat:
+    def __init__(self, api_key="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"):
+        # Initialize the LLM
+        self.llm = Groq(model="llama3-70b-8192", api_key=api_key)
+        
+        # Initialize text-to-speech engine
+        self.tts_engine = pyttsx3.init()
+        self.setup_tts()
+        
+        # Initialize speech enabled flag
+        self.speech_enabled = True
+        
+    def setup_tts(self):
+        """Configure text-to-speech settings"""
+        # Get available voices
+        voices = self.tts_engine.getProperty('voices')
+        
+        # Set voice (use first available voice)
+        if voices:
+            self.tts_engine.setProperty('voice', voices[0].id)
+        else:
+            print("⚠️ No voices available for text-to-speech")
+        
+        # Set speech rate (words per minute)
+        self.tts_engine.setProperty('rate', 150)
+        
+        # Set volume (0.0 to 1.0)
+        self.tts_engine.setProperty('volume', 0.9)
+        
+        print("🔊 Text-to-speech engine initialized")
+    
+    def speak_text(self, text):
+        """Convert text to speech with error handling"""
+        if not text.strip():
+            return
+            
+        try:
+            print(f"🗣️ Speaking paragraph: {text}")
+            self.tts_engine.say(text)
+            self.tts_engine.runAndWait()
+        except Exception as e:
+            print(f"❌ Speech error: {e}")
+    
+    def get_llm_response(self, prompt):
+        """Get response from LLM and speak paragraphs in real-time if enabled"""
+        try:
+            print("🤖 Generating response...")
+            response = self.llm.stream_complete(prompt)
+            
+            # Collect the full response and buffer for paragraphs
+            full_response = ""
+            buffer = ""
+            
+            for r in response:
+                delta = r.delta
+                full_response += delta
+                buffer += delta
+                print(delta, end="", flush=True)
+                
+                # Check for paragraph boundary (\n\n)
+                if self.speech_enabled and "\n\n" in buffer:
+                    # Split buffer into paragraphs
+                    paragraphs = buffer.split("\n\n")
+                    # Speak all complete paragraphs (all but the last part)
+                    for paragraph in paragraphs[:-1]:
+                        if paragraph.strip():
+                            self.speak_text(paragraph.strip())
+                    # Keep the last part (incomplete paragraph) in buffer
+                    buffer = paragraphs[-1]
+            
+            # Speak any remaining buffered text
+            if self.speech_enabled and buffer.strip():
+                self.speak_text(buffer.strip())
+            
+            print()  # New line after response
+            return full_response
+            
+        except Exception as e:
+            error_msg = f"❌ Error getting LLM response: {e}"
+            print(error_msg)
+            if self.speech_enabled:
+                self.speak_text(error_msg)
+            return error_msg
+    
+    def chat_with_speech(self):
+        """Interactive chat with text-to-speech output"""
+        print("🎯 TEXT-TO-SPEECH CHAT")
+        print("=" * 50)
+        print("💡 Type your message and press Enter")
+        print("🗣️ The AI response will be spoken aloud paragraph by paragraph as it generates")
+        print("💬 Type 'quit' or 'exit' to end the chat")
+        print("🔇 Type 'mute' to disable speech")
+        print("🔊 Type 'unmute' to enable speech")
+        print("=" * 50)
+        
+        # Test speech engine at startup
+        self.speak_text("Text-to-speech chat started.")
+        
+        while True:
+            try:
+                # Get user input
+                user_input = input("\n👤 You: ").strip()
+                
+                if not user_input:
+                    continue
+                    
+                # Check for commands
+                if user_input.lower() in ['quit', 'exit', 'q']:
+                    self.speak_text("Goodbye!")
+                    print("👋 Goodbye!")
+                    break
+                elif user_input.lower() == 'mute':
+                    self.speech_enabled = False
+                    print("🔇 Speech disabled")
+                    continue
+                elif user_input.lower() == 'unmute':
+                    self.speech_enabled = True
+                    print("🔊 Speech enabled")
+                    self.speak_text("Speech enabled.")
+                    continue
+                
+                # Get AI response, which now speaks paragraphs in real-time
+                print("\n🤖 AI: ", end="")
+                response = self.get_llm_response(user_input)
+                
+            except KeyboardInterrupt:
+                self.speak_text("Goodbye!")
+                print("\n👋 Goodbye!")
+                break
+            except Exception as e:
+                print(f"❌ Error: {e}")
+                if self.speech_enabled:
+                    self.speak_text(f"Error: {str(e)}")
+
+def main(user_input):
+    chat = TextToSpeechChat()
+    response = chat.get_llm_response(user_input)
+    return response
+
+#-----------------------------------
 
-import spck_groq as sg
 
 # Title of the app
 st.title("Simple Streamlit UI")
@@ -15,5 +157,5 @@ user_input = st.text_input("Enter something:")
 
 # Button
 if st.button("Submit"):
-    st.write("You entered:", user_input)
-    sg.main(user_input)
+    st.write("You entered:", main(user_input))
+