import re
bodhi_seeds_strains = {
    "Goji OG": {
        "type": "Sativa-dominant hybrid",
        "THC_content": "18-26%",
        "flavors": "Sweet, Berry, Earthy",
        "effects": "Uplifted, Euphoric, Energetic",
        "description": "Goji OG is a sativa-dominant hybrid known for its sweet berry and earthy flavors. It's known for energetic and uplifting effects, making it a favorite for daytime use."
    },
    "Snow Lotus": {
        "type": "Indica-dominant hybrid",
        "THC_content": "18-22%",
        "flavors": "Herbal, Pine, Sweet",
        "effects": "Relaxing, Calming, Euphoric",
        "description": "Snow Lotus is an indica-dominant strain that delivers calming and euphoric effects. It's perfect for unwinding after a long day."
    },
    "Sunshine Daydream": {
        "type": "Indica-dominant hybrid",
        "THC_content": "20-25%",
        "flavors": "Berry, Earthy, Creamy",
        "effects": "Happy, Relaxed, Uplifted",
        "description": "Sunshine Daydream is an indica-dominant hybrid with creamy berry flavors. Its effects are relaxing and uplifting, ideal for stress relief."
    },
    "Tiger Mountain": {
        "type": "Hybrid",
        "THC_content": "Unknown",
        "flavors": "Spicy, Earthy, Pine",
        "effects": "Uplifting, Focused, Relaxing",
        "description": "Tiger Mountain is a balanced hybrid that provides a spicy, earthy aroma with uplifting and focused effects, making it great for both daytime and nighttime use."
    },
    "Temple": {
        "type": "Sativa-dominant hybrid",
        "THC_content": "20-22%",
        "flavors": "Spicy, Herbal, Sweet",
        "effects": "Energizing, Creative, Uplifting",
        "description": "Temple is a sativa-dominant hybrid with energizing effects and a sweet herbal aroma. It promotes creativity and focus, making it a great choice for daytime use."
    },
    "The Fuzz": {
        "type": "Indica-dominant hybrid",
        "THC_content": "19-23%",
        "flavors": "Sweet, Earthy, Skunky",
        "effects": "Relaxing, Sleepy, Euphoric",
        "description": "The Fuzz is an indica-dominant hybrid known for its strong relaxation effects and skunky, earthy flavor. It's a perfect strain for relaxing in the evening and managing stress."
    },
    "Cherry Lotus": {
        "type": "Indica-dominant hybrid",
        "THC_content": "20-24%",
        "flavors": "Cherry, Sweet, Fruity",
        "effects": "Euphoric, Relaxed, Happy",
        "description": "Cherry Lotus is an indica-dominant hybrid known for its sweet cherry flavor and deeply relaxing and euphoric effects."
    },
    "Blueberry Hashplant": {
        "type": "Indica-dominant hybrid",
        "THC_content": "17-21%",
        "flavors": "Blueberry, Sweet, Earthy",
        "effects": "Relaxed, Sleepy, Euphoric",
        "description": "Blueberry Hashplant is an indica-dominant hybrid with a sweet blueberry flavor profile. It provides relaxing, sleepy effects, ideal for winding down at night."
    },
    "Soul Mate": {
        "type": "Hybrid",
        "THC_content": "Unknown",
        "flavors": "Citrus, Earthy, Fruity",
        "effects": "Euphoric, Uplifted, Happy",
        "description": "Soul Mate is a balanced hybrid with fruity, citrus flavors. It delivers uplifting and euphoric effects, making it great for social situations."
    },
    "Tiger's Milk": {
        "type": "Indica-dominant hybrid",
        "THC_content": "18-23%",
        "flavors": "Creamy, Earthy, Sweet",
        "effects": "Relaxed, Euphoric, Sleepy",
        "description": "Tiger's Milk is an indica-dominant hybrid with a creamy, earthy flavor. It provides deeply relaxing and euphoric effects, making it ideal for nighttime use."
    },
    "Mother's Milk": {
        "type": "Sativa-dominant hybrid",
        "THC_content": "20-26%",
        "flavors": "Creamy, Sweet, Herbal",
        "effects": "Happy, Relaxed, Uplifted",
        "description": "Mother's Milk is a sativa-dominant hybrid known for its creamy, sweet aroma and uplifting, happy effects."
    },
    "Dream Lotus": {
        "type": "Hybrid",
        "THC_content": "18-23%",
        "flavors": "Berry, Herbal, Earthy",
        "effects": "Relaxed, Happy, Euphoric",
        "description": "Dream Lotus is a balanced hybrid with berry and earthy flavors. Its relaxing effects make it perfect for evening use."
    },
    "Dragon Fruit": {
        "type": "Sativa-dominant hybrid",
        "THC_content": "20-24%",
        "flavors": "Tropical, Sweet, Citrus",
        "effects": "Energized, Euphoric, Creative",
        "description": "Dragon Fruit is a tropical and sweet strain with energetic effects that are great for creativity and productivity."
    },
    "Hashplant #4": {
        "type": "Indica-dominant hybrid",
        "THC_content": "17-23%",
        "flavors": "Earthy, Pine, Spicy",
        "effects": "Relaxing, Sedative, Euphoric",
        "description": "Hashplant #4 delivers deeply relaxing effects, with earthy and pine notes, making it perfect for unwinding."
    },
    "Ancient OG": {
        "type": "Indica-dominant hybrid",
        "THC_content": "20-24%",
        "flavors": "Lemon, Earthy, Spicy",
        "effects": "Calm, Sleepy, Euphoric",
        "description": "Ancient OG is an indica-dominant strain with lemony and earthy flavors. It has strong sedative effects, perfect for evening relaxation."
    },
    "Love Triangle": {
        "type": "Hybrid",
        "THC_content": "20-26%",
        "flavors": "Citrus, Pine, Sweet",
        "effects": "Happy, Relaxed, Euphoric",
        "description": "Love Triangle offers balanced effects with a citrusy and sweet flavor. It promotes relaxation and happiness."
    },
    "Strawberry Mountain": {
        "type": "Indica-dominant hybrid",
        "THC_content": "18-22%",
        "flavors": "Strawberry, Sweet, Earthy",
        "effects": "Relaxed, Happy, Sleepy",
        "description": "Strawberry Mountain delivers a sweet, fruity flavor profile with deeply relaxing effects that make it a favorite for unwinding and sleep support."
    },
    "Purple Lotus": {
        "type": "Indica-dominant hybrid",
        "THC_content": "Unknown",
        "flavors": "Grape, Sweet, Earthy",
        "effects": "Relaxed, Calm, Sleepy",
        "description": "Purple Lotus has a rich grape and earthy flavor, with deeply relaxing effects, great for managing stress and promoting sleep."
    },
    "Cobra Lips": {
        "type": "Hybrid",
        "THC_content": "Unknown",
        "flavors": "Sweet, Spicy, Earthy",
        "effects": "Euphoric, Relaxed, Happy",
        "description": "Cobra Lips is a balanced hybrid with sweet and spicy flavors. It delivers relaxing and euphoric effects, ideal for creative and social settings."
    }
}
class BodhiSeedsChatbot:
    def __init__(self, knowledge_base):
        self.knowledge_base = knowledge_base

    def get_strain_info(self, strain_name):
        """Get information about a strain if it exists in the knowledge base."""
        strain = self.knowledge_base.get(strain_name)
        if strain:
            return (f"Strain: {strain_name}\n"
                    f"Type: {strain['type']}\n"
                    f"THC Content: {strain['THC_content']}\n"
                    f"Flavors: {strain['flavors']}\n"
                    f"Effects: {strain['effects']}\n"
                    f"Description: {strain['description']}")
        else:
            return f"Sorry, I don't have information about the strain '{strain_name}'."

    def extract_strain_name(self, user_input):
        """Basic method to extract the strain name from the user's question."""
        for strain_name in self.knowledge_base.keys():
            if re.search(strain_name, user_input, re.IGNORECASE):
                return strain_name
        return None

    def respond(self, user_input):
        """Generate a response based on the user's input."""
        strain_name = self.extract_strain_name(user_input)
        if strain_name:
            return self.get_strain_info(strain_name)
        else:
            return "Sorry, I didn't understand that. Please ask about a specific strain from Bodhi Seeds."

# Main logic for interaction
if __name__ == "__main__":
    # Initialize the chatbot with the Bodhi Seeds strains knowledge base
    bot = BodhiSeedsChatbot(bodhi_seeds_strains)

    print("Bodhi Seeds Strains Chatbot: Ask me about Bodhi Seeds strains!")

    while True:
        user_input = input("You: ")
        if user_input.lower() in ['exit', 'quit']:
            print("Chatbot: Goodbye!")
            break
        response = bot.respond(user_input)
        print(f"Chatbot: {response}")