Update app.py

app.py

@@ -14,10 +14,13 @@ if "messages" not in st.session_state:
 
 # Affichage de tous les messages précédents
 for msg in st.session_state.messages:
-    st.chat_message(role=msg["role"], content=msg["content"])
+    if msg["role"] == "user":
+        st.write(f"User: {msg['content']}")
+    else:
+        st.write(f"Assistant: {msg['content']}")
 
 # Entrée pour les nouvelles questions
-prompt = st.chat_input("Posez votre question :")
+prompt = st.text_input("Posez votre question :", key="chat")
 
 if prompt:
     if not api_key:
@@ -40,4 +43,4 @@ if prompt:
 
         # Ajouter et afficher la réponse de l'assistant
         st.session_state.messages.append({"role": "assistant", "content": msg})
-        st.chat_message(role="assistant", content=msg)
+        st.write(f"Assistant: {msg}")