fix

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, fs
+    return interactions_fg, videos_fg, ranking_deployment, query_model_deployment
 
 def insert_interaction(user_id, video_id, interactions_fg):
     generic = Generic(locale=Locale.EN)
@@ -67,7 +67,7 @@ def insert_interaction(user_id, video_id, interactions_fg):
         'watch_time': [watch_time]
     })
 
-    interactions_fg.insert(interaction_df, write_options={"start_offline_materialization": False})
+    interactions_fg.insert(interaction_df)
 
 
 # Define function to fetch recommendations
@@ -75,7 +75,6 @@ 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
@@ -83,12 +82,13 @@ 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, st.session_state.fs = get_deployments()
+        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['deployments_initialized'] = True
 
     # User selection box
@@ -107,27 +107,9 @@ def main():
 
     print_header('📝 Top 3 Recommendations:')
     displayed_recommendations = st.session_state.recommendations[:3]
-
-    # 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):
+    for recommendation in displayed_recommendations:
+        video_id = recommendation[1]
+        if st.button(f"🔗 Video ID: {video_id}", key=video_id):
             new_recommendations = handle_interaction(
                 user_id_option,
                 video_id,