Update app.py

app.py

@@ -15,7 +15,7 @@ with st.container():
         submitted = st.form_submit_button("Envoyer")
         if submitted and user_input:
             # Ajout de la question de l'utilisateur
-            st.session_state['messages'].append(f"Vous: {user_input}")
+            st.session_state['messages'].append({"type": "user", "message": user_input})
             
             # Génération de la réponse via le modèle Replicate
             output = replicate.run(
@@ -23,14 +23,26 @@ with st.container():
                 input={
                     "top_p": 0.9,
                     "prompt": user_input,
-                    "max_tokens": 2341,
-                    "min_tokens": 0,
+                    "max_tokens": 2334,
+                    "min_tokens": 40,
                     "temperature": 0.3,
                     "prompt_template": "system\n\nYou are a helpful assistantuser\n\n{prompt}assistant\n\n",
                     "presence_penalty": 0,
                     "frequency_penalty": 0
                 }
             )
+            
+            # Ajout de la réponse du modèle
+            st.session_state['messages'].append({"type": "bot", "message": output})
+
+    # Affichage des messages précédents dans des "chat messages"
+    for msg in st.session_state['messages']:
+        if msg["type"] == "user":
+            with st.chat_message("user"):
+                st.write(msg["message"])
+        else:
+            with st.chat_message("bot"):
+                st.write(msg["message"])
 
             # Formatage de la sortie pour assurer que c'est un texte simple
             response_text = output["choices"][0]["message"] if "choices" in output else "Désolé, je ne peux pas traiter votre demande."