Update app.py

app.py

@@ -8,47 +8,48 @@ 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
-for msg in st.session_state.messages:
-    if msg["role"] == "user":
-        st.write(f"User: {msg['content']}")
-    else:
-        st.write(f"Assistant: {msg['content']}")
+def display_messages():
+    for msg in st.session_state.messages:
+        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.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.")
-        st.stop()
-
-    # Ajouter le nouveau message utilisateur à l'état de la session
-    st.session_state.messages.append({"role": "user", "content": prompt})
-
-    # Utiliser un indicateur pour montrer que le modèle génère une réponse
-    with st.spinner('Thinking...'):
-        response = client.messages.create(
-            model="claude-3-opus-20240229",
-            max_tokens=1024,
-            messages=[{"role": msg["role"], "content": msg["content"]} for msg in st.session_state.messages]
-        )
-        
-        # Récupérer le contenu de la réponse
-        if isinstance(response.content, list):
-            # Supposons que tous les éléments de la liste sont des objets TextBlock
-            msg = " ".join(block.text for block in response.content)
-        elif isinstance(response.content, dict) and 'text' in response.content:
-            # Supposons que la réponse soit un dictionnaire avec une clé 'text'
-            msg = response.content['text']
-        else:
-            # Si ce n'est ni une liste ni un dictionnaire avec 'text'
-            msg = str(response.content)
+def input_and_response():
+    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
+
+        # Ajouter le nouveau message utilisateur à l'état de la session
+        st.session_state.messages.append({"role": "user", "content": prompt})
+
+        # Utiliser un indicateur pour montrer que le modèle génère une réponse
+        with st.spinner('Thinking...'):
+            response = client.messages.create(
+                model="claude-3-opus-20240229",
+                max_tokens=1024,
+                messages=[{"role": msg["role"], "content": msg["content"]} for msg in st.session_state.messages]
+            )
+            
+            # Récupérer et traiter le contenu de la réponse
+            if isinstance(response.content, list):
+                msg = " ".join(block.text for block in response.content)
+            elif isinstance(response.content, dict) and 'text' in response.content:
+                msg = response.content['text']
+            else:
+                msg = str(response.content)
+
+            # 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"] = []
 
-        # Ajouter et afficher la réponse de l'assistant
-        st.session_state.messages.append({"role": "assistant", "content": msg})
-        st.write(f"Assistant: {msg}")
+# Placer les fonctions d'affichage des messages et de saisie des questions
+display_messages()
+input_and_response()