Update pages/🧠_sickle_cell_chatbot.py

pages/🧠_sickle_cell_chatbot.py

@@ -18,6 +18,29 @@ st.set_page_config(page_title="Sickle cell chatbot", page_icon="🧠")
 st.header('Sickle cell chatbot')
 st.write('Enhancing Chatbot Interactions through sickle cell training')
 
+from audio_recorder_streamlit import audio_recorder
+
+# Subheading for audio input button
+st.sidebar.header("Use Audio Input 🔊")
+with st.sidebar:
+    audio_bytes = audio_recorder(
+    pause_threshold=3.0, 
+    text="Start Talking  👉",
+    recording_color="#e8b62c",
+    neutral_color="#6aa36f",
+    icon_name="microphone",
+    icon_size="3x",)
+
+
+def transcribe_audio(audio_bytes):
+    openai.api_key = os.getenv('OPENAI_API_KEY')
+    response = openai.audio.transcriptions.create(
+        model="whisper-1",
+        file=("temp." + "wav", audio_bytes, "audio/wav"),
+        )
+    
+    return response.text
+
 class ContextChatbot:
 
     def __init__(self):
@@ -89,9 +112,29 @@ class ContextChatbot:
 
     @utils.enable_chat_history
     def main(self):
+
+        
         chain = self.setup_chain()
         user_query = st.chat_input(placeholder="Ask me anything!")
-        if user_query:
+        if audio_bytes:
+            transcript = transcribe_audio(audio_bytes)
+            if transcript:
+                question = transcript
+                utils.display_msg(question, 'user')
+                with st.chat_message("assistant"):
+                    st_cb = StreamHandler(st.empty())
+                    result = chain.invoke(
+                        {"question":question},
+                        {"callbacks": [st_cb]}
+                    )
+                    #response = ""
+                    #for source in result['source_documents']:
+                        #response += source.metadata['source']
+                        #response += "Content: " + source.page_content
+                    response = result["answer"]
+                    st.session_state.messages.append({"role": "assistant", "content": response})
+
+        elif user_query:
             utils.display_msg(user_query, 'user')
             with st.chat_message("assistant"):
                 st_cb = StreamHandler(st.empty())