Upload app.py

app.py

@@ -177,7 +177,7 @@ def query_summary(text):
     return text
 
 
-def query_generate(text):
+def query_generate(text, genres):
     """
     Get image from HuggingFace Inference API
     -param text: text to generate image
@@ -185,7 +185,8 @@ def query_generate(text):
     """
     API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
     headers = {"Authorization": f"Bearer {st.secrets['hf_token']}"}
-    text = "Poster of movie. " + text
+    text = "Poster of movie. " + text + '. ' + genres
+    st.wrtie(text)
     payload = {"inputs": f"{text}",}
     
     response = requests.post(API_URL, headers=headers, json=payload)
@@ -209,6 +210,10 @@ def generate_poster(movie_data):
     -return: image of generated alternative poster
     """
 
+    # Get movie genres
+    genres = [i['name'] for i in eval(movie_data['genres'])]
+    genres_string = ', '.join(genres)
+
     # Get summarization of movie synopsis
     st.text("")
     with st.spinner("Please wait while the synopsis is being summarized..."):
@@ -227,7 +232,7 @@ def generate_poster(movie_data):
 
     # Get image based on synopsis
     with st.spinner("Generating poster image..."):
-        response_content = query_generate(synopsis_sum)
+        response_content = query_generate(synopsis_sum, genres_string)
 
     # Show image
     try: