Update app.py

app.py

@@ -63,7 +63,10 @@ def get_similar_and_recommend(input_text):
     most_similar_index = np.argmax(similarities)
 
     # Get all features of the most similar video
-    most_similar_video_features = df.iloc[most_similar_index].to_dict()
+   unwanted_keys = ["text_for_embedding", "embeddings"]
+    for key in unwanted_keys:
+        if key in most_similar_video_features:
+            del most_similar_video_features[key]
 
     # Recommend the top 10 videos based on GNN embeddings and dot product
     def recommend_next_10_videos(given_video_index, all_video_embeddings):
@@ -73,14 +76,20 @@ def get_similar_and_recommend(input_text):
         ]
         dot_products[given_video_index] = -float("inf")
 
-        top_10_indices = np.argsort(dot_products)[::-1][:10]
+        top_10_indices = np.argsort(dot_products)[[::-1][:10]
         return [df.iloc[idx].to_dict() for idx in top_10_indices]
 
     top_10_recommended_videos_features = recommend_next_10_videos(
         most_similar_index, all_video_embeddings
     )
 
-    # Create the output JSON with all features
+    # Exclude unwanted features for recommended videos
+    for recommended_video in top_10_recommended_videos_features:
+        for key in unwanted_keys:
+            if key in recommended_video:
+                del recommended_video[key]
+
+    # Create the output JSON with all features except the unwanted ones
     output = {
         "most_similar_video": most_similar_video_features,
         "top_10_recommended_videos": top_10_recommended_videos_features,
@@ -88,13 +97,13 @@ def get_similar_and_recommend(input_text):
 
     return output
 
-# Update the Gradio interface to output a JSON object containing all features
+# Update the Gradio interface to output a JSON object without unwanted features
 interface = gr.Interface(
     fn=get_similar_and_recommend,
     inputs=gr.components.Textbox(label="Enter Text to Find Most Similar Video"),
     outputs=gr.JSON(),
     title="Video Recommendation System with GNN-based Recommendations",
-    description="Enter text to find the most similar video and get the top 10 recommended videos with all features in a JSON object.",
+    description="Enter text to find the most similar video and get the top 10 recommended videos with all features except embeddings-related fields in a JSON object.",
 )
 
 interface.launch()