select box

app.py

@@ -8,16 +8,24 @@ import random
 
 random_number = random.randint(3000, 10000)
 model = GroqModel('llama-3.1-70b-versatile', api_key='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')
-agent = Agent(model,system_prompt=(
+
+
+async def onClick(vacationType,familyType):
+    agent = Agent(model,system_prompt=(
     f"imagine the year is {random_number} and space travel is normal", 
     "Mankind has discovered aliens some are more advance then us some are not",
-    "You are a travel agent and depending on the budget low, medium and high you will list down some destinations",
-
-))
+    "You are a travel agent and depending on the" + vacationType +"you will list down some destinations",
+    "Also answer based on "+familyType+ ""
+    ))
+    result_1 = agent.run_sync("for space travel which planet is the best")
+    print(result_1)
+    
+def asyncOnClick(vacationType,familyType):
+    asyncio.run(onClick(vacationType,familyType))
 
 async def main():
 
-    result_1 = agent.run_sync("for space travel which planet is the best")
+    #result_1 = agent.run_sync("for space travel which planet is the best")
 ##The Zorvath
 
 
@@ -28,7 +36,14 @@ async def main():
 #print(result_2.data)
     
 
-    st.markdown(result_1.data)
+    #st.markdown(result_1.data)
+    #st.button(label="Search",type="primary",on_click=onClick)
+
+    vacationType = st.selectbox(label="Select Vacation type",options=("Budget","Mid-budget","High-end"))
+    family = st.selectbox(label="Select Family type",options=("Solo","Small family","Large family","Destination weeding"))
+
+    if st.button("Search"):
+        asyncOnClick(vacationType,family)
 
 if __name__ == '__main__':
     import asyncio