import streamlit as st
from pydantic_ai import Agent
from pydantic_ai.models.groq import GroqModel
import nest_asyncio
import random



random_number = random.randint(3000, 10000)
model = GroqModel('llama-3.1-70b-versatile', api_key='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')


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" + 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")
##The Zorvath


#print(result_1.data)

#result_2 = agent.run_sync("What are the major tourust attraction in planet The Zorvath",message_history=result_1.new_messages())

#print(result_2.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
    nest_asyncio.apply()
    asyncio.run(main())