updated examples

app.py

@@ -62,6 +62,7 @@ def respond(
   openai_chat_history,
   min_year,
   max_year,
+  allowed_movies,
 ):
   '''
   :param user_message: string, the user's message
@@ -172,15 +173,31 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
       retry_btn = gr.Button('Retry', variant='secondary',)
       clear_btn = gr.Button('Clear', variant='secondary')
 
+    # Filters
+    gr.Markdown("<h4 style='margin-left: 1rem'>Filters</h4>")
+
     # let user pick minimum and maximum year
     min_year = gr.Slider(1900, 2024, 1900, label='Min year', interactive=True,)
     max_year = gr.Slider(1900, 2024, 2024, label='Max year', interactive=True,)
 
+    # let user write down a list of movies. results must be a subset of this list
+    allowed_movies = gr.Textbox(
+      # container=False,
+      show_label=True,
+      label='Allowed movie list',
+      placeholder='Type a list of movies. Result will be a subset of this list.',
+    )
+
     # example inputs
     gr.Examples(
       [
-        'Please recommend some movies with lots of jumpscares or something with lots of blood.. I want to watch some movie that will not let me nor my cousins sleep soundly tonight.',
-        "Which movie is better? Warrior (2011) or Southpaw (2015)?. I'm looking to watch a boxing movie, and am not sure what to pick between Warrior (2011) or Southpaw (2015). I'm a big fan of both, Jake Gyllenhall and Tom Hardy and honestly just couldn't pick between the two",
+        # 'Please recommend some movies with lots of jumpscares or something with lots of blood.. I want to watch some movie that will not let me nor my cousins sleep soundly tonight.',
+        # "Which movie is better? Warrior (2011) or Southpaw (2015)?. I'm looking to watch a boxing movie, and am not sure what to pick between Warrior (2011) or Southpaw (2015). I'm a big fan of both, Jake Gyllenhall and Tom Hardy and honestly just couldn't pick between the two",
+        "What are some good action movies for kids under 10?",
+        "I want to watch a film about space travel but not too scientific.",
+        "I need a family-friendly movie that deals with environmental themes for an Earth Day event.",
+        "Find culturally diverse films for an international film festival at our college.",
+        "What are some uplifting movies for a charity fundraiser supporting mental health awareness?",
       ],
       inputs=[input_box],
     )
@@ -190,7 +207,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     triggers=[input_box.submit, submit_btn.click],
     fn=respond,
     api_name='respond',
-    inputs=[input_box, chatbot, openai_history_state, min_year, max_year,],
+    inputs=[input_box, chatbot, openai_history_state, min_year, max_year, allowed_movies,],
     outputs=[input_box, chatbot, openai_history_state],
   )
 

tools.py

@@ -6,12 +6,13 @@ import openai
 
 openai.api_key = os.environ['OPENAI_API_KEY']
 
-def get_movie_recs(context, K=5, min_year=1900, max_year=2024,):
+def get_movie_recs(context, K=5, min_year=1900, max_year=2024, allowed_movies=None):
+  allow_list = "" if allowed_movies is None else f"Restrict responses to movies in this list: {allowed_movies}."
   system_prompt=f"""
   Pretend you are a movie recommendation system. I will give you a conversation between a user
   and you (a recommender system). Based on the conversation, reply to me with {K} recommendations
   without extra sentences. Give the year the movie was released in your response, e.g. Inception (2010).
-  Give movies that were released between the years {min_year} and {max_year}.
+  Give movies that were released between the years {min_year} and {max_year}. {allow_list}
   """
   user_query=f"Here is the conversation:\n{context}"
   response = openai.ChatCompletion.create(