Create knowledgebase.py

knowledgebase.py

@@ -0,0 +1,542 @@
+"""
+STANLEY AI - Kiswahili Knowledge Base
+Advanced cultural and linguistic database for enhanced AI responses
+"""
+
+class KiswahiliKnowledgeBase:
+    def __init__(self):
+        self.kiswahili_data = self._initialize_kiswahili_data()
+        self.cultural_context = self._initialize_cultural_context()
+        self.lion_king_lore = self._initialize_lion_king_lore()
+        self.advanced_phrases = self._initialize_advanced_phrases()
+    
+    def _initialize_kiswahili_data(self):
+        """Comprehensive Kiswahili vocabulary and grammar"""
+        return {
+            "greetings": {
+                "formal": {
+                    "Shikamoo": "Greeting to elders (response: Marahaba)",
+                    "Marahaba": "Response to Shikamoo",
+                    "Habari za asubuhi": "Good morning news",
+                    "Habari za mchana": "Good afternoon news", 
+                    "Habari za jioni": "Good evening news",
+                    "Habari gani?": "What news? (How are you?)"
+                },
+                "informal": {
+                    "Mambo": "What's up? (response: Poa)",
+                    "Poa": "Cool (response to Mambo)",
+                    "Vipi?": "How? What's up?",
+                    "Sema": "Speak (greeting)",
+                    "Niaje": "How's it going?"
+                },
+                "responses": {
+                    "Nzuri": "Good",
+                    "Safi": "Clean/Cool", 
+                    "Salama": "Peaceful",
+                    "Poa kichizi kama ndizi": "Cool crazy like a banana",
+                    "Freshi": "Fresh"
+                }
+            },
+            
+            "common_phrases": {
+                "courtesy": {
+                    "Asante": "Thank you",
+                    "Asante sana": "Thank you very much",
+                    "Tafadhali": "Please",
+                    "Samahani": "Sorry/Excuse me",
+                    "Pole": "Sorry (for misfortune)",
+                    "Pole sana": "Very sorry",
+                    "Karibu": "You're welcome",
+                    "Karibu sana": "You're very welcome",
+                    "Starehe": "Make yourself comfortable"
+                },
+                "essential": {
+                    "Ndio": "Yes",
+                    "Hapana": "No",
+                    "Sawa": "Okay",
+                    "Labda": "Maybe",
+                    "Bila shaka": "Without doubt",
+                    "Hamna shida": "No problem",
+                    "Hakuna matata": "No worries",
+                    "Hakuna wasiwasi": "No worries (more formal)"
+                }
+            },
+            
+            "lion_king_vocabulary": {
+                "characters": {
+                    "Simba": "Lion",
+                    "Nala": "Gift",
+                    "Rafiki": "Friend", 
+                    "Mufasa": "King",
+                    "Sarabi": "Mirage",
+                    "Sarafina": "Bright star",
+                    "Pumbaa": "Simpleton/Be careless",
+                    "Timon": "Respect",
+                    "Shenzi": "Barbarian/Uncouth",
+                    "Banzai": "Skulk/Move stealthily",
+                    "Ed": "Short for Edward (no Swahili meaning)"
+                },
+                "locations": {
+                    "Pride Rock": "Mwamba wa Pride",
+                    "Elephant Graveyard": "Makaburi ya Tembo",
+                    "Watering Hole": "Kisima cha Maji",
+                    "Jungle": "Msitu",
+                    "Savannah": "Nyika"
+                },
+                "concepts": {
+                    "Circle of Life": "Mduara wa Uhai",
+                    "King": "Mfalme",
+                    "Queen": "Malkia",
+                    "Prince": "Mwana Mfalme",
+                    "Pride": "Fahari/Majivuno",
+                    "Hunt": "Windla",
+                    "Roar": "Koroma"
+                }
+            },
+            
+            "grammar_essentials": {
+                "noun_classes": {
+                    "M-/WA-": "People (mtu - person, watu - people)",
+                    "M-/MI-": "Plants, natural things (mti - tree, miti - trees)",
+                    "JI-/MA-": "Fruits, foreign objects (tunda - fruit, matunda - fruits)",
+                    "KI-/VI-": "Tools, languages (kiti - chair, viti - chairs)",
+                    "N-": "Animals, things (ndege - bird/birds)",
+                    "U-": "Abstract, mass nouns (ugonjwa - disease)",
+                    "KU-": "Verbal nouns (kusoma - to read)",
+                    "PA-": "Locatives (mahali - place)",
+                    "MU-": "Locatives (ndani - inside)",
+                    "KU-": "Locatives (kule - there)"
+                },
+                "verb_tenses": {
+                    "present": "Na- (ninasoma - I am reading)",
+                    "past": "Li- (nilisoma - I read)",
+                    "future": "Ta- (nitasoma - I will read)",
+                    "perfect": "Me- (nimesoma - I have read)",
+                    "habitual": "Hu- (ninasoma - I usually read)"
+                }
+            },
+            
+            "numbers": {
+                "1": "Moja", "2": "Mbili", "3": "Tatu", "4": "Nne", "5": "Tano",
+                "6": "Sita", "7": "Saba", "8": "Nane", "9": "Tisa", "10": "Kumi",
+                "20": "Ishirini", "30": "Thelathini", "40": "Arobaini", 
+                "50": "Hamsini", "100": "Mia moja", "1000": "Elfu moja"
+            },
+            
+            "time_expressions": {
+                "days": {
+                    "Leo": "Today",
+                    "Kesho": "Tomorrow",
+                    "Jana": "Yesterday",
+                    "Jumanne": "Tuesday",
+                    "Jumatatu": "Monday",
+                    "Jumatano": "Wednesday",
+                    "Alhamisi": "Thursday",
+                    "Ijumaa": "Friday",
+                    "Jumamosi": "Saturday",
+                    "Jumapili": "Sunday"
+                },
+                "time": {
+                    "Sasa hivi": "Right now",
+                    "Baadaye": "Later",
+                    "Mara moja": "Immediately",
+                    "Asubuhi": "Morning",
+                    "Mchana": "Afternoon",
+                    "Jioni": "Evening",
+                    "Usiku": "Night",
+                    "Saa ngapi?": "What time?"
+                }
+            }
+        }
+    
+    def _initialize_cultural_context(self):
+        """East African cultural knowledge and context"""
+        return {
+            "cultural_concepts": {
+                "ujamaa": "Familyhood/Socialism - Community cooperation",
+                "harambee": "Pull together - Community self-help",
+                "ubuntu": "Humanity towards others - I am because we are",
+                "pole_pole": "Slowly slowly - Take your time, be careful",
+                "mwalimu": "Teacher - Respect for educators",
+                "mzee": "Elder - Respect for older people",
+                "dunia": "The world - Global perspective",
+                "imani": "Faith/Belief - Trust and confidence"
+            },
+            
+            "traditional_values": {
+                "respect_for_elders": "Elders are highly respected in Swahili culture",
+                "community_first": "Community needs come before individual desires",
+                "hospitality": "Guests are treated with great honor (karibu)",
+                "indirect_communication": "Often prefer indirectness to avoid conflict",
+                "time_flexibility": "African time' - more relaxed about punctuality",
+                "oral_tradition": "Strong tradition of storytelling and proverbs"
+            },
+            
+            "swahili_proverbs": {
+                "Mgeni siku mbili; siku ya tatu mpe jembe": "A guest for two days; on the third day give him a hoe",
+                "Haraka haraka haina baraka": "Hurry hurry has no blessing",
+                "Asiyefunzwa na mamae hufunzwa na ulimwengu": "He who is not taught by his mother will be taught by the world",
+                "Mwacha mila ni mtumwa": "He who abandons his culture is a slave",
+                "Pole pole ndio mwendo": "Slowly slowly is the way to go",
+                "Fimbo ya mbali haiui nyoka": "A stick from far away doesn't kill a snake",
+                "Mwenye pupa hadiriki kula tamu": "The hurried person doesn't get to eat sweets"
+            },
+            
+            "food_culture": {
+                "staple_foods": {
+                    "ugali": "Maize porridge - the main staple",
+                    "wali": "Rice",
+                    "nyama_choma": "Grilled meat",
+                    "sukuma_wiki": "Collard greens - 'push the week'",
+                    "kachumbari": "Tomato and onion salad",
+                    "pilau": "Spiced rice dish",
+                    "chapati": "Flatbread",
+                    "mandazi": "Swahili donuts"
+                },
+                "beverages": {
+                    "chai": "Tea (usually spiced)",
+                    "kahawa": "Coffee",
+                    "maji": "Water",
+                    "uji": "Porridge",
+                    "madafu": "Coconut water"
+                }
+            }
+        }
+    
+    def _initialize_lion_king_lore(self):
+        """Comprehensive Lion King story and character knowledge"""
+        return {
+            "character_backstories": {
+                "simba": {
+                    "name_meaning": "Lion",
+                    "birth": "Born to Mufasa and Sarabi at Pride Rock",
+                    "personality": "Playful, brave, initially naive but grows wise",
+                    "journey": "Exiled after Mufasa's death, raised by Timon and Pumbaa",
+                    "transformation": "Returns to reclaim his kingdom after Rafiki's guidance",
+                    "key_phrases": ["Hakuna Matata", "I just can't wait to be king"]
+                },
+                "nala": {
+                    "name_meaning": "Gift",
+                    "role": "Simba's childhood friend and future queen",
+                    "personality": "Brave, independent, strong-willed",
+                    "journey": "Leaves Pride Lands to find help, reunites with Simba",
+                    "significance": "Convinces Simba to return and claim his destiny"
+                },
+                "mufasa": {
+                    "name_meaning": "King",
+                    "role": "Wise and noble king of Pride Rock",
+                    "philosophy": "Believes in the Circle of Life",
+                    "death": "Killed by Scar in the wildebeest stampede",
+                    "legacy": "Appears as a spirit to guide Simba"
+                },
+                "rafiki": {
+                    "name_meaning": "Friend",
+                    "role": "Mystical baboon shaman and advisor",
+                    "abilities": "Spiritual wisdom, connection to ancestors",
+                    "key_scenes": ["Presenting Simba, guiding adult Simba"],
+                    "famous_line": "The past can hurt. But the way I see it, you can either run from it, or learn from it."
+                }
+            },
+            
+            "story_arcs": {
+                "circle_of_life": {
+                    "concept": "The delicate balance connecting all living things",
+                    "representation": "Shown through the ecosystem of Pride Lands",
+                    "mufasa_explanation": "We are all connected in the great Circle of Life",
+                    "violation": "Scar's rule disrupts the balance",
+                    "restoration": "Simba's return restores harmony"
+                },
+                "hakuna_matata_philosophy": {
+                    "meaning": "No worries - problem-free philosophy",
+                    "origin": "Taught to Simba by Timon and Pumbaa",
+                    "purpose": "Helps Simba cope with trauma and guilt",
+                    "limitation": "Avoidance rather than confrontation",
+                    "growth": "Simba learns to face problems rather than ignore them"
+                },
+                "redemption_journey": {
+                    "exile": "Simba runs away believing he caused Mufasa's death",
+                    "carefree_life": "Lives with Timon and Pumbaa practicing Hakuna Matata",
+                    "awakening": "Nala's arrival and Rafiki's guidance",
+                    "confrontation": "Faces Scar and the truth about Mufasa's death",
+                    "restoration": "Defeats Scar and restores the kingdom"
+                }
+            },
+            
+            "cultural_influences": {
+                "swahili_language": "Character names and phrases throughout",
+                "african_landscapes": "Based on Kenyan and Tanzanian savannahs",
+                "masai_influence": "Pride Lands inspired by Maasai Mara",
+                "african_philosophy": "Ubuntu and community values",
+                "animal_symbolism": "Traditional African animal representations"
+            }
+        }
+    
+    def _initialize_advanced_phrases(self):
+        """Advanced Kiswahili for sophisticated conversations"""
+        return {
+            "philosophical_concepts": {
+                "Umoja": "Unity - First day of Kwanzaa",
+                "Kujichagulia": "Self-determination - Second day of Kwanzaa",
+                "Ujima": "Collective work and responsibility - Third day of Kwanzaa",
+                "Ujamaa": "Cooperative economics - Fourth day of Kwanzaa",
+                "Nia": "Purpose - Fifth day of Kwanzaa", 
+                "Kuumba": "Creativity - Sixth day of Kwanzaa",
+                "Imani": "Faith - Seventh day of Kwanzaa"
+            },
+            
+            "poetic_expressions": {
+                "Moyo wangu unaimba": "My heart is singing",
+                "Upepo wa upendo": "The wind of love",
+                "Nyota za bahati": "Stars of luck",
+                "Mstari wa hatima": "The line of destiny",
+                "Sauti ya amani": "The voice of peace"
+            },
+            
+            "business_phrases": {
+                "Biashara": "Business",
+                "Mkataba": "Contract",
+                "Mapato": "Income",
+                "Gharama": "Cost",
+                "Faida": "Profit",
+                "Hasara": "Loss",
+                "Soko": "Market",
+                "Bei": "Price",
+                "Bei ya chini": "Lowest price",
+                "Tafadhali punguza bei": "Please reduce the price"
+            },
+            
+            "medical_terms": {
+                "Daktari": "Doctor",
+                "Hospitali": "Hospital",
+                "Maumivu": "Pain",
+                "Dawa": "Medicine",
+                "Kuhara": "Diarrhea",
+                "Kichwa kinaniuma": "My head hurts",
+                "Mguu unaniuma": "My leg hurts",
+                "Nina homa": "I have fever"
+            },
+            
+            "travel_phrases": {
+                "Ninaenda wapi?": "Where am I going?",
+                "Ninahitaji teksi": "I need a taxi",
+                "Bei gani?": "How much?",
+                "Ninaenda hotelini": "I'm going to the hotel",
+                "Ninaitwa...": "My name is...",
+                "Siwezi kuzungumza Kiswahili vizuri": "I can't speak Swahili well",
+                "Unaweza kusema polepole?": "Can you speak slowly?",
+                "Tafadhali nisaidie": "Please help me"
+            }
+        }
+    
+    def get_kiswahili_phrase(self, category, subcategory=None, phrase=None):
+        """Retrieve specific Kiswahili phrases"""
+        if subcategory and phrase:
+            return self.kiswahili_data.get(category, {}).get(subcategory, {}).get(phrase)
+        elif subcategory:
+            return self.kiswahili_data.get(category, {}).get(subcategory, {})
+        else:
+            return self.kiswahili_data.get(category, {})
+    
+    def get_cultural_insight(self, concept):
+        """Get cultural context for concepts"""
+        return self.cultural_context.get('cultural_concepts', {}).get(concept)
+    
+    def get_lion_king_character(self, character):
+        """Get detailed Lion King character information"""
+        return self.lion_king_lore.get('character_backstories', {}).get(character.lower())
+    
+    def get_random_proverb(self):
+        """Get a random Swahili proverb"""
+        import random
+        proverbs = self.cultural_context.get('swahili_proverbs', {})
+        if proverbs:
+            proverb, meaning = random.choice(list(proverbs.items()))
+            return f"{proverb} - '{meaning}'"
+        return "Hakuna matata - No worries"
+    
+    def generate_kiswahili_response(self, english_text, complexity="medium"):
+        """
+        Generate Kiswahili-enhanced response based on English input
+        """
+        # Analyze the input for appropriate Kiswahili integration
+        kiswahili_elements = self._analyze_for_kiswahili(english_text)
+        
+        enhanced_response = english_text
+        
+        # Add appropriate Kiswahili elements based on context
+        if kiswahili_elements.get('greeting'):
+            enhanced_response = f"Jambo! {enhanced_response}"
+        
+        if kiswahili_elements.get('lion_king_context'):
+            # Add Lion King relevant Kiswahili
+            lion_phrase = self._get_lion_king_phrase()
+            enhanced_response = f"{enhanced_response} {lion_phrase}"
+        
+        if kiswahili_elements.get('cultural_context'):
+            # Add cultural wisdom
+            proverb = self.get_random_proverb()
+            enhanced_response = f"{enhanced_response}\n\nKama wasemavyo Wasawnili: {proverb}"
+        
+        return enhanced_response
+    
+    def _analyze_for_kiswahili(self, text):
+        """Analyze text to determine appropriate Kiswahili integration"""
+        text_lower = text.lower()
+        
+        analysis = {
+            'greeting': any(word in text_lower for word in ['hello', 'hi', 'greeting']),
+            'lion_king_context': any(word in text_lower for word in ['lion king', 'simba', 'mufasa', 'rafiki']),
+            'cultural_context': any(word in text_lower for word in ['culture', 'tradition', 'africa', 'wisdom']),
+            'philosophical': any(word in text_lower for word in ['life', 'purpose', 'meaning', 'philosophy'])
+        }
+        
+        return analysis
+    
+    def _get_lion_king_phrase(self):
+        """Get appropriate Lion King Kiswahili phrase"""
+        import random
+        phrases = [
+            "Hakuna Matata!",
+            "Kumbuka wewe ni nani - Remember who you are!",
+            "Mduara wa Uhai unadumu - The Circle of Life continues!",
+            "Asante sana kwa kukumbuka - Thank you for remembering!"
+        ]
+        return random.choice(phrases)
+    
+    def get_comprehensive_kiswahili_lesson(self, topic):
+        """Generate comprehensive Kiswahili learning content"""
+        lessons = {
+            "greetings": self._generate_greetings_lesson(),
+            "basics": self._generate_basics_lesson(),
+            "lion_king": self._generate_lion_king_lesson(),
+            "culture": self._generate_culture_lesson()
+        }
+        return lessons.get(topic, "Hakuna somo kwa hilo mada - No lesson for that topic")
+    
+    def _generate_greetings_lesson(self):
+        """Generate greetings lesson"""
+        return """
+📚 **Kiswahili Greetings Lesson**
+
+**Formal Greetings:**
+- Shikamoo (to elders) → Marahaba (response)
+- Habari za asubuhi? (Good morning?)
+- Habari za mchana? (Good afternoon?)
+
+**Informal Greetings:**
+- Mambo? → Poa (What's up? → Cool)
+- Vipi? (How's it going?)
+- Sema! (Speak!)
+
+**Responses:**
+- Nzuri (Good)
+- Safi (Clean/Cool)
+- Salama (Peaceful)
+
+**Example Conversation:**
+Person A: "Mambo?"
+Person B: "Poa! Na wewe?"
+Person A: "Safi sana!"
+        """
+    
+    def _generate_basics_lesson(self):
+        """Generate basic phrases lesson"""
+        return """
+📚 **Kiswahili Basics Lesson**
+
+**Essential Phrases:**
+- Asante (Thank you)
+- Tafadhali (Please) 
+- Samahani (Excuse me)
+- Karibu (You're welcome)
+
+**Common Responses:**
+- Ndio (Yes)
+- Hapana (No) 
+- Sawa (Okay)
+- Labda (Maybe)
+
+**Key Expressions:**
+- Hakuna matata (No worries)
+- Pole pole (Slowly slowly)
+- Asante sana (Thank you very much)
+        """
+    
+    def _generate_lion_king_lesson(self):
+        """Generate Lion King Kiswahili lesson"""
+        return """
+🎬 **Lion King Kiswahili Lesson**
+
+**Character Names & Meanings:**
+- Simba = Lion 🦁
+- Nala = Gift 🎁
+- Rafiki = Friend 🤝
+- Mufasa = King 👑
+- Pumbaa = Simpleton 🐗
+- Timon = Respect 🙏
+
+**Key Phrases:**
+- Hakuna Matata = No worries
+- Kumbuka wewe ni nani = Remember who you are
+- Mduara wa Uhai = Circle of Life
+
+**Cultural Context:**
+All character names are authentic Swahili words that reflect their personalities and roles in the story!
+        """
+    
+    def _generate_culture_lesson(self):
+        """Generate cultural context lesson"""
+        return """
+🌍 **Swahili Culture Lesson**
+
+**Important Concepts:**
+- Ujamaa = Familyhood/Community
+- Harambee = Pull together
+- Ubuntu = Humanity towards others
+- Pole pole = Take your time
+
+**Traditional Values:**
+- Respect for elders (Shikamoo/Marahaba)
+- Community before individual
+- Hospitality to guests (Karibu!)
+- Indirect communication style
+
+**Famous Proverbs:**
+- "Haraka haraka haina baraka" (Hurry hurry has no blessing)
+- "Mwacha mila ni mtumwa" (He who abandons his culture is a slave)
+        """
+
+
+# Utility functions for easy access
+def get_kiswahili_knowledge():
+    """Quick access to Kiswahili knowledge base"""
+    return KiswahiliKnowledgeBase()
+
+def enhance_with_kiswahili(text, context=None):
+    """Enhance English text with appropriate Kiswahili elements"""
+    kb = KiswahiliKnowledgeBase()
+    return kb.generate_kiswahili_response(text)
+
+def get_lion_king_insights(character=None):
+    """Get Lion King character insights"""
+    kb = KiswahiliKnowledgeBase()
+    if character:
+        return kb.get_lion_king_character(character)
+    return kb.lion_king_lore
+
+
+if __name__ == "__main__":
+    # Test the knowledge base
+    kb = KiswahiliKnowledgeBase()
+    print("🦁 STANLEY AI - Kiswahili Knowledge Base Loaded!")
+    print(f"📚 Contains {len(kb.kiswahili_data)} main categories")
+    print(f"🌍 {len(kb.cultural_context)} cultural sections")
+    print(f"🎬 {len(kb.lion_king_lore)} Lion King knowledge areas")
+    print(f"💬 {len(kb.advanced_phrases)} advanced phrase categories")
+    
+    # Test a sample
+    print("\n" + "="*50)
+    print("Sample Kiswahili Greeting:")
+    print(kb.get_kiswahili_phrase("greetings", "formal", "Shikamoo"))
+    print("\nSample Proverb:")
+    print(kb.get_random_proverb())