app.py

import streamlit as st
import main
from PIL import Image
import requests
from streamlit_option_menu import option_menu


stop = False


nav = option_menu(
    None,
    ["Home", "Contribute"],
    icons=["house", "pencil"],
    default_index=0,
    menu_icon="list",
    orientation="horizontal",
)

if nav == "Home":

    st.title("FlickFinder")

    st.markdown(
        """Introducing FlickFinder, your movie matchmaker with a twist! Tired of endlessly scrolling through streaming platforms, 
    desperately seeking that perfect movie night gem? Say no more! 
    FlickFinder is here to save the day, bringing you personalized movie recommendations like no other."""
    )

    st.markdown(
        """We're not your ordinary movie recommender; we've mastered the art of cinematic matchmaking. 
    Just tell us your favorite flick, and we'll unleash our algorithmic Cupids to curate a handpicked 
    selection of movies that capture the same essence, style, or thrills you adore. Get ready to embark on an unforgettable movie adventure 
    tailored just for you. Lights, camera, discover! """
    )

    option = st.selectbox("Choose your movie here", (i for i in main.titles))
    # print(option)
    st.subheader("You selected:")
    option_img_response = requests.get(
        f"http://www.omdbapi.com/?t={option}&apikey=f5f824aa"
    )
    option_img = option_img_response.json()

    try:
        st.image(option_img["Poster"], width=200, caption=option)
    except KeyError:
        st.warning("We haven't found sufficient information on this movie yet")
        stop = True

    if not stop:
        rec = main.recommend(option)
        images = []
        summaries = []
        for i in rec:
            images.append(
                (
                    i,
                    (
                        requests.get(f"http://www.omdbapi.com/?t={i}&apikey=f5f824aa")
                    ).json(),
                )
            )
            # print(((main.data.loc[main.data['title'] == i])['overview']).to_list())
            try:
                summaries.append(
                    (
                        " ".join(
                            str(v)
                            for v in (
                                (
                                    (main.data.loc[main.data["title"] == i])["overview"]
                                ).to_list()
                            )[0]
                        )
                    )
                )
            except IndexError:
                summaries.append("Summary not found")

        st.subheader("Check out these recommendations:")

        col1, col2, col3 = st.columns(3)

        with col1:
            if images[0][1]["Response"] != False:
                if images[0][1]['Poster']=='N/A':
                    im = 'not_found.jpg'
                else:
                    im=images[0][1]['Poster']
                st.image(
                    im,
                    width=200,
                    caption=f"{images[0][0]}: {summaries[0]}",
                )
            else:
                st.image(image="not_found.jpg", width=200)
        with col2:
            if images[1][1]["Response"] != False:
                if images[1][1]['Poster']=='N/A':
                    im = 'not_found.jpg'
                else:
                    im=images[1][1]['Poster']
                st.image(
                    im,
                    width=200,
                    caption=f"{images[1][0]}: {summaries[1]}",
                )
            else:
                st.image(image="not_found.jpg", width=200)
        with col3:
            if images[2][1]["Response"] != False:
                if images[2][1]['Poster']=='N/A':
                    im = 'not_found.jpg'
                else:
                    im=images[2][1]['Poster']
                st.image(
                    im,
                    width=200,
                    caption=f"{images[2][0]}: {summaries[2]}",
                )
            else:
                st.image(image="not_found.jpg", width=200)

        col4, col5, col6 = st.columns(3)
        with col4:
            if images[3][1]["Response"] != False:
                if images[3][1]['Poster']=='N/A':
                    im = 'not_found.jpg'
                else:
                    im=images[3][1]['Poster']
                st.image(
                    im,
                    width=200,
                    caption=f"{images[3][0]}: {summaries[3]}",
                )
            else:
                st.image(image="not_found.jpg", width=200)
        with col5:
            if images[4][1]["Response"] != False:
                if images[4][1]['Poster']=='N/A':
                    im = 'not_found.jpg'
                else:
                    im=images[4][1]['Poster']
                st.image(
                    im,
                    width=200,
                    caption=f"{images[4][0]}: {summaries[4]}",
                )
            else:
                st.image(image="not_found.jpg", width=200)
        with col6:
            if images[5][1]["Response"] != "False":
                if images[5][1]['Poster']=='N/A':
                    im = 'not_found.jpg'
                else:
                    im=images[5][1]['Poster']
                st.image(
                    im,
                    width=200,
                    caption=f"{images[5][0]}: {summaries[5]}",
                )
            else:
                st.image(image="not_found.jpg", width=200)

if nav == "Contribute":
    st.header("Contribute")
    st.subheader("Are you aware of a movie that you want as a part of this database?")
    st.write(
        "Submit the title of the movie here, and we'll make sure it's there the next time you visit this site!"
    )
    with st.form("my_form"):
        title = st.text_input("Movie title", "")
        submitted = st.form_submit_button("Submit")
        if submitted:
            st.write("Movie to be updated:", title)
            st.wrte("Your contribution is much appreciated!")