Upload app.py

app.py

@@ -177,7 +177,7 @@ def query_summary(text):
     return text
 
 
-def query_generate(text, genres):
+def query_generate(text, genres, year):
     """
     Get image from HuggingFace Inference API
     -param text: text to generate image
@@ -185,7 +185,7 @@ def query_generate(text, genres):
     """
     API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
     headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
-    text = 'A movie Poster based on the following synopsis: \"' + text + '\". Style: ' + genres
+    text = 'A movie Poster based on the following synopsis: \"' + text + '\". Style: ' + genres + ', year ' + year
     payload = {"inputs": f"{text}",}
     
     response = requests.post(API_URL, headers=headers, json=payload)
@@ -210,10 +210,12 @@ def generate_poster(movie_data):
     -return: image of generated alternative poster
     """
 
-    # Get movie genres
+    # Get movie metadata
     genres = [i['name'] for i in eval(movie_data['genres'].values[0])]
     genres_string = ', '.join(genres)
 
+    year = movie_data['year'].values[0]
+
 
     # Get summarization of movie synopsis
     st.text("")
@@ -233,7 +235,7 @@ def generate_poster(movie_data):
 
     # Get image based on synopsis
     with st.spinner("Generating poster image..."):
-        response_content = query_generate(synopsis_sum, genres_string)
+        response_content = query_generate(synopsis_sum, genres_string, year)
 
     # Show image
     try: 
@@ -245,6 +247,7 @@ def generate_poster(movie_data):
         col1, col2, col3 = st.columns([1, 5, 1])
         with col2:
             st.image(image, caption="Movie: \"" + movie_data.title.values[0] + "\"")
+        del image
 
     except:
         col1, col2 = st.columns([5, 1])