update readme

Home.py

@@ -33,9 +33,9 @@ def save_user_id(user_id):
 
 
 def logout():
-    st.session_state.pop('user_id')
-    st.session_state.pop('selected_dict')
-    st.session_state.pop('score_weights')
+    st.session_state.pop('user_id', None)
+    st.session_state.pop('selected_dict', None)
+    st.session_state.pop('score_weights', None)
 
 
 if __name__ == '__main__':

README.md

@@ -1,5 +1,5 @@
 ---
-title: GEMRec-ModelCofferGallery
+title: GEMRec-Gallery
 emoji: 🖼️
 colorFrom: blue
 colorTo: purple

pages/Gallery.py

@@ -5,6 +5,7 @@ import altair as alt
 import numpy as np
 import pandas as pd
 import streamlit as st
+import streamlit.components.v1 as components
 
 from bs4 import BeautifulSoup
 from datasets import load_dataset, Dataset, load_from_disk
@@ -497,6 +498,7 @@ class GalleryApp:
             # normalize optimal_weight
             optimal_weight = [round(weight/len(selected), 2) for weight in optimal_weight]
             print('optimal weight:', optimal_weight)
+            print('optimal weight:', optimal_weight)
 
         st.session_state.score_weights[0: 3] = optimal_weight
 
@@ -619,3 +621,16 @@ if __name__ == "__main__":
 
         app = GalleryApp(promptBook=promptBook, images_ds=images_ds)
         app.app()
+
+        # components.html(
+        #     """
+        #     <script>
+        #       var iframe = window.parent.document.querySelector('[title="streamlit_agraph.agraph"]');
+        #       console.log(iframe);
+        #       var targetElement = iframe.contentDocument.querySelector('div.vis-network div.vis-navigation div.vis-button.vis-zoomExtends');
+        #       console.log(targetElement);
+        #       targetElement.style.background-image = "url(https://www.flaticon.com/free-icon-font/menu-burger_3917215?related_id=3917215#)";
+        #     </script>
+        #     """,
+        #     # unsafe_allow_html=True,
+        # )