Update app.py

app.py

@@ -97,7 +97,7 @@ def launch_bot():
                     #st.write("reroute to LLM")
                     #call in Mistral
                 prompt3 = master_prompt + prompt2 + "context:" + response
-                print("Called in Mistral")
+                print("Here's where we would call in Mistral")
                     # ADD IN LLM
                 # st.write("Mistral:" ) #Needs finishing
               #  else:
@@ -106,46 +106,48 @@ def launch_bot():
         st.session_state.messages.append(message)
 
     # If assistant has most recently reaponded create audio of response 
-        if st.session_state.messages[-1]["role"] == "assistant":
-            audio_result = st.button("Convert to Audio 🔊") 
-            if audio_result:
-                st.session_state.messages.append({"role": "assistant", "content": prompt})
-                with st.chat_message("text-to-speech (may take a few minutes"):
-                #text-to-speech
-                    print("Calling in Text-to-speech via suno/bark-small")
+    if st.session_state.messages[-1]["role"] == "assistant":
+        audio_result = st.button("Convert to Audio 🔊") 
+        if audio_result:
+            st.session_state.messages.append({"role": "assistant", "content": prompt})
+            with st.chat_message("text-to-speech (may take a few minutes"):
+            #text-to-speech
+                print("Calling in Text-to-speech via suno/bark-small")
                     
-                    synthesiser = pipeline("text-to-speech", "suno/bark-small")
+                synthesiser = pipeline("text-to-speech", "suno/bark-small")
             
-                    speech = synthesiser(response, forward_params={"do_sample": True})
+                speech = synthesiser(response, forward_params={"do_sample": True})
 
-                    # write to temp file: the scipy wav write bark_out.wav
-                    fp.write(scipy.io.wavfile.write("bark_out.wav", rate=speech["sampling_rate"], data=speech["audio"]))
+                # write to temp file: the scipy wav write bark_out.wav
+                fp.write(scipy.io.wavfile.write("bark_out.wav", rate=speech["sampling_rate"], data=speech["audio"]))
                     
-                # ST interface for audio
-                    print("Now we try to display the audio file in the app")
-                    
-                    # Temp file access
-                    fp.seek(0)
-                    audio_file = fp.read('bark_out.wav', 'rb')
+            # ST interface for audio
+                print("Now we try to display the audio file in the app")
+                
+                # Temp file access
+                fp.seek(0)
+                audio_file = fp.read('bark_out.wav', 'rb')
                     
-                    audio_bytes = audio_file.read()
-        
-                    audio_convert = st.audio(audio_bytes, format='audio/wav')
-                    st.session_state.messages.append(audio_convert)
-                    sample_rate = 44100  # 44100 samples per second
-                    seconds = 2  # Note duration of 2 seconds
+                audio_bytes = audio_file.read()
+                
+                st.audio(audio_bytes, format='audio/wav')
+                
+                #audio_convert = st.audio(audio_bytes, format='audio/wav')
+                #st.session_state.messages.append(audio_convert)
+                #sample_rate = 44100  # 44100 samples per second
+                #seconds = 2  # Note duration of 2 seconds
                     
-                    frequency_la = 440  # Our played note will be 440 Hz
+                #frequency_la = 440  # Our played note will be 440 Hz
                     
-                    # Generate array with seconds*sample_rate steps, ranging between 0 and seconds
-                    t = np.linspace(0, seconds, seconds * sample_rate, False)
+                # Generate array with seconds*sample_rate steps, ranging between 0 and seconds
+                #t = np.linspace(0, seconds, seconds * sample_rate, False)
                     
-                    # Generate a 440 Hz sine wave
-                    note_la = np.sin(frequency_la * t * 2 * np.pi)
-                    st.audio(note_la, sample_rate=sample_rate)
+                # Generate a 440 Hz sine wave
+                #note_la = np.sin(frequency_la * t * 2 * np.pi)
+                #st.audio(note_la, sample_rate=sample_rate)
 
-                    # Close and delete temp file
-                    fp.close()
+                # Close and delete temp file
+                fp.close()
                 
 if __name__ == "__main__":
     launch_bot()