fix bug

solve chat disappearing

app.py

@@ -242,14 +242,19 @@ def main():
     if st.button(f"Double-Check Response", key="google_button"):
         search_query = get_search_query(st.session_state['generated'][-1])
         retrieved_links = search_url(search_query)
+
+        for i in range(len(st.session_state['generated'])):
+                    message(st.session_state['past'][i], is_user=True, key=str(i) + '_user', logo="https://i.pinimg.com/originals/d5/b2/13/d5b21384ccaaa6f9ef32986f17c50638.png")
+                    message(st.session_state["generated"][i], key=str(i), logo= "https://emojiisland.com/cdn/shop/products/Robot_Emoji_Icon_7070a254-26f7-4a54-8131-560e38e34c2e_large.png?v=1571606114")
+        
         if retrieved_links:
             st.markdown("**Google Search Results:**")
             for j, link in enumerate(retrieved_links):
                 st.markdown(f"{j + 1}. [{link['title']}]({link['link']})")
 
                 # Display Google logo
+                st.image(google_logo_url, width=15, caption="Google Logo")
                 google_logo_url = "https://companieslogo.com/img/orig/GOOG-0ed88f7c.png"
-                st.image(google_logo_url, width=50, caption="Google Logo")
         
 if __name__ == "__main__":
     main()