Update app.py

app.py

@@ -70,14 +70,14 @@ print(top_results)
 # HUGGING FACE PROJECT
 client = InferenceClient("Qwen/Qwen2.5-72B-instruct")
 
-def respond(message, history): # Generate a response using the most relevant travel info chunks
+def respond(message, history, name, country): # Generate a response using the most relevant travel info chunks
     top_chunks = get_top_chunks(message, chunk_embeddings, cleaned_chunks)
     str_top_chunks = "\n".join(top_chunks)
     
     messages = [
         {
             "role": "system",
-            "content": f"You are a friendly travel agent. Base your response on {top_chunks}. Don't create anything, only take from our 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 {top_chunks}. The user's name is {name} and they are traveling to {country}. Don't create anything, only take from our data. Help the customer to plan outfits for their vacation and etiquette for the country."
         }, 
         {
             "role": "user",
@@ -98,12 +98,18 @@ def respond(message, history): # Generate a response using the most relevant tra
 
     return response["choices"][0]["message"]["content"].strip()
 
+
 # SURVEY
-def submit_survey(name, gender):
-    print(f"Survey received - Name: {name}, Gender: {gender}")
-    return f"Thank you, {name}! Your survey response has been recorded."
+def submit_survey(name, country): 
+    print(f"Survey received - Name: {name}, Country: {country}")
+    return f"Thank you, {name}! Your survey response has been recorded.", name, country
+
 
 with gr.Blocks() as demo:
+    name_state = gr.State()
+    country_state = gr.State()
+    chat_state = gr.State([])
+
     gr.Markdown("## Survey")
 
     with gr.Column():
@@ -114,14 +120,22 @@ with gr.Blocks() as demo:
         survey_output = gr.Textbox(label="Response", interactive=False)
 
     submit_btn.click(
-        fn=lambda name, country: f"{name} and {country}",
+        fn=submit_survey,
         inputs=[name, country],
-        outputs=survey_output
+        outputs=[survey_output, name_state, country_state]
     )
 
     gr.Markdown("---")
     gr.Markdown("## Chatbot")
-    gr.ChatInterface(respond)
 
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Ask me anything about your trip!")
+    send_btn = gr.Button("Send")
+
+    send_btn.click(
+        fn=respond,
+        inputs=[msg, chat_state, name_state, country_state],
+        outputs=chatbot
+    )
 
 demo.launch()