Update app.py

app.py

@@ -105,6 +105,8 @@ def main():
     # Initialize session state
     if "chat" not in st.session_state:
         st.session_state.chat = model.start_chat(history=[])
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
     if "web_search" not in st.session_state:
         st.session_state.web_search = False
 
@@ -120,13 +122,16 @@ def main():
     uploaded_file = st.file_uploader("Télécharger un fichier (image/document)", 
                                    type=['jpg', 'mp4', 'mp3', 'jpeg', 'png', 'pdf', 'txt'])
 
-    # Display chat history
-    for message in st.session_state.chat.history:
-        with st.chat_message(role_to_streamlit(message.role)):
-            st.markdown(message.parts[0].text)
+    # Display chat history from session state
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
 
     # Chat input
     if prompt := st.chat_input("Hey?"):
+        # Add user message to chat history
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        
         # Display user message
         st.chat_message("user").markdown(prompt)
 
@@ -158,11 +163,8 @@ def main():
                 # Remove the cursor and update with the final response
                 message_placeholder.markdown(full_response)
             
-            # Add the interaction to chat history
-            st.session_state.chat.history.extend([
-                {"role": "user", "parts": [prompt]},
-                {"role": "model", "parts": [full_response]}
-            ])
+            # Add assistant response to chat history
+            st.session_state.messages.append({"role": "assistant", "content": full_response})
 
         except Exception as e:
             st.error(f"Erreur lors de l'envoi du message : {e}")