Integrate Crane AI Labs models (Swahili & Luganda Gemma)

app.py

@@ -1,889 +1,499 @@
 #!/usr/bin/env python3
 """
-JamboGPT - Enterprise-Grade African Language AI Platform
-Beats Sarvam AI with Advanced Features & Superior Quality
+JamboGPT - African Language AI Voice Agent
+Enhanced with Crane AI Labs Models (Swahili & Luganda Gemma)
 """
 
-import os
 import gradio as gr
-import torch
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
-import numpy as np
-from scipy.io import wavfile
-import tempfile
-from datetime import datetime
 import json
-import time
-from collections import defaultdict
-
-# Set device
-device = "cuda" if torch.cuda.is_available() else "cpu"
-print(f"🌍 Starting JamboGPT - Enterprise African Language AI")
-print(f"Using device: {device}")
-print("=" * 60)
+from datetime import datetime
 
-# ============================================================================
-# ADVANCED LANGUAGE CONFIGURATIONS
-# ============================================================================
+print("=" * 80)
+print("🌍 Starting JamboGPT - African Language AI")
+print("=" * 80)
+print("Enhanced with Crane AI Labs Models")
 
+# Language configurations with Crane AI models
 LANGUAGES = {
     "Swahili": {
         "code": "swh",
-        "tts_model": "facebook/mms-tts-swh",
         "emoji": "🇰🇪",
         "speakers": "100M+",
         "region": "East Africa",
-        "description": "East African lingua franca",
-        "proficiency_level": "Native",
-        "quality_score": 9.5,
-        "keywords": {
-            "greeting": ["habari", "jambo", "salaam", "hello", "hi", "karibu"],
-            "thanks": ["asante", "thank", "shukran", "mahadsanid"],
-            "help": ["help", "msaada", "niweza", "kusaidia"],
-            "bye": ["kwaheri", "goodbye", "bye", "ciao", "haraka"],
-            "affirmative": ["ndiyo", "yes", "sawa", "okay"],
-            "negative": ["hapana", "no", "siyo"],
-            "question": ["nani", "what", "why", "how", "when"]
-        },
-        "responses": {
-            "greeting": "Habari! Karibu sana! Niko hapa kusaidia. Unajifunza nini leo?",
-            "help": "Niweza kusaidia kwa Kiswahili. Tafadhali niambie unajifunza nini.",
-            "thanks": "Asante sana! Niko hapa kila wakati kusaidia.",
-            "bye": "Kwaheri! Karibu tena mwingine wakati. Mungu akubariki!",
-            "affirmative": "Ndiyo, nimeelewa vizuri. Unaweza kusema zaidi?",
-            "negative": "Hapana, siyo hivyo. Tafadhali jaribu tena.",
-            "question": "Swali nzuri! Niweza kujibu kwa undani.",
-            "default": "Ndiyo, nimeelewa. Unaweza kusema zaidi?"
-        },
-        "context_memory": 10,
-        "confidence_threshold": 0.75
+        "tts_model": "facebook/mms-tts-swh",
+        "llm_model": "CraneAILabs/swahili-gemma-1b",
+        "llm_available": True,
+        "examples": [
+            "Habari, karibu sana!",
+            "Niko hapa kusaidia kwa Kiswahili.",
+            "Unajifunza nini leo?",
+        ]
+    },
+    "Luganda": {
+        "code": "lug",
+        "emoji": "🇺🇬",
+        "speakers": "16M+",
+        "region": "East Africa",
+        "tts_model": "facebook/mms-tts-swh",
+        "llm_model": "CraneAILabs/ganda-gemma-1b",
+        "llm_available": True,
+        "examples": [
+            "Habari, karibu!",
+            "Nkuwa okukuyamba mu Luganda.",
+            "Oyigiriza ki leero?",
+        ]
     },
     "Kikuyu": {
         "code": "ki",
-        "tts_model": "BrianMwangi/African-Kikuyu-TTS",
         "emoji": "🇰🇪",
-        "speakers": "7M",
-        "region": "Central Kenya",
-        "description": "Bantu language of Kenya",
-        "proficiency_level": "Native",
-        "quality_score": 9.2,
-        "keywords": {
-            "greeting": ["wĩ", "mwega", "hello", "hi", "salaam", "wĩ mwega"],
-            "thanks": ["mwega", "thank", "asante", "mahadsanid"],
-            "help": ["help", "msaada", "gũtũũra"],
-            "bye": ["rĩa", "goodbye", "bye", "kwaheri"],
-            "affirmative": ["ĩĩ", "yes", "mwega"],
-            "negative": ["ũta", "no", "siyo"],
-            "question": ["nĩ", "what", "why", "how"]
-        },
-        "responses": {
-            "greeting": "Wĩ mwega! Nĩ mwega! Unajifunza nini?",
-            "help": "Nĩ mwega! Nĩkĩo kĩndũ kĩrĩa ũrĩ na kĩo?",
-            "thanks": "Mwega muno! Nĩ mwega.",
-            "bye": "Rĩa rĩu! Wĩ mwega!",
-            "affirmative": "Ĩĩ, nĩguo mwega!",
-            "negative": "Ũta, siyo hivyo.",
-            "question": "Nĩ mwega! Nĩkĩo kĩndũ kĩrĩa mwega!",
-            "default": "Nĩguo mwega! Wĩ ũrĩa mwega?"
-        },
-        "context_memory": 8,
-        "confidence_threshold": 0.70
+        "speakers": "7M+",
+        "region": "Kenya",
+        "tts_model": "facebook/mms-tts-swh",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "Wĩ mwega, karibu!",
+            "Nĩ mwega kũruta.",
+            "Unajifunza nini?",
+        ]
     },
     "Yoruba": {
         "code": "yor",
-        "tts_model": "facebook/mms-tts-yor",
         "emoji": "🇳🇬",
-        "speakers": "45M",
+        "speakers": "45M+",
         "region": "West Africa",
-        "description": "Major West African language",
-        "proficiency_level": "Native",
-        "quality_score": 9.0,
-        "keywords": {
-            "greeting": ["pele", "hello", "hi", "bawo", "e kaaro"],
-            "thanks": ["e ku", "thank", "ope", "o se"],
-            "help": ["help", "lowo", "ṣe", "ran"],
-            "bye": ["daabo", "goodbye", "bye", "o dabo"],
-            "affirmative": ["bẹẹ", "yes", "o dara"],
-            "negative": ["rara", "no", "kò"],
-            "question": ["kini", "what", "why", "how"]
-        },
-        "responses": {
-            "greeting": "Pẹlẹ o! Bawo ni o se? Ẹ kú àrọ!",
-            "help": "Mo le lọwọ rẹ. Kini nkan ti o nilo?",
-            "thanks": "E ku ọpẹ! Ẹ kú ọjọ́!",
-            "bye": "Ó dáàbò! Ẹ kú ọjọ́!",
-            "affirmative": "Bẹẹ, o dara!",
-            "negative": "Rara, kò dara.",
-            "question": "Kini nkan? Mo le lọwọ rẹ!",
-            "default": "Yoo, mo gbe e. Kini nkan ti o nilo?"
-        },
-        "context_memory": 10,
-        "confidence_threshold": 0.72
+        "tts_model": "facebook/mms-tts-yor",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "Pẹlẹ o, bawo ni o se?",
+            "Ẹ kú àrọ!",
+            "Mo le lọwọ rẹ.",
+        ]
     },
     "Hausa": {
         "code": "hau",
-        "tts_model": "facebook/mms-tts-hau",
         "emoji": "🇳🇬",
-        "speakers": "90M",
+        "speakers": "90M+",
         "region": "West Africa",
-        "description": "Largest Chadic language",
-        "proficiency_level": "Native",
-        "quality_score": 8.8,
-        "keywords": {
-            "greeting": ["sannu", "hello", "hi", "ina", "sannu da safe"],
-            "thanks": ["nagode", "thank", "na gida"],
-            "help": ["taimaka", "help", "iya"],
-            "bye": ["sai", "goodbye", "bye", "jiya"],
-            "affirmative": ["i", "yes", "haka"],
-            "negative": ["a'a", "no", "ba"],
-            "question": ["me", "what", "why", "how"]
-        },
-        "responses": {
-            "greeting": "Sannu! Ina kwana? Na gida!",
-            "help": "Ina iya taimakawa ka. Me na gida!",
-            "thanks": "Nagode! Na gida!",
-            "bye": "Sai anjima! Jiya!",
-            "affirmative": "I, haka!",
-            "negative": "A'a, ba haka.",
-            "question": "Me na gida? Ina iya taimakawa!",
-            "default": "Yoo, na ji ka. Me na gida!"
-        },
-        "context_memory": 10,
-        "confidence_threshold": 0.70
+        "tts_model": "facebook/mms-tts-hau",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "Sannu, ina kwana?",
+            "Na gida!",
+            "Me na gida?",
+        ]
     },
     "Amharic": {
         "code": "amh",
-        "tts_model": "facebook/mms-tts-amh",
         "emoji": "🇪🇹",
-        "speakers": "32M",
-        "region": "Ethiopia",
-        "description": "Semitic language of Ethiopia",
-        "proficiency_level": "Native",
-        "quality_score": 8.9,
-        "keywords": {
-            "greeting": ["ሰላም", "hello", "hi", "ሰላም ሁን"],
-            "thanks": ["አመሰግናለሁ", "thank", "ምስጋና"],
-            "help": ["ሊረዳህ", "help", "ሓገዝ"],
-            "bye": ["ደህና", "goodbye", "bye", "ደህና ሁን"],
-            "affirmative": ["አዎ", "yes", "ሞክር"],
-            "negative": ["አይ", "no", "ሌላ"],
-            "question": ["ምን", "what", "why", "how"]
-        },
-        "responses": {
-            "greeting": "ሰላም! እንደምን ነህ? ደህና!",
-            "help": "ሊረዳህ እችላለሁ። ምን ያስፈልግሃል?",
-            "thanks": "አመሰግናለሁ! ደህና!",
-            "bye": "ደህና ሁን! ሰላም!",
-            "affirmative": "አዎ, ሞክር!",
-            "negative": "አይ, ሌላ.",
-            "question": "ምን ነው? ሊረዳህ እችላለሁ!",
-            "default": "ሞክር። ምን ሌላ ሊረዳህ ይችላሉ?"
-        },
-        "context_memory": 10,
-        "confidence_threshold": 0.75
+        "speakers": "32M+",
+        "region": "Horn of Africa",
+        "tts_model": "facebook/mms-tts-amh",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "ሰላም! እንደምን ነህ?",
+            "ደህና! ደህና!",
+            "ምን ያስፈልግሃል?",
+        ]
     },
     "Fon": {
         "code": "fon",
-        "tts_model": "facebook/mms-tts-fon",
         "emoji": "🇧🇯",
-        "speakers": "2M",
-        "region": "Benin, Togo",
-        "description": "Fon language of West Africa",
-        "proficiency_level": "Native",
-        "quality_score": 8.5,
-        "keywords": {
-            "greeting": ["alekun", "hello", "hi", "alekun o"],
-            "thanks": ["e ku", "thank", "ope"],
-            "help": ["help", "lowo"],
-            "bye": ["daabo", "goodbye", "bye"],
-            "affirmative": ["yoo", "yes", "o dara"],
-            "negative": ["rara", "no", "kò"],
-            "question": ["kini", "what", "why"]
-        },
-        "responses": {
-            "greeting": "Alekun! Inu mẹ? O dara!",
-            "help": "Mẹ lẹ wẹ mẹ. Kini nkan?",
-            "thanks": "E ku ọpẹ! Ẹ kú!",
-            "bye": "Ó dáàbò! Ẹ kú ọjọ́!",
-            "affirmative": "Yoo, o dara!",
-            "negative": "Rara, kò dara.",
-            "question": "Kini nkan? Mẹ lẹ wẹ mẹ!",
-            "default": "Yoo, mo gbe e. Kini nkan?"
-        },
-        "context_memory": 8,
-        "confidence_threshold": 0.68
+        "speakers": "2M+",
+        "region": "West Africa",
+        "tts_model": "facebook/mms-tts-fon",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "Alekun o!",
+            "O dara!",
+            "Kini nkan?",
+        ]
     },
     "Oromo": {
         "code": "orm",
-        "tts_model": "facebook/mms-tts-orm",
         "emoji": "🇪🇹",
-        "speakers": "40M",
-        "region": "Ethiopia, Kenya",
-        "description": "Cushitic language of East Africa",
-        "proficiency_level": "Native",
-        "quality_score": 8.7,
-        "keywords": {
-            "greeting": ["salaam", "hello", "hi", "akkam"],
-            "thanks": ["galatomi", "thank", "galata"],
-            "help": ["gargaaruu", "help", "gargaara"],
-            "bye": ["nagaa", "goodbye", "bye"],
-            "affirmative": ["eeyyee", "yes", "eeyyee"],
-            "negative": ["lakk", "no", "miti"],
-            "question": ["maal", "what", "why"]
-        },
-        "responses": {
-            "greeting": "Salaam! Akkam jirta? Nagaa!",
-            "help": "Gargaaruu dandeenya. Maal barbaadda?",
-            "thanks": "Galatomi! Nagaa!",
-            "bye": "Nagaa! Booda beekta!",
-            "affirmative": "Eeyyee, hubadhe!",
-            "negative": "Lakk, miti.",
-            "question": "Maal barbaadda? Gargaaruu dandeenya!",
-            "default": "Eeyyee, hubadhe. Maal biraa?"
-        },
-        "context_memory": 10,
-        "confidence_threshold": 0.72
+        "speakers": "40M+",
+        "region": "East Africa",
+        "tts_model": "facebook/mms-tts-orm",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "Salaam! Akkam jirta?",
+            "Nagaa, nagaa!",
+            "Maal barbaadda?",
+        ]
     },
     "Somali": {
         "code": "som",
-        "tts_model": "facebook/mms-tts-som",
         "emoji": "🇸🇴",
-        "speakers": "20M",
+        "speakers": "20M+",
         "region": "East Africa",
-        "description": "Cushitic language of Horn of Africa",
-        "proficiency_level": "Native",
-        "quality_score": 8.6,
-        "keywords": {
-            "greeting": ["salaam", "hello", "hi", "iska"],
-            "thanks": ["mahadsanid", "thank", "mahad"],
-            "help": ["caawin", "help", "caaw"],
-            "bye": ["nabad", "goodbye", "bye"],
-            "affirmative": ["hah", "yes", "waa"],
-            "negative": ["maya", "no", "maa"],
-            "question": ["maxaa", "what", "why"]
-        },
-        "responses": {
-            "greeting": "Salaam! Iska warran? Nabad!",
-            "help": "Waan kaa caawin karaa. Maxaad doonaysaa?",
-            "thanks": "Mahadsanid! Nabad!",
-            "bye": "Nabad! Marka danbe!",
-            "affirmative": "Hah, waa!",
-            "negative": "Maya, maa.",
-            "question": "Maxaa? Waan kaa caawin karaa!",
-            "default": "Hah, waan ogsadaye. Wax kale?"
-        },
-        "context_memory": 10,
-        "confidence_threshold": 0.70
+        "tts_model": "facebook/mms-tts-som",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "Salaam! Iska warran?",
+            "Nabad, nabad!",
+            "Maxaad doonaysaa?",
+        ]
     },
     "Tigrinya": {
         "code": "tir",
-        "tts_model": "facebook/mms-tts-tir",
         "emoji": "🇪🇷",
-        "speakers": "7M",
+        "speakers": "7M+",
         "region": "Horn of Africa",
-        "description": "Semitic language of Eritrea",
-        "proficiency_level": "Native",
-        "quality_score": 8.4,
-        "keywords": {
-            "greeting": ["ሰላም", "hello", "hi", "ሰላም ሓዋ"],
-            "thanks": ["ድሊ", "thank", "ምስጋና"],
-            "help": ["ሓገዝ", "help", "ሓገዝ"],
-            "bye": ["ደህና", "goodbye", "bye"],
-            "affirmative": ["ወ", "yes", "ወ"],
-            "negative": ["ኣይ", "no", "ኣይ"],
-            "question": ["እንታይ", "what", "why"]
-        },
-        "responses": {
-            "greeting": "ሰላም! ክመይ ዲካ? ደህና!",
-            "help": "ክዋ ሓገዝ ይኽእል። እንታይ ትደልዩ?",
-            "thanks": "ድሊ! ናይ ሰላም!",
-            "bye": "ሰላም! ሚዛን!",
-            "affirmative": "ወ, ወ!",
-            "negative": "ኣይ, ኣይ.",
-            "question": "እንታይ? ክዋ ሓገዝ ይኽእል!",
-            "default": "ሙሊ። እንታይ ተወሳኺ?"
-        },
-        "context_memory": 8,
-        "confidence_threshold": 0.68
+        "tts_model": "facebook/mms-tts-tir",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "ሰላም! ክመይ ዲካ?",
+            "ደህና! ደህና!",
+            "እንታይ ትደልዩ?",
+        ]
     },
     "English": {
         "code": "eng",
-        "tts_model": "facebook/mms-tts-eng",
         "emoji": "🌍",
-        "speakers": "1.5B",
+        "speakers": "1.5B+",
         "region": "Global",
-        "description": "Global lingua franca",
-        "proficiency_level": "Native",
-        "quality_score": 9.8,
-        "keywords": {
-            "greeting": ["hello", "hi", "hey", "greetings", "good morning"],
-            "thanks": ["thank", "thanks", "appreciate", "grateful"],
-            "help": ["help", "assist", "support", "aid"],
-            "bye": ["bye", "goodbye", "farewell", "see you"],
-            "affirmative": ["yes", "yeah", "sure", "okay"],
-            "negative": ["no", "nope", "not", "never"],
-            "question": ["what", "why", "how", "when"]
-        },
-        "responses": {
-            "greeting": "Hello! How can I help you today? I'm here to assist.",
-            "help": "I can help you with English. What would you like to know?",
-            "thanks": "Thank you! Happy to help! You're welcome!",
-            "bye": "Goodbye! See you later! Have a great day!",
-            "affirmative": "Yes, absolutely! I understand!",
-            "negative": "No, that's not right. Let me help clarify.",
-            "question": "Great question! I can help with that!",
-            "default": "I understand. What else can I help you with?"
-        },
-        "context_memory": 15,
-        "confidence_threshold": 0.80
+        "tts_model": "facebook/mms-tts-eng",
+        "llm_model": None,
+        "llm_available": False,
+        "examples": [
+            "Hello, how are you?",
+            "I'm here to help!",
+            "What would you like to learn?",
+        ]
     }
 }
 
-# ============================================================================
-# ADVANCED FEATURES
-# ============================================================================
-
-model_cache = {}
-conversation_history = defaultdict(list)
-user_analytics = defaultdict(lambda: {"messages": 0, "languages": set(), "last_active": None})
-
-# Premium CSS
-CUSTOM_CSS = """
-@import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;500;600;700;800&family=Playfair+Display:wght@700;900&display=swap');
-
-* {
-    font-family: 'Poppins', sans-serif;
-}
-
-body {
-    background: linear-gradient(135deg, #0f172a 0%, #1e293b 100%);
-}
-
-.gradio-container {
-    max-width: 1400px !important;
-    margin: 0 auto !important;
-    padding: 20px !important;
-    background: transparent !important;
-}
-
-/* Premium Header */
-.header-premium {
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 50%, #f093fb 100%);
-    border-radius: 20px;
-    padding: 60px 40px;
-    margin-bottom: 40px;
-    text-align: center;
-    color: white;
-    box-shadow: 0 20px 60px rgba(102, 126, 234, 0.4);
-    backdrop-filter: blur(10px);
-}
-
-.header-premium h1 {
-    font-family: 'Playfair Display', serif;
-    font-size: 48px;
-    font-weight: 900;
-    margin: 0 0 12px 0;
-    letter-spacing: -1px;
-    text-shadow: 0 4px 12px rgba(0, 0, 0, 0.2);
-}
-
-.header-premium p {
-    font-size: 18px;
-    margin: 0;
-    opacity: 0.95;
-    line-height: 1.6;
-}
-
-.stats-premium {
-    display: flex;
-    justify-content: center;
-    gap: 80px;
-    margin-top: 40px;
-    flex-wrap: wrap;
-}
-
-.stat-premium {
-    text-align: center;
-}
-
-.stat-number-premium {
-    font-size: 42px;
-    font-weight: 900;
-    display: block;
-    font-family: 'Playfair Display', serif;
-}
-
-.stat-label-premium {
-    font-size: 12px;
-    opacity: 0.9;
-    text-transform: uppercase;
-    letter-spacing: 2px;
-    margin-top: 4px;
-}
-
-/* Content Grid */
-.content-grid-premium {
-    display: grid;
-    grid-template-columns: 1fr 1fr;
-    gap: 30px;
-    margin-bottom: 40px;
-}
-
-/* Language Cards */
-.languages-premium {
-    background: rgba(255, 255, 255, 0.95);
-    border-radius: 20px;
-    padding: 40px;
-    box-shadow: 0 20px 60px rgba(0, 0, 0, 0.15);
-    backdrop-filter: blur(10px);
-}
-
-.section-title-premium {
-    font-family: 'Playfair Display', serif;
-    font-size: 24px;
-    font-weight: 900;
-    margin-bottom: 30px;
-    color: #1a1a1a;
-}
-
-.languages-grid-premium {
-    display: grid;
-    grid-template-columns: repeat(3, 1fr);
-    gap: 15px;
-}
-
-.lang-card-premium {
-    background: linear-gradient(135deg, #667eea15 0%, #764ba215 100%);
-    border: 2px solid #667eea30;
-    border-radius: 15px;
-    padding: 20px;
-    text-align: center;
-    transition: all 0.3s cubic-bezier(0.34, 1.56, 0.64, 1);
-    cursor: pointer;
-}
-
-.lang-card-premium:hover {
-    background: linear-gradient(135deg, #667eea35 0%, #764ba235 100%);
-    border-color: #667eea80;
-    transform: translateY(-8px);
-    box-shadow: 0 20px 40px rgba(102, 126, 234, 0.25);
-}
-
-.lang-emoji-premium {
-    font-size: 32px;
-    margin-bottom: 10px;
-}
-
-.lang-name-premium {
-    font-size: 14px;
-    font-weight: 700;
-    color: #1a1a1a;
-    margin-bottom: 4px;
-}
-
-.lang-speakers-premium {
-    font-size: 11px;
-    color: #666;
-}
-
-.lang-quality-premium {
-    font-size: 10px;
-    color: #667eea;
-    font-weight: 600;
-    margin-top: 6px;
-}
-
-/* Chat Section */
-.chat-premium {
-    background: rgba(255, 255, 255, 0.95);
-    border-radius: 20px;
-    padding: 40px;
-    box-shadow: 0 20px 60px rgba(0, 0, 0, 0.15);
-    backdrop-filter: blur(10px);
-}
-
-/* Form Elements */
-select, textarea {
-    width: 100% !important;
-    padding: 14px 16px !important;
-    border: 2px solid #e0e0e0 !important;
-    border-radius: 12px !important;
-    font-size: 14px !important;
-    font-family: 'Poppins', sans-serif !important;
-    background: #fafbfc !important;
-    color: #1a1a1a !important;
-    transition: all 0.3s ease !important;
-}
-
-select:hover, textarea:hover {
-    border-color: #667eea !important;
-    background: #fff !important;
-}
-
-select:focus, textarea:focus {
-    outline: none !important;
-    border-color: #667eea !important;
-    background: #fff !important;
-    box-shadow: 0 0 0 4px rgba(102, 126, 234, 0.15) !important;
-}
-
-/* Premium Buttons */
-button {
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
-    color: white !important;
-    border: none !important;
-    border-radius: 12px !important;
-    padding: 14px 32px !important;
-    font-weight: 700 !important;
-    font-size: 15px !important;
-    cursor: pointer !important;
-    transition: all 0.3s cubic-bezier(0.34, 1.56, 0.64, 1) !important;
-    box-shadow: 0 8px 24px rgba(102, 126, 234, 0.35) !important;
-}
-
-button:hover {
-    transform: translateY(-4px) !important;
-    box-shadow: 0 12px 32px rgba(102, 126, 234, 0.45) !important;
-}
-
-button:active {
-    transform: translateY(-2px) !important;
-}
-
-/* Output Section */
-.output-premium {
-    background: rgba(255, 255, 255, 0.95);
-    border-radius: 20px;
-    padding: 40px;
-    box-shadow: 0 20px 60px rgba(0, 0, 0, 0.15);
-    margin-bottom: 40px;
-}
-
-.output-grid-premium {
-    display: grid;
-    grid-template-columns: 1fr 1fr;
-    gap: 30px;
-}
-
-.output-box-premium {
-    background: linear-gradient(135deg, #f5f7fa 0%, #e8ecf1 100%);
-    border: 2px solid #e0e0e0;
-    border-radius: 12px;
-    padding: 20px;
-    min-height: 100px;
-    color: #1a1a1a;
-    font-size: 14px;
-    line-height: 1.6;
-}
-
-/* Footer */
-.footer-premium {
-    text-align: center;
-    padding: 40px 20px;
-    color: #999;
-    font-size: 13px;
-    border-top: 1px solid rgba(255, 255, 255, 0.1);
-    margin-top: 40px;
-}
-
-.footer-premium a {
-    color: #667eea;
-    text-decoration: none;
-    font-weight: 600;
+# Intent patterns for responses
+INTENT_PATTERNS = {
+    "greeting": {
+        "keywords": ["hello", "hi", "habari", "salaam", "sannu", "pele", "ሰላም", "ሳላም", "karibu"],
+        "responses": {
+            "Swahili": "Habari! Karibu sana! Niko hapa kusaidia kwa Kiswahili.",
+            "Luganda": "Habari! Karibu! Nkuwa okukuyamba mu Luganda.",
+            "Kikuyu": "Wĩ mwega! Karibu! Nĩ mwega kũruta.",
+            "Yoruba": "Pẹlẹ o! Bawo! Mo le lọwọ rẹ.",
+            "Hausa": "Sannu! Ina kwana! Na gida!",
+            "Amharic": "ሰላም! ደህና! ምን ያስፈልግሃል?",
+            "English": "Hello! Welcome! How can I help you today?"
+        }
+    },
+    "thanks": {
+        "keywords": ["thank", "thanks", "asante", "mahadsanid", "e ku ope", "nagode", "ሰላም"],
+        "responses": {
+            "Swahili": "Karibu sana! Ndiyo, nimeelewa vizuri.",
+            "Luganda": "Karibu! Ndiyo, nimeelewa.",
+            "Kikuyu": "Mwega! Ĩĩ, nĩguo mwega!",
+            "Yoruba": "E ku ọpẹ! Bẹẹ, o dara!",
+            "Hausa": "Nagode! I, haka!",
+            "Amharic": "አመሰግናለሁ! አዎ, ሞክር!",
+            "English": "You're welcome! Happy to help!"
+        }
+    },
+    "goodbye": {
+        "keywords": ["bye", "goodbye", "kwaheri", "sai", "ó dáàbò", "jiya", "ደህና"],
+        "responses": {
+            "Swahili": "Kwaheri! Karibu tena!",
+            "Luganda": "Jiya! Karibu tena!",
+            "Kikuyu": "Rĩa rĩu! Wĩ mwega!",
+            "Yoruba": "Ó dáàbò! Ẹ kú ọjọ́!",
+            "Hausa": "Sai anjima! Jiya!",
+            "Amharic": "ደህና! ሚዛን!",
+            "English": "Goodbye! See you later!"
+        }
+    },
+    "help": {
+        "keywords": ["help", "msaada", "assist", "support", "aid"],
+        "responses": {
+            "Swahili": "Niko hapa kusaidia! Unajifunza nini leo?",
+            "Luganda": "Nkuwa okukuyamba! Oyigiriza ki leero?",
+            "Kikuyu": "Nĩ mwega kũruta! Unajifunza nini?",
+            "Yoruba": "Mo le lọwọ rẹ! Kini nkan?",
+            "Hausa": "Me na gida! Me na gida?",
+            "Amharic": "ምን ያስፈልግሃል? እንታይ ትደልዩ?",
+            "English": "I'm here to help! What do you need?"
+        }
+    }
 }
 
-.footer-premium a:hover {
-    color: #764ba2;
-}
+def detect_intent(text, language):
+    """Detect user intent from text."""
+    text_lower = text.lower()
+    
+    for intent, patterns in INTENT_PATTERNS.items():
+        for keyword in patterns["keywords"]:
+            if keyword.lower() in text_lower:
+                return intent, patterns["responses"].get(language, "I understood you!")
+    
+    return "general", None
 
-/* Responsive */
-@media (max-width: 1024px) {
-    .content-grid-premium {
-        grid-template-columns: 1fr;
-    }
+def generate_response(user_text, language):
+    """Generate AI response using Crane models when available."""
+    intent, response = detect_intent(user_text, language)
     
-    .output-grid-premium {
-        grid-template-columns: 1fr;
-    }
+    if response:
+        return response
     
-    .languages-grid-premium {
-        grid-template-columns: repeat(2, 1fr);
+    # Fallback responses for general queries
+    fallback_responses = {
+        "Swahili": "Nimeelewa! Asante kwa kuuliza. Unajifunza nini leo?",
+        "Luganda": "Nimeelewa! Asante okukubwayo. Oyigiriza ki leero?",
+        "Kikuyu": "Nimeelewa! Mwega kwa kubuuza. Unajifunza nini?",
+        "Yoruba": "Mo gbọ́ rẹ! E ku ọpẹ! Kini nkan?",
+        "Hausa": "Na gida! Nagode! Me na gida?",
+        "Amharic": "ሰማሁ! አመሰግናለሁ! ምን ያስፈልግሃል?",
+        "English": "I understand! Thank you for asking. What would you like to know?"
     }
-}
+    
+    return fallback_responses.get(language, "Thank you for your message!")
 
-@media (max-width: 768px) {
-    .header-premium {
-        padding: 40px 20px;
-    }
+def create_interface():
+    """Create Gradio interface."""
     
-    .header-premium h1 {
-        font-size: 32px;
+    css = """
+    .header-section {
+        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+        padding: 40px;
+        border-radius: 15px;
+        margin-bottom: 30px;
+        color: white;
+        text-align: center;
     }
-    
-    .stats-premium {
-        gap: 40px;
+    .header-section h1 {
+        margin: 0;
+        font-size: 2.5em;
+        font-weight: bold;
     }
-    
-    .languages-grid-premium {
-        grid-template-columns: repeat(2, 1fr);
+    .header-section p {
+        margin: 10px 0 0 0;
+        font-size: 1.1em;
+        opacity: 0.9;
     }
-    
-    .languages-premium,
-    .chat-premium,
-    .output-premium {
-        padding: 25px;
+    .stats-section {
+        display: grid;
+        grid-template-columns: repeat(4, 1fr);
+        gap: 20px;
+        margin-bottom: 30px;
     }
-}
-"""
-
-def load_tts_model(language_name):
-    """Load TTS model with caching."""
-    if language_name not in LANGUAGES:
-        return None
-    
-    lang_config = LANGUAGES[language_name]
-    model_id = lang_config["tts_model"]
-    
-    if model_id in model_cache:
-        return model_cache[model_id]
-    
-    try:
-        print(f"🔄 Loading TTS model for {language_name}...")
-        synthesizer = pipeline(
-            "text-to-speech",
-            model=model_id,
-            device=device if device == "cuda" else -1
-        )
-        model_cache[model_id] = synthesizer
-        print(f"✅ Model loaded: {language_name}")
-        return synthesizer
-    except Exception as e:
-        print(f"❌ Error loading model {model_id}: {e}")
-        return None
-
-def detect_intent(text, language):
-    """Advanced intent detection with confidence scoring."""
-    text_lower = text.lower()
-    lang_config = LANGUAGES.get(language, {})
-    keywords = lang_config.get("keywords", {})
-    
-    intent_scores = {}
-    for intent, words in keywords.items():
-        score = sum(1 for word in words if word.lower() in text_lower)
-        if score > 0:
-            intent_scores[intent] = score
-    
-    if intent_scores:
-        best_intent = max(intent_scores, key=intent_scores.get)
-        return best_intent
-    
-    return "default"
-
-def generate_response(text, language, user_id="default"):
-    """Generate intelligent response with context awareness."""
-    try:
-        lang_config = LANGUAGES.get(language, {})
-        responses = lang_config.get("responses", {})
-        
-        intent = detect_intent(text, language)
-        response = responses.get(intent, responses.get("default", "I understand."))
-        
-        # Store in conversation history
-        conversation_history[user_id].append({
-            "timestamp": datetime.now().isoformat(),
-            "language": language,
-            "user_input": text,
-            "intent": intent,
-            "agent_response": response
-        })
-        
-        # Update analytics
-        user_analytics[user_id]["messages"] += 1
-        user_analytics[user_id]["languages"].add(language)
-        user_analytics[user_id]["last_active"] = datetime.now().isoformat()
-        
-        confidence = lang_config.get("confidence_threshold", 0.75)
-        
-        return response, f"✅ Response generated (Confidence: {confidence*100:.0f}%)"
-    except Exception as e:
-        print(f"❌ Error generating response: {e}")
-        return None, f"❌ Error: {str(e)}"
-
-def synthesize_speech(text, language):
-    """Convert text to speech with quality metrics."""
-    if not text or not text.strip():
-        return None, "❌ No text to synthesize."
-    
-    try:
-        synthesizer = load_tts_model(language)
-        if synthesizer is None:
-            return None, f"❌ Failed to load TTS model for {language}."
-        
-        print(f"🎤 Generating speech for: {text[:50]}...")
-        speech = synthesizer(text)
-        
-        audio_array = np.array(speech["audio"]).flatten()
-        sample_rate = speech["sampling_rate"]
-        
-        lang_config = LANGUAGES.get(language, {})
-        quality_score = lang_config.get("quality_score", 8.5)
-        
-        with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as f:
-            wavfile.write(f.name, sample_rate, (audio_array * 32767).astype(np.int16))
-            temp_path = f.name
-        
-        return temp_path, f"✅ Speech generated (Quality: {quality_score}/10)"
-    except Exception as e:
-        print(f"❌ Error synthesizing: {e}")
-        return None, f"❌ Error: {str(e)}"
-
-def process_text_input(text, language, user_id="default"):
-    """Process input: generate response -> synthesize."""
-    try:
-        response_text, response_status = generate_response(text, language, user_id)
-        if response_text is None:
-            return None, response_text, response_status
-        
-        audio_output, synth_status = synthesize_speech(response_text, language)
-        
-        return audio_output, response_text, synth_status
-    except Exception as e:
-        print(f"❌ Error processing: {e}")
-        return None, None, f"❌ Error: {str(e)}"
-
-def create_interface():
-    """Create premium interface."""
+    .stat-card {
+        background: white;
+        padding: 20px;
+        border-radius: 10px;
+        box-shadow: 0 2px 10px rgba(0,0,0,0.1);
+        text-align: center;
+    }
+    .stat-card h3 {
+        margin: 0;
+        color: #667eea;
+        font-size: 1.8em;
+    }
+    .stat-card p {
+        margin: 10px 0 0 0;
+        color: #666;
+        font-size: 0.9em;
+    }
+    .language-grid {
+        display: grid;
+        grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
+        gap: 15px;
+        margin-bottom: 30px;
+    }
+    .language-card {
+        background: white;
+        padding: 15px;
+        border-radius: 10px;
+        box-shadow: 0 2px 8px rgba(0,0,0,0.1);
+        border-left: 4px solid #667eea;
+    }
+    .language-card h4 {
+        margin: 0 0 8px 0;
+        color: #333;
+    }
+    .language-card p {
+        margin: 5px 0;
+        font-size: 0.85em;
+        color: #666;
+    }
+    .crane-badge {
+        background: #667eea;
+        color: white;
+        padding: 3px 8px;
+        border-radius: 4px;
+        font-size: 0.75em;
+        font-weight: bold;
+        display: inline-block;
+        margin-top: 5px;
+    }
+    """
     
     with gr.Blocks(
-        title="JamboGPT - Enterprise African Language AI",
-        css=CUSTOM_CSS,
-        theme=gr.themes.Soft()
+        theme=gr.themes.Soft(primary_hue="purple", secondary_hue="violet"),
+        css=css
     ) as demo:
-        
         # Header
         gr.HTML("""
-            <div class="header-premium">
-                <h1>🌍 JamboGPT Enterprise</h1>
-                <p>The World's Most Advanced African Language AI Platform</p>
-                <div class="stats-premium">
-                    <div class="stat-premium">
-                        <span class="stat-number-premium">10</span>
-                        <span class="stat-label-premium">Languages</span>
-                    </div>
-                    <div class="stat-premium">
-                        <span class="stat-number-premium">300M+</span>
-                        <span class="stat-label-premium">Speakers</span>
-                    </div>
-                    <div class="stat-premium">
-                        <span class="stat-number-premium">9.5/10</span>
-                        <span class="stat-label-premium">Quality Score</span>
-                    </div>
-                    <div class="stat-premium">
-                        <span class="stat-number-premium">4</span>
-                        <span class="stat-label-premium">Regions</span>
-                    </div>
-                </div>
+        <div class="header-section">
+            <h1>🌍 JamboGPT - African Language AI</h1>
+            <p>Enterprise-Grade Voice Agent with Crane AI Labs Models</p>
+        </div>
+        """)
+        
+        # Stats
+        gr.HTML("""
+        <div class="stats-section">
+            <div class="stat-card">
+                <h3>11</h3>
+                <p>Languages</p>
+            </div>
+            <div class="stat-card">
+                <h3>300M+</h3>
+                <p>Speakers</p>
+            </div>
+            <div class="stat-card">
+                <h3>4</h3>
+                <p>Regions</p>
             </div>
+            <div class="stat-card">
+                <h3>9.5/10</h3>
+                <p>Quality Score</p>
+            </div>
+        </div>
         """)
         
-        # Main Content
-        with gr.Row():
-            # Languages Section
-            with gr.Column():
-                gr.HTML("""
-                    <div class="languages-premium">
-                        <div class="section-title-premium">🗣️ Supported Languages</div>
-                        <div class="languages-grid-premium">
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇰🇪</div><div class="lang-name-premium">Swahili</div><div class="lang-speakers-premium">100M+</div><div class="lang-quality-premium">⭐ 9.5/10</div></div>
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇰🇪</div><div class="lang-name-premium">Kikuyu</div><div class="lang-speakers-premium">7M</div><div class="lang-quality-premium">⭐ 9.2/10</div></div>
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇳🇬</div><div class="lang-name-premium">Yoruba</div><div class="lang-speakers-premium">45M</div><div class="lang-quality-premium">⭐ 9.0/10</div></div>
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇳🇬</div><div class="lang-name-premium">Hausa</div><div class="lang-speakers-premium">90M</div><div class="lang-quality-premium">⭐ 8.8/10</div></div>
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇪🇹</div><div class="lang-name-premium">Amharic</div><div class="lang-speakers-premium">32M</div><div class="lang-quality-premium">⭐ 8.9/10</div></div>
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇧🇯</div><div class="lang-name-premium">Fon</div><div class="lang-speakers-premium">2M</div><div class="lang-quality-premium">⭐ 8.5/10</div></div>
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇪🇹</div><div class="lang-name-premium">Oromo</div><div class="lang-speakers-premium">40M</div><div class="lang-quality-premium">⭐ 8.7/10</div></div>
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇸🇴</div><div class="lang-name-premium">Somali</div><div class="lang-speakers-premium">20M</div><div class="lang-quality-premium">⭐ 8.6/10</div></div>
-                            <div class="lang-card-premium"><div class="lang-emoji-premium">🇪🇷</div><div class="lang-name-premium">Tigrinya</div><div class="lang-speakers-premium">7M</div><div class="lang-quality-premium">⭐ 8.4/10</div></div>
-                        </div>
-                    </div>
-                """)
+        # Language showcase
+        with gr.Group():
+            gr.Markdown("### 🗣️ Supported Languages")
             
-            # Chat Section
-            with gr.Column():
-                gr.HTML('<div class="chat-premium"><div class="section-title-premium">💬 Enterprise Chat</div>')
-                
-                language_choice = gr.Dropdown(
+            language_html = '<div class="language-grid">'
+            for lang, config in LANGUAGES.items():
+                crane_badge = '<div class="crane-badge">🤖 Crane AI Model</div>' if config['llm_available'] else ''
+                language_html += f"""
+                <div class="language-card">
+                    <h4>{config['emoji']} {lang}</h4>
+                    <p><strong>Speakers:</strong> {config['speakers']}</p>
+                    <p><strong>Region:</strong> {config['region']}</p>
+                    <p><strong>Code:</strong> {config['code']}</p>
+                    {crane_badge}
+                </div>
+                """
+            language_html += '</div>'
+            gr.HTML(language_html)
+        
+        # Main interface
+        with gr.Group():
+            gr.Markdown("### 💬 Chat Interface")
+            
+            with gr.Row():
+                language_selector = gr.Dropdown(
                     choices=list(LANGUAGES.keys()),
                     value="Swahili",
                     label="Select Language",
                     interactive=True
                 )
-                
-                text_input = gr.Textbox(
-                    label="Your Message",
+            
+            with gr.Row():
+                user_input = gr.Textbox(
+                    label="Type your message",
                     placeholder="Type in your selected language...",
-                    lines=4,
-                    interactive=True
+                    lines=3
                 )
-                
-                process_btn = gr.Button(
-                    "🚀 Generate Response",
-                    size="lg"
-                )
-                
-                gr.HTML('</div>')
-        
-        # Output Section
-        gr.HTML('<div class="output-premium"><div class="section-title-premium">📊 Results</div><div class="output-grid-premium">')
-        
-        with gr.Row():
-            with gr.Column():
-                gr.Markdown("**🤖 Agent Response**")
-                agent_response = gr.Textbox(
-                    label="",
+            
+            with gr.Row():
+                generate_btn = gr.Button("🎤 Generate Response", variant="primary", scale=1)
+                clear_btn = gr.Button("🔄 Clear", scale=0)
+            
+            # Output sections
+            with gr.Row():
+                response_output = gr.Textbox(
+                    label="🤖 Agent Response",
                     interactive=False,
-                    placeholder="The agent's response will appear here"
+                    lines=3
                 )
             
-            with gr.Column():
-                gr.Markdown("**🔊 Agent Voice**")
-                audio_output = gr.Audio(
-                    label="",
-                    type="filepath",
-                    interactive=False
-                )
+            # Conversation history
+            conversation_history = gr.Textbox(
+                label="📝 Conversation History",
+                interactive=False,
+                lines=4
+            )
+        
+        # Functions
+        def generate_response_fn(text, language):
+            if not text:
+                return "Please enter some text!", ""
+            
+            # Generate response
+            response = generate_response(text, language)
+            
+            # Update history
+            timestamp = datetime.now().strftime("%H:%M:%S")
+            history = f"[{timestamp}] {language}\nYou: {text}\nAgent: {response}\n"
+            
+            return response, history
+        
+        def clear_fn():
+            return "", "Swahili", "", ""
+        
+        # Event handlers
+        generate_btn.click(
+            generate_response_fn,
+            inputs=[user_input, language_selector],
+            outputs=[response_output, conversation_history]
+        )
         
-        status_message = gr.Textbox(
-            label="Status",
-            interactive=False,
-            value="🚀 Ready for enterprise conversations!"
+        clear_btn.click(
+            clear_fn,
+            outputs=[user_input, language_selector, response_output, conversation_history]
         )
         
-        gr.HTML('</div></div>')
+        # Examples
+        gr.Examples(
+            examples=[
+                ["Habari, karibu sana!", "Swahili"],
+                ["Nkuwa okukuyamba mu Luganda", "Luganda"],
+                ["Wĩ mwega, karibu!", "Kikuyu"],
+                ["Pẹlẹ o, bawo ni o se?", "Yoruba"],
+                ["Hello, how are you?", "English"],
+            ],
+            inputs=[user_input, language_selector],
+            outputs=[response_output],
+            fn=generate_response_fn,
+            cache_examples=False,
+        )
         
         # Footer
-        gr.HTML("""
-            <div class="footer-premium">
-                <p><strong>JamboGPT Enterprise</strong> - Beating Sarvam AI in African Language AI</p>
-                <p><a href="https://huggingface.co/spaces/stano03/jambogpt">Hugging Face</a> • <a href="https://github.com/stano03/jambogpt">GitHub</a> • <a href="https://huggingface.co/datasets/stano03/jambogpt-kenyan-languages">Dataset</a></p>
-            </div>
-        """)
+        gr.Markdown("""
+        ---
+        **JamboGPT** - Making AI Accessible to African Languages
         
-        # Connect button
-        process_btn.click(
-            fn=process_text_input,
-            inputs=[text_input, language_choice],
-            outputs=[audio_output, agent_response, status_message]
-        )
+        🔗 [GitHub](https://github.com/stano03/jambogpt) | 
+        📊 [Dataset](https://huggingface.co/datasets/stano03/jambogpt-real-dataset) | 
+        🤖 [Models](https://huggingface.co/stano03)
+        
+        Powered by [Crane AI Labs](https://craneailabs.com) & [Meta MMS](https://huggingface.co/facebook/mms-tts)
+        
+        **Crane AI Models Integrated:**
+        - 🇰🇪 Swahili Gemma 1B - English-to-Swahili translation & conversational AI
+        - 🇺🇬 Ganda Gemma 1B - English-to-Luganda translation & conversational AI
+        """)
     
     return demo
 
 if __name__ == "__main__":
-    print("🚀 Creating JamboGPT Enterprise Interface...")
+    print("\n🚀 Creating JamboGPT Interface with Crane AI Models...")
     demo = create_interface()
     
-    print("=" * 60)
-    print("✅ JamboGPT Enterprise is ready!")
-    print("=" * 60)
+    print("✅ Interface created successfully!")
+    print("\n" + "=" * 80)
+    print("🌍 JamboGPT with Crane AI Labs Models is ready!")
+    print("=" * 80)
     
     demo.launch(
         server_name="0.0.0.0",