Update app.py

app.py

@@ -115,12 +115,12 @@ def get_top_chunks(query, chunk_embeddings, text_chunks):
 
 
 # ======================================
-def respond(message, history):
+def respond(message, history, hobby_type, age):
     best_chunks = get_top_chunks(message, chunk_embeddings, cleaned_chunks) 
     print(best_chunks)
 
     str_chunks = "/n".join(best_chunks)
-    messages = [{'role':'system', 'content': 'You are a very kind chatbot giving people hobby suggestions to help them spend less time on their electronic devices. You answer their questions based on ' + str_chunks + '.'}]
+    messages = [{'role':'system', 'content': f'You are a very kind chatbot giving people hobby suggestions to help them spend less time on their electronic devices. Make sure to be specific to {hobby_type} hobbies. You answer their questions based on ' + str_chunks + '.'}]
       
     if history:
         messages.extend(history)
@@ -135,8 +135,11 @@ def respond(message, history):
 
 with gr.Blocks(theme='earneleh/paris') as chatbot:
     with gr.Row(scale=1):
+        with gr.Column(scale=1):
+            hobby_type = gr.CheckboxGroup(['Long-term budget plan', 'Short-term budget plan','Cut back spending'], label = "What category of hobbies are you interested in?")
+            age = gr.Textbox(label = "How old are you", info = "Enter your age", placeholder = "Type a sentence here...")
         gr.Image("ai.png")
     with gr.Row(scale=1):
-        gr.ChatInterface(respond, type= 'messages')
+        gr.ChatInterface(fn=respond, additional_inputs = [hobby_type, age], type= "messages")
 
 chatbot.launch()