Upload app.py

app.py

@@ -14,6 +14,9 @@ from datetime import time, datetime
 from zipfile import ZipFile
 from sentence_transformers import SentenceTransformer
 from diffusers import DiffusionPipeline
+from htbuilder import HtmlElement, div, ul, li, br, hr, a, p, img, styles, classes, fonts
+from htbuilder.units import percent, px
+from htbuilder.funcs import rgba, rgb
 
 
 ###############################
@@ -39,13 +42,104 @@ from kaggle.api.kaggle_api_extended import KaggleApi
 ## ------- FUNCTIONS ------- ##
 ###############################
 
-#@st.cache(persist=True, show_spinner=False, suppress_st_warning=True)
+def link(link, text, **style):
+    return a(_href=link, _target="_blank", style=styles(**style))(text)
+
+def image(src_as_string, **style):
+    return img(src=src_as_string, style=styles(**style))
+
+def layout(*args):
+
+    style = """
+    <style>
+      # MainMenu {visibility: hidden;}
+      footer {visibility: hidden;}
+     .stApp { bottom: 105px; }
+    </style>
+    """
+
+    style_div = styles(
+        position="fixed",
+        left=0,
+        bottom=0,
+        margin=px(0, 0, 0, 0),
+        width=percent(100),
+        color="black",
+        text_align="center",
+        height="auto",
+        opacity=1
+    )
+
+    style_hr = styles(
+        display="block",
+        margin=px(8, 8, "auto", "auto"),
+        border_style="inset",
+        border_width=px(2)
+    )
+
+    body = p()
+    foot = div(
+        style=style_div
+    )(
+        hr(
+            style=style_hr
+        ),
+        body
+    )
+
+    st.markdown(style, unsafe_allow_html=True)
+
+    for arg in args:
+        if isinstance(arg, str):
+            body(arg)
+
+        elif isinstance(arg, HtmlElement):
+            body(arg)
+
+    st.markdown(str(foot), unsafe_allow_html=True)
+
+def footer():
+    myargs = [
+        "Made in ",
+        image('https://avatars3.githubusercontent.com/u/45109972?s=400&v=4',
+              width=px(25), height=px(25)),
+        " with ❤️ by ",
+        link("https://www.linkedin.com/in/gaspar-avit/", "Gaspar Avit"),
+    ]
+    layout(*myargs)
+
+def authenticate_kaggle():
+    # Connect to kaggle API
+
+    if not os.path.exists(PATH_JSON):
+        api_token = {"username":st.secrets['username'],"key":st.secrets['key']}
+        with open(PATH_JSON, 'w') as file:
+            json.dump(api_token, file)
+
+
+    # Activate Kaggle API
+    api = KaggleApi()
+    api.authenticate()
+
+
+    try:
+        api.authenticate()
+    except:
+        with open('/home/appuser/.kaggle/kaggle.json', 'w') as file:
+            json.dump(api_token, file)
+        api.authenticate() 
+
+
 @st.experimental_memo(persist=True, show_spinner=False, suppress_st_warning=True)
 def load_dataset():
     """
     Load Dataset from Kaggle
     -return: dataframe containing dataset
     """
+
+    # Connect to kaggle API
+    authenticate_kaggle()
+
     # Downloading Movies dataset
     api.dataset_download_file('rounakbanik/the-movies-dataset', 'movies_metadata.csv')
 
@@ -105,69 +199,57 @@ def generate_poster(movie_data):
     return None #poster_image
 
 
-###############################
-## --- CONNECT TO KAGGLE --- ##
-###############################
-
 
-if not os.path.exists(PATH_JSON):
-    api_token = {"username":st.secrets['username'],"key":st.secrets['key']}
-    with open(PATH_JSON, 'w') as file:
-        json.dump(api_token, file)
-
-
-# Activate Kaggle API
-api = KaggleApi()
-api.authenticate()
 
+###############################
+## --------- MAIN ---------- ##
+###############################
 
-try:
-    api.authenticate()
-except:
-    with open('/home/appuser/.kaggle/kaggle.json', 'w') as file:
-        json.dump(api_token, file)
-    api.authenticate() 
 
+if __name__ == "__main__":
 
 
-###############################
-## --------- MAIN ---------- ##
-###############################
+    # Initialize image variable
+    image = None
 
-image = None
+    ## Create dataset
+    data = load_dataset()
 
-# Create dataset
-data = load_dataset()
 
+    ## --- Page config --- ##
+    # Set page title
+    st.title("""
+    Alternative Movie Poster Generator :film_frames:
+    This is a movie poster generator based on movie's synopsis :sunglasses:
+    Just select the title of a movie to generate an alternative poster.
+     """)
 
-st.title("""
-Alternative Movie Poster Generator :film_frames:
-This is a movie poster generator based on movie's synopsis :sunglasses:
-Just select the title of a movie to generate an alternative poster.
- """)
+    ## Set page footer
+    footer()
+    ## ------------------- ##
 
-st.text("")
-st.text("")
-st.text("")
-st.text("")
+    st.text("")
+    st.text("")
+    st.text("")
+    st.text("")
 
-session.selected_movie = st.selectbox(label="Select a movie to generate alternative poster", options=data.title)
+    session.selected_movie = st.selectbox(label="Select a movie to generate alternative poster", options=data.title)
 
-st.text("")
-st.text("")
+    st.text("")
+    st.text("")
 
-buffer1, col1, buffer2 = st.columns([1.3, 1, 1])
+    buffer1, col1, buffer2 = st.columns([1.3, 1, 1])
 
-is_clicked = col1.button(label="Generate poster!")
+    is_clicked = col1.button(label="Generate poster!")
 
 
-if is_clicked:
-    image = generate_poster(data[data.title==session.selected_movie])
+    if is_clicked:
+        image = generate_poster(data[data.title==session.selected_movie])
 
-st.text("")
-st.text("")
-st.text("")
-st.text("")
+    st.text("")
+    st.text("")
+    st.text("")
+    st.text("")
 
-#if data is not None:
-#    st.table(data)
+    #if image is not None:
+    #    st.image(image, caption=session.selected_movie.title.values[0])