Update app.py

app.py

@@ -74,7 +74,7 @@ print(top_results)
 # HUGGING FACE PROJECT
 client = InferenceClient("Qwen/Qwen2.5-72B-instruct")
 
-def respond(message, history, name, country): # Generate a response using the most relevant travel info chunks
+def respond(message, history, name, country, season): # Generate a response using the most relevant travel info chunks
     new_message = message + " " + country
     top_chunks = get_top_chunks(new_message, chunk_embeddings, cleaned_chunks)
     str_top_chunks = "\n".join(top_chunks)
@@ -82,7 +82,7 @@ def respond(message, history, name, country): # Generate a response using the mo
     messages = [
         {
             "role": "system",
-            "content": f"You are a friendly travel agent. Base your response on the following data: {top_chunks}. The user's name is {name} and they are traveling to {country}. Don't create anything, only take from our data. Look for {country} in the data and pull the specific information from the data. Help the customer to plan outfits for their vacation and etiquette for the country."
+            "content": f"You are a friendly travel agent. Base your response on the following data: {top_chunks}. The user's name is {name} and they are traveling to {country} in the {season}. Don't create anything, only take from our data. Look for {country} in the data and pull the specific information from the data. Help the customer to plan outfits for their vacation and etiquette for the country."
         }, 
         {
             "role": "user",
@@ -111,8 +111,8 @@ def respond(message, history, name, country): # Generate a response using the mo
 
 
 # SURVEY
-def submit_survey(name, country): 
-    print(f"Survey received - Name: {name}, Country: {country}")
+def submit_survey(name, country, season): 
+    print(f"Survey received - Name: {name}, Country: {country}, Season: {season}")
     return f"Thank you, {name}! Your survey response has been recorded. Ask Miles about any questions you have!", name, country
 
 class MyCustomTheme(gr.themes.Base):
@@ -149,14 +149,15 @@ with gr.Blocks(theme=custom_theme, css="* { color: #0c265b !important; }") as de
     with gr.Column():
         name = gr.Textbox(label="Name", placeholder="Enter your name")
         country = gr.Dropdown(label="Country", choices=["Argentina", "Australia", "Brazil", "Canada", "China", "Costa Rica", "Ecuador", "Egypt", "England", "France", "Germany", "Greece", "India", "Ireland", "Italy", "Jamaica", "Japan", "Kenya", "Malaysia", "Maldives", "Mexico", "Morocco", "Netherlands", "New Zealand", "Norway", "Panama", "Peru", "Philippines", "Poland", "Portugal", "Russia", "South Africa", "South Korea", "Spain", "Sweden", "Switzerland", "Taiwan", "Thailand", "Turkey", "USA", "Vietnam", "Zimbabwe"])
-
+        season = gr.Radio(label="Season", choices=["Spring", "Summer", "Autumn", "Winter"])
+        
         submit_btn = gr.Button("Submit Survey")
         survey_output = gr.Textbox(label="Response", interactive=False)
 
     submit_btn.click(
         fn=submit_survey,
-        inputs=[name, country],
-        outputs=[survey_output, name_state, country_state]
+        inputs=[name, country, season],
+        outputs=[survey_output, name_state, country_state, season_state]
     )
 
     gr.Markdown("---")
@@ -167,9 +168,9 @@ with gr.Blocks(theme=custom_theme, css="* { color: #0c265b !important; }") as de
     send_btn = gr.Button("Send")
 
     send_btn.click(
-    fn=respond,
-    inputs=[msg, chat_state, name_state, country_state],
-    outputs=[chat_state, chat_state]
+        fn=respond,
+        inputs=[msg, chat_state, name_state, country_state, season_state],
+        outputs=[chat_state, chat_state]
     ).then(
         fn=lambda history: history,
         inputs=[chat_state],