from gift_finder_model import GiftFinderModel

# Initialize the model
model = GiftFinderModel()

# Example conversation
conversation = """
USER: I need to find a gift for my brother's birthday. He's in his 20s and loves playing basketball and video games. I don't want to spend more than $50.
ASSISTANT: I'd be happy to help you find a birthday gift for your brother who loves basketball and video games within your $50 budget.
"""

# Process the conversation
result = model.process_conversation(conversation)

# Print the result
print("Extracted Data:")
print(result["extracted_data"])
print("\nGenerated Gift Search Query:")
print(result["gift_search_query"])