Update app.py

app.py

@@ -8,21 +8,26 @@ client = anthropic.Anthropic(api_key=api_key)
 
 st.title("💬 Chatbot avec Claude 3")
 
+# Initialisation de l'état de la session pour stocker les messages s'il n'existe pas déjà
+if "messages" not in st.session_state:
+    st.session_state["messages"] = []
+
 # Affichage de tous les messages précédents
-def display_messages():
-    for msg in st.session_state.messages:
+for msg in st.session_state.messages:
+    # Utiliser st.container pour mieux organiser l'affichage des messages
+    with st.container():
         if msg["role"] == "user":
-            st.write(f"User: {msg['content']}")
+            st.text(f"User: {msg['content']}")
         else:
-            st.write(f"Assistant: {msg['content']}")
+            st.text(f"Assistant: {msg['content']}")
 
-# Entrée pour les nouvelles questions
-def input_and_response():
+# Entrée pour les nouvelles questions avec un container en bas
+with st.container():
     prompt = st.text_input("Posez votre question :", key="chat")
     if prompt:
         if not api_key:
             st.error("Clé API de Anthropic non trouvée. Veuillez définir la variable d'environnement ANTHROPIC_API_KEY.")
-            return
+            st.stop()
 
         # Ajouter le nouveau message utilisateur à l'état de la session
         st.session_state.messages.append({"role": "user", "content": prompt})
@@ -45,11 +50,4 @@ def input_and_response():
 
             # Ajouter et afficher la réponse de l'assistant
             st.session_state.messages.append({"role": "assistant", "content": msg})
-
-# Initialisation de l'état de la session pour stocker les messages s'il n'existe pas déjà
-if "messages" not in st.session_state:
-    st.session_state["messages"] = []
-
-# Placer les fonctions d'affichage des messages et de saisie des questions
-display_messages()
-input_and_response()
+            st.text(f"Assistant: {msg}")