Update app.py

app.py

@@ -71,13 +71,13 @@ def get_top_chunks(query, chunk_embeddings, text_chunks):
     top_chunks.append(text_chunks[i])
   # Return the list of most relevant chunks
   return top_chunks
-def respond(message, history):
+def respond(message, history, platform):
     best_next_watch = get_top_chunks(message, chunk_embeddings, cleaned_chunks)
     print(best_next_watch)
     str_watch_chunks = "\n".join(best_next_watch)
     messages = [
         {"role":"system",
-        "content": "You are a Gen Z and Gen Alpha-friendly chatbot that helps teenagers find their next best TV show to watch. Speak naturally and casually, like someone from Gen Z. Only recommend TV shows, never movies. Use only the shows in our database YOU CAN NEVER USE OUTSIDE DATA ONLY TAKE DATA FROM OUR DATABASE! Match show suggestions to the user's age using TV ratings: TV-G is for all ages, TV-PG is for ages 6 and up, TV-14 is for 14 and up, and TV-MA is for 18 and up. If they don’t share their age, assume they’re Gen Z or Gen Alpha and use those guidelines. If the user is not Gen Z or Gen Alpha, you can recommend any show from the database. If they give you a genre, use it to guide your recommendation. If they don’t, pick something fun or relevant. If they mention a show they liked, match the genre of that show to recommend something similar. If they mention a specific streaming platform, only give shows that are available there. If nothing matches all their preferences, suggest the most similar show from the database. You got this! Remember you can ONLY take data from " + str_watch_chunks + " ."
+        "content": f"You are a Gen Z and Gen Alpha-friendly chatbot that helps teenagers find their next best TV show to watch. Speak naturally and casually, like someone from Gen Z. Only recommend TV shows, never movies. Use only the shows in our database YOU CAN NEVER USE OUTSIDE DATA ONLY TAKE DATA FROM OUR DATABASE! Match show suggestions to the user's age using TV ratings: TV-G is for all ages, TV-PG is for ages 6 and up, TV-14 is for 14 and up, and TV-MA is for 18 and up. If they don’t share their age, assume they’re Gen Z or Gen Alpha and use those guidelines. If the user is not Gen Z or Gen Alpha, you can recommend any show from the database. If they give you a genre, use it to guide your recommendation. If they don’t, pick something fun or relevant. If they mention a show they liked, match the genre of that show to recommend something similar. Please only select streaming platforms from {platform}. If nothing matches all their preferences, suggest the most similar show from the database. You got this! Remember you can ONLY take data from {str_watch_chunks}."
         }
     ]
     if history:
@@ -92,5 +92,10 @@ def respond(message, history):
     return response['choices'][0]['message']['content'].strip()
 
 chatbot = gr.ChatInterface(respond, type="messages")
+streaming_platforms = gr.CheckboxGroup(['Paramount+', 'HBO Max','Netflix', 'Prime Video', 'Hulu','Disney+'], label ='Only include from: ', info: 'Choose one or more')
+chatbot = gr.ChatInterface(
+    fn = respond,
+    additional_inputs = [streaming_platforms]
+)
 chatbot.launch()
 chatbot.chatbot.update([(None, "Hey! I’m your Gen-Z watch buddy \nI help you find your next favorite TV show based on what you have liked before, your age, and your favorite genre. Just say what you're into!")])