Karley8 commited on
Commit
f336a81
·
verified ·
1 Parent(s): 6b7a11c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -11
app.py CHANGED
@@ -1,18 +1,43 @@
1
- # app.py
2
  import streamlit as st
3
- from GlobalMovies import movie_titles, movie_descriptions, movie_languages, tfidf_matrix, recommend_foreign
 
 
 
 
 
 
 
4
 
5
  st.set_page_config(page_title="WorldFlix", layout="centered")
 
6
 
7
- st.title("🎬 WorldFlix: Discover Foreign Films")
8
 
9
- st.write("Enter a foreign-language movie you like, and we’ll recommend similar titles.")
 
10
 
11
- selected_language = st.selectbox("Choose a language:", ["fr", "ko", "ja", "hi", "es"])
12
- movie_input = st.text_input("Enter a movie title:")
 
 
13
 
14
- if movie_input:
15
- st.subheader("Recommended Movies:")
16
- recs = recommend_foreign(movie_input, tfidf_matrix, movie_titles, movie_languages, selected_language)
17
- for rec in recs:
18
- st.markdown(f"- **{rec}**")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
+ from GlobalMovies import (
3
+ region_languages,
4
+ movie_titles,
5
+ movie_descriptions,
6
+ movie_languages,
7
+ tfidf_matrix,
8
+ recommend_foreign
9
+ )
10
 
11
  st.set_page_config(page_title="WorldFlix", layout="centered")
12
+ st.title("🌍 WorldFlix: Discover Global Cinema")
13
 
14
+ st.markdown("Choose a region and a movie you like. We'll recommend similar international films.")
15
 
16
+ # Region dropdown
17
+ selected_region = st.selectbox("Choose a film region:", list(region_languages.keys()))
18
 
19
+ # Language options for the selected region
20
+ language_dict = region_languages[selected_region]
21
+ selected_lang_name = st.selectbox("Select a language:", list(language_dict.values()))
22
+ selected_language = [code for code, name in language_dict.items() if name == selected_lang_name][0]
23
 
24
+ # Filter titles for that language
25
+ filtered_titles = [title for i, title in enumerate(movie_titles) if movie_languages[i] == selected_language]
26
+
27
+ if not filtered_titles:
28
+ st.warning("No movies found for this language. Try another region or refresh the app.")
29
+ else:
30
+ selected_movie = st.selectbox("Choose a movie you like:", filtered_titles)
31
+
32
+ if selected_movie:
33
+ st.subheader("Recommended Films:")
34
+ recommendations = recommend_foreign(
35
+ selected_movie,
36
+ tfidf_matrix,
37
+ movie_titles,
38
+ movie_languages,
39
+ lang=selected_language
40
+ )
41
+
42
+ for rec in recommendations:
43
+ st.markdown(f"- 🎬 **{rec}**")