CropSeek-LLM / mock_models /chat_engine.py
persadian's picture
Create chat_engine.py
a76e990 verified
raw
history blame contribute delete
661 Bytes
class AgriculturalChatEngine:
def __init__(self):
self.responses = {
'fertilizer': "Organic compost is recommended for most crops in your area.",
'pest': "Neem oil spray is effective against common pests while being eco-friendly.",
'default': "For optimal results, maintain proper irrigation and soil pH levels."
}
def generate_response(self, query):
query = query.lower()
for key in self.responses:
if key in query:
return f"🌱 Agricultural Assistant: {self.responses[key]}"
return f"🌱 Agricultural Assistant: {self.responses['default']}"