Update app.py

app.py

@@ -6,7 +6,7 @@ from os import getenv
 client = OpenAI(api_key=getenv("OPENAI_API_KEY"))
 
 def main():
-    st.title("Application de Chat avec GPT-4o")
+    st.title("Application de Chat avec GPT-4")
 
     # Initialisation de l'historique des messages
     if "messages" not in st.session_state:
@@ -27,27 +27,23 @@ def main():
             st.markdown(user_input)
 
         # 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
-            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"):
-                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})
-            status.update(label="Réponse terminée", state="complete", expanded=False)
+        with st.chat_message("assistant"):
+            message_placeholder = st.empty()
+            full_response = ""
+            for response in client.chat.completions.create(
+                model="gpt-4",
+                messages=[
+                    {"role": m["role"], "content": m["content"]}
+                    for m in st.session_state.messages
+                ],
+                stream=True,
+            ):
+                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})
 
     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é.")