Update app.py

app.py

@@ -29,20 +29,21 @@ def main():
         # Traitement du message avec OpenAI
         with st.status("L'assistant est en train de répondre...", expanded=True) as status:
             # Utilisation de l'API OpenAI
-            stream = client.chat.completions.create(
-                model="gpt-4o",
-                messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
-                stream=True
-            )
+            async def generate_response():
+                stream = client.chat.completions.create(
+                    model="gpt-4o",
+                    messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
+                    stream=True
+                )
+                for chunk in stream:
+                    if chunk.choices[0].delta.content is not None:
+                        yield chunk.choices[0].delta.content
 
             # Streaming de la réponse
             response = ""
             with st.chat_message("assistant"):
-                for chunk in stream:
-                    if chunk.choices[0].delta.content is not None:
-                        response += chunk.choices[0].delta.content
-                        status.update(label="L'assistant répond...", expanded=True)
-                        st.write_stream(chunk.choices[0].delta.content)
+                status.update(label="L'assistant répond...", expanded=True)
+                st.write_stream(generate_response())
 
             # Ajout de la réponse de l'assistant à l'historique
             st.session_state.messages.append({"role": "assistant", "content": response})