Update app.py

app.py

@@ -19,7 +19,7 @@ def load_model():
 # Function to search the dataset for movie details
 def search_movie_details(query, data):
     query = query.lower()
-    filtered_data = data.dropna(subset=["title", "cast", "director"])  # Remove NaN values
+    filtered_data = data.dropna(subset=["title", "cast", "director", "country"])  # Remove NaN values
     results = filtered_data[
         filtered_data["title"].str.lower().str.contains(query, na=False) |
         filtered_data["cast"].str.lower().str.contains(query, na=False) |
@@ -70,6 +70,7 @@ if user_input:
             st.write(f"**Director:** {row['director']}")
             st.write(f"**Cast:** {row['cast']}")
             st.write(f"**Release Year:** {row['release_year']}")
+            st.write(f"**Country:** {row['country']}")  # ✅ Now country is displayed correctly
             st.write(f"**Rating:** {row['rating']}")
             st.write(f"**Description:** {row['description']}")
             st.write("---")
@@ -79,3 +80,4 @@ if user_input:
         outputs = model.generate(inputs, max_length=100, num_return_sequences=1)
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
         st.write(f"Chatbot: {response}")
+