Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from openai import OpenAI
 from os import getenv
+from streamlit_audiorecorder import st_audiorecorder
+import tempfile
 
 # fonction pour lire et retourner le contenu de fichiers textes
 def lire_fichier(nom_fichier):
@@ -13,13 +15,44 @@ def lire_fichier(nom_fichier):
     except Exception as e:
         return f"Une erreur s'est produite lors de la lecture du fichier : {str(e)}"
 
-
 # Configuration de la clé API OpenAI
 client = OpenAI(api_key=getenv("OPENAI_API_KEY"))
 
 # Définition du prompt système initial
 SYSTEM_PROMPT = f"{lire_fichier('linguascribe.prompt')}"
 
+def transcribe_audio(audio_file):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_audio:
+        temp_audio.write(audio_file)
+        temp_audio.flush()
+        
+        with open(temp_audio.name, "rb") as audio_file:
+            transcript = client.audio.transcriptions.create(
+                model="whisper-1", 
+                file=audio_file
+            )
+    
+    return transcript.text
+
+def process_message(message):
+    st.session_state.messages.append({"role": "user", "content": message})
+    with st.chat_message("user"):
+        st.markdown(message)
+
+    with st.chat_message("assistant"):
+        message_placeholder = st.empty()
+        full_response = ""
+        for response in client.chat.completions.create(
+            model="gpt-4o",
+            messages=st.session_state.messages,
+            stream=True,
+            temperature=0.1,
+        ):
+            full_response += (response.choices[0].delta.content or "")
+            message_placeholder.markdown(full_response + "▌")
+        message_placeholder.markdown(full_response)
+    
+    st.session_state.messages.append({"role": "assistant", "content": full_response})
 
 def main():
     st.title("Application de Chat avec GPT-4")
@@ -38,28 +71,18 @@ def main():
     # Interface utilisateur pour le chat
     user_input = st.chat_input("Entrez votre message ici:")
 
+    # Ajout de l'enregistreur audio
+    st.write("Ou enregistrez votre message audio :")
+    audio_bytes = st_audiorecorder()
+
     if user_input:
-        # Ajout du message de l'utilisateur à l'historique
-        st.session_state.messages.append({"role": "user", "content": user_input})
-        with st.chat_message("user"):
-            st.markdown(user_input)
+        process_message(user_input)
 
-        # Traitement du message avec OpenAI
-        with st.chat_message("assistant"):
-            message_placeholder = st.empty()
-            full_response = ""
-            for response in client.chat.completions.create(
-                model="gpt-4o",
-                messages=st.session_state.messages,
-                stream=True,
-                temperature=0.1,  # Ajustement de la température à 0.1
-            ):
-                full_response += (response.choices[0].delta.content or "")
-                message_placeholder.markdown(full_response + "▌")
-            message_placeholder.markdown(full_response)
-        
-        # Ajout de la réponse de l'assistant à l'historique
-        st.session_state.messages.append({"role": "assistant", "content": full_response})
+    if len(audio_bytes) > 0:
+        st.audio(audio_bytes, format="audio/wav")
+        transcription = transcribe_audio(audio_bytes)
+        st.write(f"Transcription : {transcription}")
+        process_message(transcription)
 
     st.sidebar.markdown("## À propos")
     st.sidebar.info("Cette application utilise Streamlit et OpenAI GPT-4 pour créer un chat interactif avec un modèle de langage avancé.")