| from django.core.management.base import BaseCommand
|
| from ai_chatbot.models import GlobalQA
|
| from ai_chatbot.services import EmbeddingService
|
|
|
| class Command(BaseCommand):
|
| help = 'Setup global greeting Q&A for all properties'
|
|
|
| def handle(self, *args, **options):
|
| service = EmbeddingService()
|
|
|
| global_qa = [
|
|
|
| ("Hello", "Hello! How can I help you with this property?", "en", 10),
|
| ("Hi", "Hi there! Feel free to ask me any questions about this property.", "en", 10),
|
| ("Hey", "Hey! How can I assist you today?", "en", 10),
|
| ("Good morning", "Good morning! How may I help you with this property?", "en", 9),
|
| ("Good afternoon", "Good afternoon! What would you like to know about this property?", "en", 9),
|
| ("Good evening", "Good evening! How can I assist you?", "en", 9),
|
| ("How are you?", "I'm doing great! How can I help you with this property?", "en", 8),
|
| ("Thanks", "You're welcome! Let me know if you have any other questions.", "en", 10),
|
| ("Thank you", "My pleasure! Feel free to ask anything else.", "en", 10),
|
| ("Thank you so much", "You're very welcome! Happy to help.", "en", 10),
|
| ("Ok thanks", "Anytime! Let me know if you need more information.", "en", 9),
|
| ("Bye", "Goodbye! Feel free to come back if you have more questions.", "en", 8),
|
| ("Goodbye", "Take care! Have a great day.", "en", 8),
|
|
|
|
|
| ("Assalam-o-Alaikum", "Wa Alaikum Assalam! Main ap ki kya madad kar sakta hoon?", "ur", 10),
|
| ("AoA", "Wa Alaikum Assalam! Kya sawal hai?", "ur", 10),
|
| ("Salam", "Wa Alaikum Assalam! Kaisay madad kar sakta hoon?", "ur", 10),
|
| ("Adaab", "Adaab! Kya poochna chahte ho property ke bare mein?", "ur", 9),
|
| ("Kya haal hai", "Allah ka shukar hai! Ap property ke bare mein kya poochna chahte ho?", "ur", 8),
|
| ("Theek ho", "Haan bilkul! Koi sawal poochain.", "ur", 8),
|
| ("Shukriya", "Apka khair maqdam hai! Koi aur sawal?", "ur", 10),
|
| ("Bohot shukriya", "Khushi hui madad karke! Aur poochain.", "ur", 10),
|
| ("Allah hafiz", "Allah hafiz! Phir kabhi ayen.", "ur", 8),
|
|
|
|
|
| ("Hello, kya haal hai", "Wa Alaikum Assalam! Mein theek hoon. Ap property ke bare mein kya janna chahte ho?", "both", 9),
|
| ("Hi, property available?", "Hello! Let me check the availability for you. Our agent will confirm shortly.", "both", 7),
|
| ("Salam, price kya hai", "Wa Alaikum Assalam! Please allow our agent to share detailed pricing information.", "both", 7),
|
| ]
|
|
|
| count = 0
|
| for question, answer, language, priority in global_qa:
|
| embedding = service.generate_embedding(question)
|
| obj, created = GlobalQA.objects.get_or_create(
|
| question=question,
|
| defaults={
|
| 'answer': answer,
|
| 'question_embedding': embedding,
|
| 'language': language,
|
| 'priority': priority,
|
| 'is_active': True
|
| }
|
| )
|
| if created:
|
| count += 1
|
| self.stdout.write(self.style.SUCCESS(f'✓ Added: {question[:40]}...'))
|
| else:
|
|
|
| obj.question_embedding = embedding
|
| obj.save()
|
|
|
| self.stdout.write(self.style.SUCCESS(f'\n✅ Successfully added/updated {count} global Q&A pairs!')) |
