Update app.py

app.py

@@ -54,27 +54,12 @@ st.markdown("""
 # --------------------------
 # Movie Dataset & Embeddings
 # --------------------------
+# Replace load_movie_data() with:
 @st.cache_resource
 def load_movie_data():
-    # Synthetic movie dataset (replace with wiki_movies loading)
-    movies = [
-        {
-            "title": "The Dark Knight",
-            "plot": "Batman faces the Joker in a battle for Gotham's soul...",
-            "cast": {"Christian Bale": "Bruce Wayne", "Heath Ledger": "Joker"},
-            "director": "Christopher Nolan",
-            "year": 2008
-        },
-        {
-            "title": "Inception",
-            "plot": "A thief who enters the dreams of others...",
-            "cast": {"Leonardo DiCaprio": "Cobb", "Tom Hardy": "Eames"},
-            "director": "Christopher Nolan",
-            "year": 2010
-        }
-    ]
-    df = pd.DataFrame(movies)
-    df['context'] = df.apply(lambda x: f"Title: {x.title}\nPlot: {x.plot}\nCast: {', '.join([f'{k} as {v}' for k,v in x.cast.items()])}\nDirector: {x.director}\nYear: {x.year}", axis=1)
+    dataset = load_dataset("wiki_movies", split="train")
+    df = pd.DataFrame(dataset)
+    df['context'] = df.apply(lambda x: f"Title: {x['title']}\nPlot: {x['plot']}\nCast: {x['cast']}", axis=1)
     return df
 
 @st.cache_resource