updates

app.py

@@ -48,7 +48,7 @@ def get_deployments():
     st.write('✅ Deployments are ready!')
 
     # Returning deployment instances
-    return interactions_fg, videos_fg, ranking_deployment, query_model_deployment
+    return interactions_fg, videos_fg, ranking_deployment, query_model_deployment, fs
 
 def insert_interaction(user_id, video_id, interactions_fg):
     generic = Generic(locale=Locale.EN)
@@ -75,6 +75,7 @@ def fetch_recommendations(user_id, query_model_deployment):
     st.write('🔮 Getting recommendations...')
     deployment_input = {"instances": {"user_id": user_id}}
     prediction = query_model_deployment.predict(deployment_input)['predictions']['ranking']
+    st.write('🔮 Got recommendations...')
     return prediction
 
 # Function to insert interaction and fetch new recommendations
@@ -82,13 +83,12 @@ def handle_interaction(user_id, video_id, interactions_fg, query_model_deploymen
     insert_interaction(user_id, video_id, interactions_fg)
     return fetch_recommendations(user_id, query_model_deployment)
 
-
 # Main Streamlit application logic
 def main():
     st.title('🎬 Video Recommender')
     # Initialize or re-use existing deployments
     if 'deployments_initialized' not in st.session_state:
-        st.session_state.interactions_fg, st.session_state.videos_fg, st.session_state.ranking_deployment, st.session_state.query_model_deployment = get_deployments()
+        st.session_state.interactions_fg, st.session_state.videos_fg, st.session_state.ranking_deployment, st.session_state.query_model_deployment, st.session_state.fs = get_deployments()
         st.session_state['deployments_initialized'] = True
 
     # User selection box
@@ -107,9 +107,27 @@ def main():
 
     print_header('📝 Top 3 Recommendations:')
     displayed_recommendations = st.session_state.recommendations[:3]
-    for recommendation in displayed_recommendations:
-        video_id = recommendation[1]
-        if st.button(f"🔗 Video ID: {video_id}", key=video_id):
+
+    # Retrieve video information
+    video_id_list = [{"video_id": video_id[1]} for video_id in displayed_recommendations]
+    string_list = [f"'{i[1]}'" for i in displayed_recommendations]
+    result_string = ",".join(string_list)
+    sql_str = f"SELECT * FROM videos_1 WHERE video_id IN ({result_string})"
+    print(sql_str)
+    videos_df = st.session_state.fs.sql(sql_str, online=True)
+    videos_info = videos_df.to_dict('records')
+
+    for video_info in videos_info:
+        video_id = video_info['video_id']
+        category = video_info['category']
+        likes = video_info['likes']
+        upload_date = video_info['upload_date']
+        video_length = video_info['video_length']
+        views = video_info['views']
+
+        # Button text with video information
+        button_text = f"🔗 Video ID: {video_id} | 🎥 Category: {category} | 👍🏻 Likes: {likes} | 👁️ Views: {views} | ⚙️ Length: {video_length}"
+        if st.button(button_text, key=video_id):
             new_recommendations = handle_interaction(
                 user_id_option,
                 video_id,