Update core/language_detector.py

core/language_detector.py

@@ -1,205 +1,125 @@
-# language_detector.py - UPDATED FOR CODE FORMATTING (200+ lines)
-from langdetect import detect, DetectorFactory
-import re
-
-DetectorFactory.seed = 0
-
-def detect_input_language(text):
-    """Detect if text is Hindi, English or Mixed"""
-    try:
-        clean_text = re.sub(r'[^\w\s]', '', text)
-        if not clean_text.strip():
-            return 'mixed'
-        
-        lang = detect(clean_text)
-        
-        # Hindi detection
-        hindi_chars = re.findall(r'[\u0900-\u097F]', text)
-        if lang == 'hi' or hindi_chars:
-            # Check if mixed with English
-            english_chars = re.findall(r'[a-zA-Z]', text)
-            if hindi_chars and english_chars:
-                return 'mixed'
-            return 'hindi'
-        
-        # English detection
-        if lang == 'en':
-            return 'english'
-        
-        return 'mixed'
-    except:
-        return 'mixed'
-
-def get_system_prompt(lang_mode, username):
-    """Generate system prompt based on language and intent"""
-    
-    # AUMCORE AI SYSTEM KNOWLEDGE (CRITICAL - ADDED NEW)
-    SYSTEM_KNOWLEDGE = f"""AUMCORE AI SYSTEM INFORMATION (NEVER FORGET):
-    1. ENDPOINTS: /system/status, /system/task, /reset, /chat, /
-    2. /system/status → AumCoreMaster orchestrator status check
-    3. /system/task → Heavy computational tasks
-    4. URL: https://aumcoreai-aumcore-ai.hf.space
-    5. Version: 2.0.1-Stable, Logs: logs/aumcore_main.log
-    6. When asked about system, ALWAYS mention these endpoints"""
-    
-    # CORE RULES - UPDATED FOR MARKDOWN FORMATTING
-    core_rules = f"""
-    ROLE: AumCore AI - Senior Coding Assistant
-    USER: {username}
-    
-    CRITICAL RULES:
-    1. CODE vs CHAT DECISION:
-       - CODE WHEN: User says 'code', 'program', 'script', 'function', 'create', 'build'
-       - CHAT WHEN: General conversation, greetings, knowledge questions
-       - EXAMPLES: 
-           * "google drive code" → CODE IN MARKDOWN BLOCKS
-           * "hello how are you" → TEXT RESPONSE
-           * "koi bhajan aata hai" → TEXT RESPONSE
-    
-    2. CODE FORMATTING (CRITICAL):
-       - ALWAYS use markdown code blocks for Python code
-       - FORMAT: ```python\\ncode here\\n```
-       - NEVER output raw code without markdown blocks
-       - EXAMPLES:
-           ✅ CORRECT: ```python\\nfrom google.colab import drive\\ndrive.mount('/content/gdrive')\\n```
-           ❌ WRONG: from google.colab import drive
-       
-    3. ERROR HANDLING:
-       - If user shows error, analyze and provide corrected code IN MARKDOWN
-       - Include brief explanation before/after code block
-    
-    4. CODE QUALITY:
-       - Production-ready code
-       - Error handling included
-       - Proper structure
-       
-    5. RESPONSE STRUCTURE:
-       - Code responses: Explanation (if needed) + Markdown code block
-       - Chat responses: Plain text only
-       - Error fixes: Error analysis + Corrected code in markdown
-       
-    6. SYSTEM KNOWLEDGE: {SYSTEM_KNOWLEDGE}
-    """
+# SIMPLE CODE GENERATOR WITH COLORFUL SYNTAX HIGHLIGHTING
+def generate_basic_code(task):
+    """Generate basic code templates with colorful syntax highlighting"""
+    task_lower = task.lower()
     
-    # LANGUAGE SPECIFIC STYLES - UPDATED WITH SYSTEM EXAMPLES
-    styles = {
-        'hindi': f"""
-        STYLE: 100% Hindi (code markdown blocks ke alawa)
-        EXAMPLES:
-        - User: "नमस्ते, कोड बताओ" 
-          You: ```python\\nfrom google.colab import drive\\ndrive.mount('/content/gdrive')\\n```
-        - User: "system status check करो" 
-          You: "/system/status endpoint से AumCoreMaster orchestrator की status check करें"
-        - User: "क्या हाल है" 
-          You: "सब ठीक है {username} भाई!"
-        - User: "endpoints बताओ" 
-          You: "/system/status, /system/task, /reset, /chat, / - ये सभी endpoints हैं"
-        """,
-        
-        'english': f"""
-        STYLE: 100% English (except code in markdown blocks)
-        EXAMPLES:
-        - User: "hello, give code" 
-          You: ```python\\nfrom google.colab import drive\\ndrive.mount('/content/gdrive')\\n```
-        - User: "check system orchestrator status" 
-          You: "Use /system/status endpoint to check AumCoreMaster orchestrator"
-        - User: "how are you" 
-          You: "I'm good {username}!"
-        - User: "list endpoints" 
-          You: "Endpoints: /system/status, /system/task, /reset, /chat, /"
-        """,
-        
-        'mixed': f"""
-        STYLE: 60% English + 40% Hindi (natural blend)
-        EXAMPLES:
-        - User: "hi bhai, code de" 
-          You: ```python\\nfrom google.colab import drive\\ndrive.mount('/content/gdrive')\\n```
-        - User: "system ki status bata" 
-          You: "/system/status endpoint se system orchestrator ki status check karo"
-        - User: "are yaar, kya haal hai" 
-          You: "Sab badhiya hai {username} bhai!"
-        - User: "endpoints batana" 
-          You: "Endpoints hai: /system/status, /system/task, /reset, /chat, /"
-        """
+    # GitHub-like syntax highlighting colors
+    colors = {
+        'keyword': '#569CD6',      # def, import, from
+        'function': '#DCDCAA',     # function names
+        'string': '#CE9178',       # strings
+        'comment': '#608B4E',      # comments
+        'number': '#B5CEA8',       # numbers
+        'operator': '#D4D4D4',     # =, (), ., :
+        'class': '#4EC9B0',        # class names
+        'decorator': '#C586C0',    # @app.get
     }
     
-    # COMBINE EVERYTHING
-    full_prompt = f"""{core_rules}
-    
-    {styles.get(lang_mode, styles['mixed'])}
-    
-    FINAL REMINDER: 
-    - You are {username}'s helpful AI assistant
-    - FOR CODE: ALWAYS use ```python\\ncode\\n``` format
-    - KNOW YOUR SYSTEM: AumCore-AI with specific endpoints
-    - When asked about system, REFER to /system/status and /system/task
-    """
+    # Colorful Google Drive Mount Code
+    drive_code = f'''<div class="code-container">
+<div class="code-header">
+<div class="code-lang"><i class="fab fa-google-drive"></i> Google Colab</div>
+<button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy Code</button>
+</div>
+<pre><code style="color: #f8f8f2; background: #0d1117; padding: 20px; border-radius: 0 0 8px 8px; display: block; overflow-x: auto; font-family: 'Fira Code', monospace; line-height: 1.5;">
+<span style="color: {colors['comment']}"># Mount Google Drive in Google Colab</span>
+<span style="color: {colors['keyword']}">from</span> <span style="color: {colors['function']}">google.colab</span> <span style="color: {colors['keyword']}">import</span> <span style="color: {colors['function']}">drive</span>
+
+<span style="color: {colors['function']}">drive</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">mount</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&#x27;/content/gdrive&#x27;</span><span style="color: {colors['operator']}">)</span>
+
+<span style="color: {colors['comment']}"># After authorization, access your drive</span>
+<span style="color: {colors['keyword']}">import</span> <span style="color: {colors['function']}">os</span>
+
+<span style="color: {colors['comment']}"># List files in drive</span>
+<span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&quot;Mounted at: /content/gdrive&quot;</span><span style="color: {colors['operator']}">)</span>
+<span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&quot;Files:&quot;</span><span style="color: {colors['operator']}">,</span> <span style="color: {colors['function']}">os</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">listdir</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&#x27;/content/gdrive&#x27;</span><span style="color: {colors['operator']}">))</span>
+</code></pre>
+</div>'''
     
-    return full_prompt.strip()
+    # Colorful FastAPI Web App Code
+    web_code = f'''<div class="code-container">
+<div class="code-header">
+<div class="code-lang"><i class="fas fa-globe"></i> FastAPI Web App</div>
+<button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy Code</button>
+</div>
+<pre><code style="color: #f8f8f2; background: #0d1117; padding: 20px; border-radius: 0 0 8px 8px; display: block; overflow-x: auto; font-family: 'Fira Code', monospace; line-height: 1.5;">
+<span style="color: {colors['keyword']}">from</span> <span style="color: {colors['function']}">fastapi</span> <span style="color: {colors['keyword']}">import</span> <span style="color: {colors['function']}">FastAPI</span>
+<span style="color: {colors['keyword']}">import</span> <span style="color: {colors['function']}">uvicorn</span>
 
-# SIMPLE CODE GENERATOR WITH MARKDOWN
-def generate_basic_code(task):
-    """Generate basic code templates with markdown"""
-    templates = {
-        'web': """```python
-from fastapi import FastAPI
-import uvicorn
+<span style="color: {colors['comment']}"># Create FastAPI app</span>
+<span style="color: {colors['function']}">app</span> <span style="color: {colors['operator']}">=</span> <span style="color: {colors['class']}">FastAPI</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['operator']}">)</span>
 
-app = FastAPI()
+<span style="color: {colors['decorator']}">@app</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">get</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&quot;/&quot;</span><span style="color: {colors['operator']}">)</span>
+<span style="color: {colors['keyword']}">def</span> <span style="color: {colors['function']}">home</span><span style="color: {colors['operator']}">():</span>
+    <span style="color: {colors['keyword']}">return</span> <span style="color: {colors['operator']}">{{</span><span style="color: {colors['string']}">&quot;message&quot;</span><span style="color: {colors['operator']}">:</span> <span style="color: {colors['string']}">&quot;Hello from AumCore AI! 🚀&quot;</span><span style="color: {colors['operator']}">}}</span>
 
-@app.get("/")
-def home():
-    return {"message": "Hello from AumCore AI"}
+<span style="color: {colors['decorator']}">@app</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">get</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&quot;/health&quot;</span><span style="color: {colors['operator']}">)</span>
+<span style="color: {colors['keyword']}">def</span> <span style="color: {colors['function']}">health_check</span><span style="color: {colors['operator']}">():</span>
+    <span style="color: {colors['keyword']}">return</span> <span style="color: {colors['operator']}">{{</span><span style="color: {colors['string']}">&quot;status&quot;</span><span style="color: {colors['operator']}">:</span> <span style="color: {colors['string']}">&quot;healthy&quot;</span><span style="color: {colors['operator']}">,</span> <span style="color: {colors['string']}">&quot;service&quot;</span><span style="color: {colors['operator']}">:</span> <span style="color: {colors['string']}">&quot;AumCore AI&quot;</span><span style="color: {colors['operator']}">}}</span>
 
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
-```""",
-        'data': """```python
-import pandas as pd
-import numpy as np
+<span style="color: {colors['keyword']}">if</span> <span style="color: {colors['variable']}">__name__</span> <span style="color: {colors['operator']}">==</span> <span style="color: {colors['string']}">&quot;__main__&quot;</span><span style="color: {colors['operator']}">:</span>
+    <span style="color: {colors['function']}">uvicorn</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">run</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['function']}">app</span><span style="color: {colors['operator']}">,</span> <span style="color: {colors['variable']}">host</span><span style="color: {colors['operator']}">=</span><span style="color: {colors['string']}">&quot;0.0.0.0&quot;</span><span style="color: {colors['operator']}">,</span> <span style="color: {colors['variable']}">port</span><span style="color: {colors['operator']}">=</span><span style="color: {colors['number']}">8000</span><span style="color: {colors['operator']}">)</span>
+</code></pre>
+</div>'''
+    
+    # Colorful Data Analysis Code
+    data_code = f'''<div class="code-container">
+<div class="code-header">
+<div class="code-lang"><i class="fas fa-chart-bar"></i> Data Analysis</div>
+<button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy Code</button>
+</div>
+<pre><code style="color: #f8f8f2; background: #0d1117; padding: 20px; border-radius: 0 0 8px 8px; display: block; overflow-x: auto; font-family: 'Fira Code', monospace; line-height: 1.5;">
+<span style="color: {colors['keyword']}">import</span> <span style="color: {colors['function']}">pandas</span> <span style="color: {colors['keyword']}">as</span> <span style="color: {colors['function']}">pd</span>
+<span style="color: {colors['keyword']}">import</span> <span style="color: {colors['function']}">numpy</span> <span style="color: {colors['keyword']}">as</span> <span style="color: {colors['function']}">np</span>
+<span style="color: {colors['keyword']}">import</span> <span style="color: {colors['function']}">matplotlib.pyplot</span> <span style="color: {colors['keyword']}">as</span> <span style="color: {colors['function']}">plt</span>
 
-# Load data
-df = pd.read_csv("data.csv")
+<span style="color: {colors['comment']}"># Load dataset</span>
+<span style="color: {colors['function']}">df</span> <span style="color: {colors['operator']}">=</span> <span style="color: {colors['function']}">pd</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">read_csv</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&quot;data.csv&quot;</span><span style="color: {colors['operator']}">)</span>
 
-# Basic analysis
-print(f"Shape: {df.shape}")
-print(f"Columns: {list(df.columns)}")
-print(f"Summary:\\n{df.describe()}")
-```""",
-        'drive': """```python
-from google.colab import drive
-drive.mount('/content/gdrive')
-```"""
-    }
+<span style="color: {colors['comment']}"># Basic information</span>
+<span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">f&quot;📊 Dataset Shape: </span><span style="color: {colors['operator']}">{{</span><span style="color: {colors['function']}">df</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">shape</span><span style="color: {colors['operator']}">}}</span><span style="color: {colors['string']}">&quot;</span><span style="color: {colors['operator']}">)</span>
+<span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">f&quot;📋 Columns: </span><span style="color: {colors['operator']}">{{</span><span style="color: {colors['function']}">list</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['function']}">df</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">columns</span><span style="color: {colors['operator']}">)}}</span><span style="color: {colors['string']}">&quot;</span><span style="color: {colors['operator']}">)</span>
+<span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&quot;\\n📈 Basic Statistics:&quot;</span><span style="color: {colors['operator']}">)</span>
+<span style="color: {colors='function'}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['function']}">df</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">describe</span><span style="color: {colors['operator']}">())</span>
+
+<span style="color: {colors['comment']}"># Check for missing values</span>
+<span style="color: {colors['function']}">missing</span> <span style="color: {colors['operator']}">=</span> <span style="color: {colors['function']}">df</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">isnull</span><span style="color: {colors['operator']}">()</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">sum</span><span style="color: {colors['operator']}">()</span>
+<span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">f&quot;\\n⚠️  Missing Values:\\n</span><span style="color: {colors['operator']}">{{</span><span style="color: {colors['function']}">missing</span><span style="color: {colors['operator']}">}}</span><span style="color: {colors['string']}">&quot;</span><span style="color: {colors['operator']}">)</span>
+
+<span style="color: {colors['comment']}"># Create a simple visualization</span>
+<span style="color: {colors['function']}">plt</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">figure</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['variable']}">figsize</span><span style="color: {colors['operator']}">=</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['number']}">10</span><span style="color: {colors['operator']}">,</span> <span style="color: {colors['number']}">6</span><span style="color: {colors['operator']}">))</span>
+<span style="color: {colors['function']}">df</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">hist</span><span style="color: {colors['operator']}">()</span>
+<span style="color: {colors['function']}">plt</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">tight_layout</span><span style="color: {colors['operator']}">()</span>
+<span style="color: {colors['function']}">plt</span><span style="color: {colors['operator']}">.</span><span style="color: {colors['function']}">show</span><span style="color: {colors['operator']}">()</span>
+</code></pre>
+</div>'''
     
-    task_lower = task.lower()
+    # Determine which code to return
     if 'drive' in task_lower or 'mount' in task_lower:
-        return templates['drive']
-    elif 'web' in task_lower or 'app' in task_lower:
-        return templates['web']
-    elif 'data' in task_lower or 'analy' in task_lower:
-        return templates['data']
-    
-    return templates['drive']  # Default
+        return drive_code
+    elif 'web' in task_lower or 'app' in task_lower or 'api' in task_lower:
+        return web_code
+    elif 'data' in task_lower or 'analy' in task_lower or 'pandas' in task_lower:
+        return data_code
+    else:
+        # Default colorful hello world
+        return f'''<div class="code-container">
+<div class="code-header">
+<div class="code-lang"><i class="fas fa-code"></i> Python</div>
+<button class="copy-btn" onclick="copyCode(this)"><i class="fas fa-copy"></i> Copy Code</button>
+</div>
+<pre><code style="color: #f8f8f2; background: #0d1117; padding: 20px; border-radius: 0 0 8px 8px; display: block; overflow-x: auto; font-family: 'Fira Code', monospace; line-height: 1.5;">
+<span style="color: {colors['keyword']}">def</span> <span style="color: {colors['function']}">hello_aumcore</span><span style="color: {colors['operator']}">():</span>
+    <span style="color: {colors['string']}">&quot;&quot;&quot;</span>
+<span style="color: {colors['string']}>    AumCore AI Sample Function</span>
+<span style="color: {colors['string']}>    With GitHub-like syntax highlighting</span>
+<span style="color: {colors['string']}>    &quot;&quot;&quot;</span>
+    <span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&quot;🚀 Welcome to AumCore AI!&quot;</span><span style="color: {colors['operator']}">)</span>
+    <span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}">&quot;🌈 Colorful syntax highlighting enabled&quot;</span><span style="color: {colors['operator']}">)</span>
+    <span style="color: {colors['keyword']}">return</span> <span style="color: {colors['string']}">&quot;Code execution successful!&quot;</span>
 
-# Test function
-if __name__ == "__main__":
-    # Test detection
-    tests = [
-        "नमस्ते",
-        "hello world",
-        "hi bhai kya haal hai",
-        "google drive mount code do"
-    ]
-    
-    for test in tests:
-        lang = detect_input_language(test)
-        print(f"{test[:20]:20} -> {lang}")
-    
-    # Test code generation with markdown
-    test_code = generate_basic_code("google drive")
-    print(f"\nGenerated code sample:\n{test_code[:100]}...")
-    
-    print("\n✅ language_detector.py UPDATED for markdown formatting")
-    print("   AI will now output code in ```python blocks")
-    print("   This enables: Code boxes, Copy button, Syntax highlighting")
+<span style="color: {colors['comment']}># Execute the function</span>
+<span style="color: {colors['function']}">result</span> <span style="color: {colors['operator']}">=</span> <span style="color: {colors['function']}">hello_aumcore</span><span style="color: {colors['operator']}">()</span>
+<span style="color: {colors['function']}">print</span><span style="color: {colors['operator']}">(</span><span style="color: {colors['string']}>f&quot;✅ Result: </span><span style="color: {colors['operator']}">{{</span><span style="color: {colors['function']}">result</span><span style="color: {colors['operator']}">}}</span><span style="color: {colors['string']}">&quot;</span><span style="color: {colors['operator']}">)</span>
+</code></pre>
+</div>'''