Upgrade: Nyaya Agent 2.0 - Judicial Lockdown, Aura Themes, and Export Suite

.gitignore

@@ -0,0 +1,17 @@
+# Environments
+.env
+.venv/
+__pycache__/
+*.pyc
+
+# Local Data
+data/
+chroma/
+*.db
+
+# Logs
+*.log
+
+# Docker/Podman local files
+.dockerignore
+docker-compose.yml

Dockerfile

@@ -0,0 +1,29 @@
+FROM python:3.11-slim
+
+# Create a non-root user for Hugging Face security
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+# Switch to root temporarily to install system packages
+USER root
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Switch back to the non-root user
+USER user
+
+# Copy project files and set ownership
+COPY --chown=user . .
+
+# Install dependencies
+RUN pip install --no-cache-dir chromadb pypdf google-genai[embeddings] python-dotenv fastapi uvicorn pydantic-settings python-multipart
+
+# Expose port (Hugging Face uses 7860)
+EXPOSE 7860
+
+# Command to run the application
+CMD ["python", "server.py"]

README.md

@@ -0,0 +1,22 @@
+---
+title: Nyaya Agent Legal Platform
+emoji: 🏛️
+colorFrom: yellow
+colorTo: gray
+sdk: docker
+pinned: false
+app_port: 7860
+---
+
+# 🏛️ Nyaya Agent: Premium Legal AI Platform
+
+A glassmorphic, persona-aware legal assistant for the Indian Judiciary.
+
+## Features:
+- **⚖️ BNS-IPC Bridge**: Instant mapping of IPC sections to Bharatiya Nyaya Sanhita 2023.
+- **🛡️ DPDPA Audit**: Automated Privacy Policy compliance auditing.
+- **🔍 Case Network**: Live judgments and legal research via Indian Kanoon.
+- **📜 Document Drafting**: RTI and Legal Notice automation.
+
+## Deployment:
+This app is hosted as a Docker container. API Keys are managed via Space Secrets.

app.py

@@ -0,0 +1,53 @@
+import sys
+import os
+
+# Add the current directory to path for imports
+sys.path.append(os.getcwd())
+
+from core.agent import NyayaAgent
+from tools.search_tool import search_indian_kanoon, get_act_details, search_legal_documents
+from tools.legal_tools import draft_rti_query, draft_legal_notice, analyze_case_summary, suggest_sc_followup
+from tools.compliance import audit_privacy_policy
+from tools.law_bridge import ipc_to_bns_lookup
+from google.genai import types
+
+def main():
+    print("--- Nyaya Agent: Indian Legal Assistant ---")
+    print("Type 'exit' to quit.\n")
+    
+    agent = NyayaAgent()
+    
+    # Define tools for function calling
+    tools = [
+        search_indian_kanoon, 
+        get_act_details, 
+        search_legal_documents,
+        draft_rti_query,
+        draft_legal_notice,
+        analyze_case_summary,
+        suggest_sc_followup,
+        audit_privacy_policy,
+        ipc_to_bns_lookup
+    ]
+    
+    config = types.GenerateContentConfig(
+        system_instruction=agent.system_instruction,
+        tools=tools,
+        temperature=0.2,
+    )
+
+    while True:
+        user_input = input("You: ")
+        if user_input.lower() in ['exit', 'quit']:
+            break
+            
+        try:
+            # Using the agent's wrapper which includes retry logic
+            response_text = agent.generate_response(user_input, config=config)
+            print(f"\nNyaya Agent: {response_text}")
+            
+        except Exception as e:
+            print(f"An error occurred: {e}")
+
+if __name__ == "__main__":
+    main()

backups/v1_production/Dockerfile

@@ -0,0 +1,29 @@
+FROM python:3.11-slim
+
+# Create a non-root user for Hugging Face security
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+# Switch to root temporarily to install system packages
+USER root
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Switch back to the non-root user
+USER user
+
+# Copy project files and set ownership
+COPY --chown=user . .
+
+# Install dependencies
+RUN pip install --no-cache-dir chromadb pypdf google-genai[embeddings] python-dotenv fastapi uvicorn pydantic-settings python-multipart
+
+# Expose port (Hugging Face uses 7860)
+EXPOSE 7860
+
+# Command to run the application
+CMD ["python", "server.py"]

backups/v1_production/app.py

@@ -0,0 +1,53 @@
+import sys
+import os
+
+# Add the current directory to path for imports
+sys.path.append(os.getcwd())
+
+from core.agent import NyayaAgent
+from tools.search_tool import search_indian_kanoon, get_act_details, search_legal_documents
+from tools.legal_tools import draft_rti_query, draft_legal_notice, analyze_case_summary, suggest_sc_followup
+from tools.compliance import audit_privacy_policy
+from tools.law_bridge import ipc_to_bns_lookup
+from google.genai import types
+
+def main():
+    print("--- Nyaya Agent: Indian Legal Assistant ---")
+    print("Type 'exit' to quit.\n")
+    
+    agent = NyayaAgent()
+    
+    # Define tools for function calling
+    tools = [
+        search_indian_kanoon, 
+        get_act_details, 
+        search_legal_documents,
+        draft_rti_query,
+        draft_legal_notice,
+        analyze_case_summary,
+        suggest_sc_followup,
+        audit_privacy_policy,
+        ipc_to_bns_lookup
+    ]
+    
+    config = types.GenerateContentConfig(
+        system_instruction=agent.system_instruction,
+        tools=tools,
+        temperature=0.2,
+    )
+
+    while True:
+        user_input = input("You: ")
+        if user_input.lower() in ['exit', 'quit']:
+            break
+            
+        try:
+            # Using the agent's wrapper which includes retry logic
+            response_text = agent.generate_response(user_input, config=config)
+            print(f"\nNyaya Agent: {response_text}")
+            
+        except Exception as e:
+            print(f"An error occurred: {e}")
+
+if __name__ == "__main__":
+    main()

backups/v1_production/core/agent.py

@@ -0,0 +1,58 @@
+import os
+from google import genai
+from google.genai import types
+from dotenv import load_dotenv
+
+load_dotenv(override=True)
+
+class NyayaAgent:
+    def __init__(self, model_id='gemini-flash-lite-latest'):
+        self.client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+        self.model_id = model_id
+        self.system_instruction = (
+            "You are 'Nyaya Agent', a specialized Legal & Judiciary Assistant for the Indian Legal System. "
+            "You serve Citizens, Law Students, and Lawyers. Your tone and technicality must adapt based on the user's background.\n\n"
+            "ADAPTATION RULES:\n"
+            "1. CITIZEN/NON-PROFESSIONAL: Use simple language, avoid jargon, and explain concepts like 'Writ' or 'Affidavit'. Always suggest drafting an RTI query if they need data from a government body.\n"
+            "2. LAW STUDENT: Provide technical details, refer to specific case precedents (e.g., Kesavananda Bharati), and suggest checking the latest Supreme Court filings.\n"
+            "3. LAWYER/PROFESSIONAL: Provide high-technicality analysis, cite specific Acts/Sections (IPC, BNS, etc.), and discuss procedural nuances.\n\n"
+            "MANDATORY GUIDELINES:\n"
+            "- Be proactive: If a query involves government inaction or data, suggest an RTI.\n"
+            "- If a query involves a complex constitutional point, suggest checking Supreme Court updates.\n"
+            "- Be precise and refer to specific sections of Indian Acts/Statutes.\n"
+            "- BE GUIDED BY TRUTH: Your underlying principle is 'Satyameva Jayate' (Truth alone triumphs). Ensure your research is accurate and evidence-based.\n"
+            "- DO NOT provide definitive legal advice; clarify that you are a research assistant."
+        )
+
+    def generate_response(self, prompt, config=None):
+        import time
+        max_retries = 5
+        retry_delay = 10  # Increased delay for free tier stability
+
+        if config is None:
+            config = types.GenerateContentConfig(
+                system_instruction=self.system_instruction,
+                temperature=0.2,
+            )
+        
+        for attempt in range(max_retries):
+            try:
+                response = self.client.models.generate_content(
+                    model=self.model_id,
+                    contents=prompt,
+                    config=config
+                )
+                return response.text
+            except Exception as e:
+                if "503" in str(e) or "429" in str(e):
+                    if attempt < max_retries - 1:
+                        print(f"\n[System] Gemini is busy. Retrying in {retry_delay}s... (Attempt {attempt + 1}/{max_retries})", flush=True)
+                        time.sleep(retry_delay)
+                        retry_delay *= 2  # Exponential backoff
+                        continue
+                raise e
+
+if __name__ == "__main__":
+    # Quick test
+    agent = NyayaAgent()
+    print(agent.generate_response("Introduce yourself briefly."))

backups/v1_production/frontend/index.html

@@ -0,0 +1,116 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Nyaya Agent | Indian Legal Assistant</title>
+    <link rel="stylesheet" href="style.css">
+    <!-- Google Fonts -->
+    <link href="https://fonts.googleapis.com/css2?family=Playfair+Display:wght@700&family=Inter:wght@300;400;600&display=swap" rel="stylesheet">
+    <!-- Lucide Icons -->
+    <script src="https://unpkg.com/lucide@latest"></script>
+    <!-- Marked.js for Markdown -->
+    <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
+</head>
+<body>
+    <!-- Onboarding Overlay -->
+    <div id="onboarding-overlay" class="onboarding-container glass">
+        <div class="onboarding-card">
+            <h2 class="playfair">Welcome to Nyaya</h2>
+            <p class="onboarding-sub">सत्यमेव जयते</p>
+            <p class="instruction">To provide the most accurate legal assistance, please select your profile:</p>
+            
+            <div class="persona-grid">
+                <button class="persona-btn" data-persona="lawyer">
+                    <i data-lucide="briefcase"></i>
+                    <span>Lawyer / Judge</span>
+                    <small>High Technicality & Procedural Analysis</small>
+                </button>
+                <button class="persona-btn" data-persona="student">
+                    <i data-lucide="graduation-cap"></i>
+                    <span>Law Student</span>
+                    <small>Landmark Cases & Legal Theory</small>
+                </button>
+                <button class="persona-btn" data-persona="citizen">
+                    <i data-lucide="user"></i>
+                    <span>Citizen</span>
+                    <small>Simple Language & RTI Support</small>
+                </button>
+                <button class="persona-btn" data-persona="explorer">
+                    <i data-lucide="compass"></i>
+                    <span>Explorer</span>
+                    <small>History & System Overview</small>
+                </button>
+            </div>
+        </div>
+    </div>
+
+    <div class="app-container blur-background">
+        <!-- Sidebar -->
+        <aside class="sidebar glass">
+            <div class="logo-section">
+                <div class="legal-tech-logo">
+                    <i data-lucide="scale" class="gold-icon scales"></i>
+                    <i data-lucide="pen-tool" class="gold-icon pen"></i>
+                </div>
+                <h1 class="playfair">Nyaya</h1>
+            </div>
+            <nav class="nav-menu">
+                <button class="nav-item active" data-tool="chat"><i data-lucide="message-square"></i> Legal Chat</button>
+                <button class="nav-item" data-tool="search"><i data-lucide="search"></i> Case Search</button>
+                <button class="nav-item" data-tool="notice"><i data-lucide="file-text"></i> Draft Notice</button>
+                <button class="nav-item" data-tool="rti"><i data-lucide="scroll"></i> RTI Portal</button>
+                <button class="nav-item" data-tool="compliance"><i data-lucide="shield-check"></i> Compliance Audit</button>
+                <button class="nav-item" data-tool="bridge"><i data-lucide="git-branch"></i> Law Bridge</button>
+                <button class="nav-item switch-profile" id="switch-btn"><i data-lucide="user-cog"></i> Switch Profile</button>
+            </nav>
+            <div class="motto-section">
+                <div id="motto-container">
+                    <p class="motto" id="motto-text">सत्यमेव जयते</p>
+                    <span class="motto-en" id="motto-lang">HINDI</span>
+                </div>
+            </div>
+        </aside>
+
+        <!-- Main Content -->
+        <main class="main-content">
+            <header class="top-bar glass">
+                <div class="status-indicator">
+                    <span class="dot online"></span>
+                    <span>Judicial Network Online</span>
+                </div>
+                <div class="user-profile">
+                    <i data-lucide="user" class="profile-icon"></i>
+                    <span id="profile-status">Professional Dashboard</span>
+                </div>
+            </header>
+
+            <div class="chat-wrapper">
+                <div id="chat-messages" class="chat-container">
+                    <div class="message system glass">
+                        <p>Greetings. I am <strong>Nyaya Agent</strong>. I am ready to assist with your legal research, drafting, or compliance queries.</p>
+                    </div>
+                </div>
+                
+                <div class="input-area glass">
+                    <input type="file" id="file-upload" style="display: none;" accept=".pdf,.txt">
+                    <button id="attach-btn" class="icon-btn" title="Attach Document">
+                        <i data-lucide="paperclip"></i>
+                    </button>
+                    <input type="text" id="user-input" placeholder="Enter your legal query here..." autocomplete="off">
+                    <button id="send-btn" class="gold-btn">
+                        <i data-lucide="gavel"></i>
+                        <span>Execute</span>
+                    </button>
+                </div>
+            </div>
+        </main>
+    </div>
+
+    <script src="script.js"></script>
+    <script>
+        // Initialize icons
+        lucide.createIcons();
+    </script>
+</body>
+</html>

backups/v1_production/frontend/script.js

@@ -0,0 +1,221 @@
+// API Configuration
+const API_BASE_URL = '/api';
+let userPersona = 'explorer'; // Default
+
+const chatMessages = document.getElementById('chat-messages');
+const userInput = document.getElementById('user-input');
+const sendBtn = document.getElementById('send-btn');
+const onboardingOverlay = document.getElementById('onboarding-overlay');
+const appContainer = document.querySelector('.app-container');
+const switchBtn = document.getElementById('switch-btn');
+const profileStatus = document.getElementById('profile-status');
+const attachBtn = document.getElementById('attach-btn');
+const fileUpload = document.getElementById('file-upload');
+
+// Handle File Upload
+attachBtn.addEventListener('click', () => fileUpload.click());
+
+fileUpload.addEventListener('change', async (e) => {
+    const file = e.target.files[0];
+    if (!file) return;
+
+    addMessage(`Uploading ${file.name}...`, 'user');
+    
+    const formData = new FormData();
+    formData.append('file', file);
+
+    try {
+        const response = await fetch(`${API_BASE_URL}/upload`, {
+            method: 'POST',
+            body: formData
+        });
+        
+        const data = await response.json();
+        if (data.text) {
+            userInput.value = `Audit this document: \n\n ${data.text.substring(0, 1000)}...`;
+            addMessage(`Document ${file.name} processed. I have extracted the text. Click Execute to start the audit.`, 'system');
+        }
+    } catch (err) {
+        addMessage(`Error uploading file: ${err.message}`, 'system');
+    }
+});
+document.querySelectorAll('.persona-btn').forEach(btn => {
+    btn.addEventListener('click', () => {
+        userPersona = btn.getAttribute('data-persona');
+        console.log('Selected Persona:', userPersona);
+        
+        // Hide onboarding and remove blur
+        onboardingOverlay.style.display = 'none';
+        appContainer.classList.remove('blur-background');
+        
+        // Update top bar status
+        profileStatus.innerText = userPersona.charAt(0).toUpperCase() + userPersona.slice(1) + " Dashboard";
+        
+        addMessage(`Welcome! You are now connected as a **${userPersona.toUpperCase()}**. How can I help you navigate the legal system today?`, 'system');
+    });
+});
+
+switchBtn.addEventListener('click', () => {
+    onboardingOverlay.style.display = 'flex';
+    appContainer.classList.add('blur-background');
+});
+
+// Motto Carousel Data
+const mottos = [
+    { text: "सत्यमेव जयते", lang: "Hindi" },
+    { text: "সত্যমেব জয়তে", lang: "Bengali" },
+    { text: "सत्यमेव जयते", lang: "Marathi" },
+    { text: "సత్యమేవ జయతే", lang: "Telugu" },
+    { text: "சத்யமேவ ஜெயதே", lang: "Tamil" },
+    { text: "સત્યમેવ જયતે", lang: "Gujarati" },
+    { text: "ستیہ میو جیتے", lang: "Urdu" },
+    { text: "ಸತ್ಯಮೇವ ಜಯತೇ", lang: "Kannada" },
+    { text: "ସତ୍ୟମେବ ଜୟତେ", lang: "Odia" },
+    { text: "സത്യമേവ ജയതേ", lang: "Malayalam" }
+];
+
+let currentMottoIndex = 0;
+const mottoText = document.getElementById('motto-text');
+const mottoLang = document.getElementById('motto-lang');
+
+function updateMotto() {
+    mottoText.classList.add('fade-out');
+    mottoLang.classList.add('fade-out');
+
+    setTimeout(() => {
+        currentMottoIndex = (currentMottoIndex + 1) % mottos.length;
+        mottoText.innerText = mottos[currentMottoIndex].text;
+        mottoLang.innerText = mottos[currentMottoIndex].lang;
+        
+        mottoText.classList.remove('fade-out');
+        mottoLang.classList.remove('fade-out');
+    }, 500);
+}
+
+setInterval(updateMotto, 4000); // Change every 4 seconds
+
+function addMessage(text, sender) {
+    const msgDiv = document.createElement('div');
+    msgDiv.classList.add('message', sender, 'glass');
+    
+    // Render Markdown using Marked.js
+    let formattedText = marked.parse(text);
+    msgDiv.innerHTML = formattedText;
+
+    // Action Button Detection (Proactive Agent Feature)
+    if (sender === 'system') {
+        const actionArea = document.createElement('div');
+        actionArea.classList.add('action-area');
+
+        if (text.toLowerCase().includes('rti query') || text.toLowerCase().includes('rti act')) {
+            const btn = createActionButton('scroll', 'Draft RTI Query', () => {
+                userInput.value = "Draft an RTI query for this matter.";
+                handleSendMessage();
+            });
+            actionArea.appendChild(btn);
+        }
+
+        if (text.toLowerCase().includes('legal notice')) {
+            const btn = createActionButton('file-text', 'Prepare Legal Notice', () => {
+                userInput.value = "Help me prepare a formal legal notice for this.";
+                handleSendMessage();
+            });
+            actionArea.appendChild(btn);
+        }
+
+        // New: Copy Draft Detection
+        if (text.includes('LEGAL NOTICE') || text.includes('Subject: Application under RTI') || text.includes('Facts:')) {
+            const btn = createActionButton('copy', 'Copy Draft', () => {
+                navigator.clipboard.writeText(text);
+                alert('Draft copied to clipboard! (Satyameva Jayate)');
+            });
+            actionArea.appendChild(btn);
+        }
+
+        if (actionArea.children.length > 0) {
+            msgDiv.appendChild(actionArea);
+        }
+    }
+    
+    chatMessages.appendChild(msgDiv);
+    chatMessages.scrollTop = chatMessages.scrollHeight;
+    lucide.createIcons();
+}
+
+function createActionButton(icon, label, callback) {
+    const btn = document.createElement('button');
+    btn.classList.add('action-btn');
+    btn.innerHTML = `<i data-lucide="${icon}"></i> <span>${label}</span>`;
+    btn.onclick = callback;
+    return btn;
+}
+
+async function handleSendMessage() {
+    const text = userInput.value.trim();
+    if (!text) return;
+
+    // Add user message to UI
+    addMessage(text, 'user');
+    userInput.value = '';
+
+    // Show animated loading indicator
+    const loadingId = 'loading-' + Date.now();
+    const loadingDiv = document.createElement('div');
+    loadingDiv.id = loadingId;
+    loadingDiv.classList.add('message', 'system', 'glass', 'legal-loader');
+    loadingDiv.innerHTML = `
+        <div class="gavel-animate"><i data-lucide="gavel"></i></div>
+        <div class="loader-text">Order in the Court... Analyzing</div>
+    `;
+    chatMessages.appendChild(loadingDiv);
+    lucide.createIcons();
+
+    try {
+        const response = await fetch(`${API_BASE_URL}/chat`, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({ 
+                message: text,
+                persona: userPersona
+            })
+        });
+
+        const data = await response.json();
+        
+        // Remove loading indicator
+        document.getElementById(loadingId).remove();
+
+        if (data.response) {
+            addMessage(data.response, 'system');
+        } else if (data.detail) {
+            addMessage(`Error: ${data.detail}`, 'system');
+        }
+    } catch (error) {
+        document.getElementById(loadingId).remove();
+        addMessage(`Connection Error: Could not reach the Judicial Network. Ensure the server is running.`, 'system');
+        console.error('API Error:', error);
+    }
+}
+
+sendBtn.addEventListener('click', handleSendMessage);
+
+userInput.addEventListener('keypress', (e) => {
+    if (e.key === 'Enter') {
+        handleSendMessage();
+    }
+});
+
+// Sidebar Navigation
+document.querySelectorAll('.nav-item').forEach(item => {
+    item.addEventListener('click', () => {
+        document.querySelectorAll('.nav-item').forEach(btn => btn.classList.remove('active'));
+        item.classList.add('active');
+        
+        const tool = item.getAttribute('data-tool');
+        if (tool !== 'chat') {
+            addMessage(`Switched to **${tool.toUpperCase()}** module. Please enter your specific request.`, 'system');
+        }
+    });
+});

backups/v1_production/frontend/style.css

@@ -0,0 +1,515 @@
+:root {
+    --advocate-black: #0d0d0d;
+    --sidebar-black: #141414;
+    --judicial-gold: #c5a059;
+    --gold-glow: rgba(197, 160, 89, 0.2);
+    --courtroom-maroon: #4a0e0e;
+    --parchment: #f5f5dc;
+    --glass: rgba(255, 255, 255, 0.05);
+    --glass-border: rgba(255, 255, 255, 0.1);
+}
+
+* {
+    margin: 0;
+    padding: 0;
+    box-sizing: border-box;
+}
+
+body {
+    font-family: 'Inter', sans-serif;
+    background-color: var(--advocate-black);
+    color: var(--parchment);
+    height: 100vh;
+    overflow: hidden;
+}
+
+.app-container {
+    display: flex;
+    height: 100%;
+}
+
+/* Glassmorphism Utility */
+.glass {
+    background: var(--glass);
+    backdrop-filter: blur(12px);
+    -webkit-backdrop-filter: blur(12px);
+    border: 1px solid var(--glass-border);
+}
+
+/* Sidebar */
+.sidebar {
+    width: 280px;
+    height: 100%;
+    background: var(--sidebar-black);
+    padding: 2rem;
+    display: flex;
+    flex-direction: column;
+    border-right: 1px solid var(--glass-border);
+}
+
+.logo-section {
+    display: flex;
+    align-items: center;
+    gap: 1.2rem;
+    margin-bottom: 3rem;
+}
+
+.legal-tech-logo {
+    position: relative;
+    width: 45px;
+    height: 45px;
+}
+
+.legal-tech-logo .scales {
+    position: absolute;
+    top: 0;
+    left: 0;
+    z-index: 2;
+    animation: balance 4s infinite ease-in-out;
+}
+
+.legal-tech-logo .pen {
+    position: absolute;
+    bottom: -5px;
+    right: -5px;
+    width: 20px;
+    height: 20px;
+    opacity: 0.8;
+    animation: write-glow 3s infinite ease-in-out;
+}
+
+@keyframes balance {
+    0%, 100% { transform: rotate(-5deg); }
+    50% { transform: rotate(5deg); }
+}
+
+@keyframes write-glow {
+    0%, 100% { filter: drop-shadow(0 0 2px var(--judicial-gold)); opacity: 0.6; }
+    50% { filter: drop-shadow(0 0 10px var(--judicial-gold)); opacity: 1; }
+}
+
+.playfair {
+    font-family: 'Playfair Display', serif;
+    font-size: 1.8rem;
+    color: var(--judicial-gold);
+    letter-spacing: 1px;
+    background: linear-gradient(
+        to right, 
+        var(--judicial-gold) 20%, 
+        #fff 40%, 
+        var(--judicial-gold) 60%
+    );
+    background-size: 200% auto;
+    background-clip: text;
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    animation: shine 5s linear infinite;
+}
+
+@keyframes shine {
+    to { background-position: 200% center; }
+}
+
+.gold-icon {
+    color: var(--judicial-gold);
+    width: 32px;
+    height: 32px;
+    filter: drop-shadow(0 0 5px var(--gold-glow));
+}
+
+.nav-menu {
+    display: flex;
+    flex-direction: column;
+    gap: 0.5rem;
+    flex-grow: 1;
+}
+
+.nav-item {
+    background: transparent;
+    border: none;
+    color: rgba(245, 245, 220, 0.6);
+    padding: 1rem;
+    display: flex;
+    align-items: center;
+    gap: 1rem;
+    cursor: pointer;
+    border-radius: 8px;
+    transition: all 0.3s ease;
+    font-size: 0.95rem;
+}
+
+.nav-item i {
+    width: 18px;
+}
+
+.nav-item:hover, .nav-item.active {
+    background: var(--glass);
+    color: var(--judicial-gold);
+    border-left: 3px solid var(--judicial-gold);
+}
+
+.motto-section {
+    margin-top: auto;
+    text-align: center;
+    padding-top: 2rem;
+    border-top: 1px solid var(--glass-border);
+}
+
+.motto {
+    font-family: 'Playfair Display', serif;
+    font-size: 1.2rem;
+    color: var(--judicial-gold);
+    transition: opacity 0.5s ease-in-out;
+}
+
+.motto.fade-out {
+    opacity: 0;
+}
+
+.motto-en {
+    font-size: 0.7rem;
+    text-transform: uppercase;
+    letter-spacing: 2px;
+    opacity: 0.5;
+    transition: opacity 0.5s ease-in-out;
+}
+
+.motto-en.fade-out {
+    opacity: 0;
+}
+
+/* Main Content */
+.main-content {
+    flex-grow: 1;
+    display: flex;
+    flex-direction: column;
+    min-height: 0; /* Ensures child flex containers can scroll */
+    background: radial-gradient(circle at top right, #1a0a0a, var(--advocate-black));
+}
+
+.top-bar {
+    padding: 1rem 2rem;
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    border-bottom: 1px solid var(--glass-border);
+}
+
+.status-indicator {
+    display: flex;
+    align-items: center;
+    gap: 0.8rem;
+    font-size: 0.8rem;
+    opacity: 0.8;
+}
+
+.dot {
+    width: 8px;
+    height: 8px;
+    border-radius: 50%;
+}
+
+.dot.online {
+    background-color: #4caf50;
+    box-shadow: 0 0 10px #4caf50;
+}
+
+.chat-wrapper {
+    flex-grow: 1;
+    display: flex;
+    flex-direction: column;
+    padding: 2rem;
+    max-width: 1000px;
+    margin: 0 auto;
+    width: 100%;
+    min-height: 0; /* Crucial for flex scroll */
+}
+
+.chat-container {
+    flex-grow: 1;
+    overflow-y: auto;
+    display: flex;
+    flex-direction: column;
+    gap: 1.5rem;
+    padding-bottom: 2rem;
+    padding-right: 1rem; /* Space for scrollbar */
+}
+
+.message {
+    padding: 1.2rem;
+    border-radius: 12px;
+    max-width: 85%;
+    line-height: 1.6;
+}
+
+/* Markdown Styling */
+.message h1, .message h2, .message h3 {
+    color: var(--judicial-gold);
+    margin-bottom: 0.8rem;
+    font-family: 'Playfair Display', serif;
+}
+
+.message ul, .message ol {
+    margin-left: 1.5rem;
+    margin-bottom: 1rem;
+}
+
+.message p {
+    margin-bottom: 0.8rem;
+}
+
+/* Action Buttons */
+.action-area {
+    display: flex;
+    gap: 0.8rem;
+    margin-top: 1rem;
+    padding-top: 1rem;
+    border-top: 1px solid var(--glass-border);
+}
+
+.action-btn {
+    background: rgba(197, 160, 89, 0.1);
+    border: 1px solid var(--judicial-gold);
+    color: var(--judicial-gold);
+    padding: 0.5rem 1rem;
+    border-radius: 6px;
+    display: flex;
+    align-items: center;
+    gap: 0.5rem;
+    font-size: 0.85rem;
+    cursor: pointer;
+    transition: all 0.2s ease;
+}
+
+.action-btn:hover {
+    background: var(--judicial-gold);
+    color: var(--advocate-black);
+}
+
+.message.system {
+    align-self: flex-start;
+    border-left: 4px solid var(--judicial-gold);
+}
+
+.message.user {
+    align-self: flex-end;
+    background: var(--judicial-gold);
+    color: var(--advocate-black);
+    font-weight: 500;
+}
+
+.input-area {
+    display: flex;
+    gap: 1rem;
+    padding: 1rem;
+    border-radius: 15px;
+    margin-top: auto;
+    align-items: center;
+}
+
+.icon-btn {
+    background: none;
+    border: none;
+    color: var(--judicial-gold);
+    cursor: pointer;
+    opacity: 0.7;
+    transition: all 0.2s;
+    padding: 0.5rem;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+}
+
+.icon-btn:hover {
+    opacity: 1;
+    transform: scale(1.1);
+}
+
+#user-input {
+    flex-grow: 1;
+    background: transparent;
+    border: none;
+    color: var(--parchment);
+    font-size: 1rem;
+    outline: none;
+}
+
+.gold-btn {
+    background: var(--judicial-gold);
+    color: var(--advocate-black);
+    border: none;
+    padding: 0.8rem 1.5rem;
+    border-radius: 8px;
+    display: flex;
+    align-items: center;
+    gap: 0.8rem;
+    font-weight: 600;
+    cursor: pointer;
+    transition: transform 0.2s ease, box-shadow 0.2s ease;
+}
+
+.gold-btn:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 5px 15px var(--gold-glow);
+}
+
+/* Legal Loader Animations */
+.legal-loader {
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    gap: 1rem;
+    padding: 1rem;
+    position: relative;
+}
+
+.legal-loader::after {
+    content: '';
+    position: absolute;
+    top: 50%;
+    left: 50%;
+    width: 20px;
+    height: 20px;
+    background: var(--judicial-gold);
+    border-radius: 50%;
+    transform: translate(-50%, -50%);
+    animation: ripple 2s infinite;
+    z-index: -1;
+    opacity: 0;
+}
+
+@keyframes ripple {
+    0% { width: 0; height: 0; opacity: 0.5; }
+    100% { width: 200px; height: 200px; opacity: 0; }
+}
+
+.gavel-animate {
+    font-size: 2.5rem;
+    color: var(--judicial-gold);
+    animation: strike 1.2s infinite ease-in-out;
+    display: inline-block;
+}
+
+.loader-text {
+    font-size: 0.8rem;
+    text-transform: uppercase;
+    letter-spacing: 2px;
+    color: var(--judicial-gold);
+    animation: pulse 1.5s infinite;
+}
+
+@keyframes strike {
+    0% { transform: rotate(0deg); }
+    30% { transform: rotate(-45deg); }
+    50% { transform: rotate(10deg); }
+    100% { transform: rotate(0deg); }
+}
+
+@keyframes pulse {
+    0%, 100% { opacity: 0.4; }
+    50% { opacity: 1; }
+}
+
+/* Onboarding Overlay */
+.onboarding-container {
+    position: fixed;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+    z-index: 1000;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    background: rgba(13, 13, 13, 0.95);
+    backdrop-filter: blur(20px);
+}
+
+.onboarding-card {
+    background: var(--sidebar-black);
+    padding: 3rem;
+    border-radius: 20px;
+    border: 1px solid var(--judicial-gold);
+    text-align: center;
+    max-width: 600px;
+    width: 90%;
+    box-shadow: 0 0 50px rgba(197, 160, 89, 0.1);
+}
+
+.onboarding-sub {
+    color: var(--judicial-gold);
+    font-size: 1.5rem;
+    margin: 1rem 0;
+    font-family: 'Playfair Display', serif;
+}
+
+.instruction {
+    opacity: 0.7;
+    margin-bottom: 2rem;
+}
+
+.persona-grid {
+    display: grid;
+    grid-template-columns: 1fr 1fr;
+    gap: 1.5rem;
+}
+
+.persona-btn {
+    background: var(--glass);
+    border: 1px solid var(--glass-border);
+    color: var(--parchment);
+    padding: 1.5rem;
+    border-radius: 12px;
+    cursor: pointer;
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    gap: 0.8rem;
+    transition: all 0.3s ease;
+}
+
+.persona-btn i {
+    width: 24px;
+    height: 24px;
+    color: var(--judicial-gold);
+}
+
+.persona-btn:hover {
+    background: var(--gold-glow);
+    border-color: var(--judicial-gold);
+    transform: translateY(-5px);
+}
+
+.persona-btn small {
+    font-size: 0.7rem;
+    opacity: 0.6;
+    margin-top: 0.3rem;
+}
+
+.switch-profile {
+    margin-top: 2rem;
+    border-top: 1px solid var(--glass-border) !important;
+    border-radius: 0 !important;
+}
+
+.blur-background {
+    filter: blur(10px);
+}
+
+/* Custom Scrollbar */
+::-webkit-scrollbar {
+    width: 6px;
+}
+
+::-webkit-scrollbar-track {
+    background: transparent;
+}
+
+::-webkit-scrollbar-thumb {
+    background: var(--glass-border);
+    border-radius: 10px;
+}
+
+::-webkit-scrollbar-thumb:hover {
+    background: var(--judicial-gold);
+}

backups/v1_production/server.py

@@ -0,0 +1,95 @@
+from fastapi import FastAPI, HTTPException, File, UploadFile
+from pydantic import BaseModel
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
+import io
+import os
+import pypdf
+import traceback
+from core.agent import NyayaAgent
+from tools.search_tool import search_indian_kanoon, get_act_details, search_legal_documents
+from tools.legal_tools import draft_rti_query, draft_legal_notice, analyze_case_summary
+from tools.compliance import audit_privacy_policy
+from tools.law_bridge import ipc_to_bns_lookup
+from google.genai import types
+
+# Initialize Agent and App
+agent = NyayaAgent()
+app = FastAPI(title="Nyaya Agent")
+
+# CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+class ChatRequest(BaseModel):
+    message: str
+    persona: str = "explorer"
+
+# API ROUTES MUST COME BEFORE STATIC MOUNT
+@app.post("/api/chat")
+async def chat(request: ChatRequest):
+    try:
+        custom_instruction = f"{agent.system_instruction}\n\nCURRENT USER PERSONA: {request.persona.upper()}. "
+        if request.persona == "lawyer":
+            custom_instruction += "Provide highly technical legal analysis with specific sections and procedural details."
+        elif request.persona == "student":
+            custom_instruction += "Provide academic analysis, citing landmark cases and legal theory."
+        else:
+            custom_instruction += "Use simple language, avoid jargon, and explain basic legal concepts clearly."
+
+        tools = [
+            search_indian_kanoon, 
+            get_act_details, 
+            search_legal_documents,
+            draft_rti_query,
+            draft_legal_notice,
+            analyze_case_summary,
+            audit_privacy_policy,
+            ipc_to_bns_lookup
+        ]
+        
+        config = types.GenerateContentConfig(
+            system_instruction=custom_instruction,
+            tools=tools,
+            temperature=0.2,
+        )
+        
+        response_text = agent.generate_response(request.message, config=config)
+        return {"response": response_text}
+    except Exception as e:
+        print(f"ERROR IN CHAT: {str(e)}", flush=True)
+        traceback.print_exc()
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/api/upload")
+async def upload_file(file: UploadFile = File(...)):
+    try:
+        content = await file.read()
+        text = ""
+        if file.filename.endswith(".pdf"):
+            reader = pypdf.PdfReader(io.BytesIO(content))
+            for page in reader.pages:
+                text += page.extract_text() + "\n"
+        else:
+            text = content.decode("utf-8")
+        return {"filename": file.filename, "text": text}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error processing file: {str(e)}")
+
+@app.get("/api/status")
+async def status():
+    return {"status": "Satyameva Jayate - Online"}
+
+# MOUNT STATIC FILES LAST
+app.mount("/", StaticFiles(directory="frontend", html=True), name="frontend")
+
+if __name__ == "__main__":
+    import uvicorn
+    # Hugging Face uses 7860
+    port = int(os.getenv("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)

backups/v1_production/tools/compliance.py

@@ -0,0 +1,17 @@
+def audit_privacy_policy(policy_text: str):
+    """
+    Analyzes a Privacy Policy against the Digital Personal Data Protection Act (DPDPA) 2023.
+    
+    Args:
+        policy_text: The full text of the privacy policy to audit.
+    """
+    # This tool will be called by the agent to perform a structured audit.
+    # The agent will use its internal knowledge of DPDPA 2023 to identify gaps.
+    return (
+        "DPDPA 2023 AUDIT REPORT\n"
+        "-----------------------\n"
+        "1. NOTICE (Section 5): Analyzing if purpose and data types are specified...\n"
+        "2. CONSENT (Section 6): Checking for 'Clear and Affirmative' consent language...\n"
+        "3. GRIEVANCE REDRESSAL (Section 12): Searching for Grievance Officer contact details...\n\n"
+        "Please provide the policy text, and I will highlight the specific compliance gaps."
+    )

backups/v1_production/tools/ingest.py

@@ -0,0 +1,66 @@
+import os
+import chromadb
+from chromadb.utils import embedding_functions
+from pypdf import PdfReader
+from dotenv import load_dotenv
+
+load_dotenv(override=True)
+
+# Configuration
+CHROMA_PATH = "data/chroma_db"
+COLLECTION_NAME = "nyaya_legal_docs"
+
+# Initialize ChromaDB
+client = chromadb.PersistentClient(path=CHROMA_PATH)
+
+# Use Gemini embeddings via a custom wrapper or OpenAI-compatible one if available
+# For simplicity in this workshop, we'll use a default sentence-transformer if available, 
+# or a simple placeholder. 
+# Better yet: Use Chroma's built-in default for now.
+embedding_func = embedding_functions.DefaultEmbeddingFunction()
+
+collection = client.get_or_create_collection(
+    name=COLLECTION_NAME,
+    embedding_function=embedding_func
+)
+
+def ingest_pdf(file_path):
+    """Reads a PDF, chunks it, and adds it to the vector database."""
+    print(f"Ingesting: {file_path}")
+    reader = PdfReader(file_path)
+    
+    doc_id = os.path.basename(file_path)
+    chunks = []
+    metadata = []
+    ids = []
+    
+    for i, page in enumerate(reader.pages):
+        text = page.extract_text()
+        if text.strip():
+            # Basic chunking by page for now
+            chunks.append(text)
+            metadata.append({"source": doc_id, "page": i + 1})
+            ids.append(f"{doc_id}_page_{i+1}")
+            
+    if chunks:
+        collection.add(
+            documents=chunks,
+            metadatas=metadata,
+            ids=ids
+        )
+        print(f"Successfully added {len(chunks)} pages from {doc_id}")
+
+def search_docs(query, n_results=3):
+    """Searches the collection for relevant snippets."""
+    results = collection.query(
+        query_texts=[query],
+        n_results=n_results
+    )
+    return results
+
+if __name__ == "__main__":
+    # Test with any PDFs in the data folder
+    data_dir = "data"
+    for file in os.listdir(data_dir):
+        if file.endswith(".pdf"):
+            ingest_pdf(os.path.join(data_dir, file))

backups/v1_production/tools/law_bridge.py

@@ -0,0 +1,29 @@
+def ipc_to_bns_lookup(ipc_section: str):
+    """
+    Maps an Indian Penal Code (IPC) section to the corresponding Bharatiya Nyaya Sanhita (BNS) 2023 section.
+    
+    Args:
+        ipc_section: The IPC section number (e.g., '302', '420').
+    """
+    # Key Mappings (Sample for MVP)
+    mappings = {
+        "302": {"bns": "101", "name": "Punishment for Murder", "change": "Punishment remains life/death; restructuring of clauses for clarity."},
+        "420": {"bns": "318", "name": "Cheating and dishonestly inducing delivery of property", "change": "Merged with other cheating sections; broader definition of 'property'."},
+        "498A": {"bns": "85/86", "name": "Cruelty by husband or relatives", "change": "Split into two sections; Section 85 defines cruelty, Section 86 defines punishment."},
+        "376": {"bns": "64-70", "name": "Punishment for Rape", "change": "Extensively expanded; separate sections for aggravated cases and gang rape."},
+        "124A": {"bns": "152", "name": "Sedition (now 'Acts endangering sovereignty')", "change": "The word 'Sedition' is removed; focuses on acts against the state/sovereignty."}
+    }
+    
+    clean_section = ipc_section.upper().replace("IPC", "").strip()
+    
+    if clean_section in mappings:
+        m = mappings[clean_section]
+        return (
+            f"--- BNS-IPC CROSS-REFERENCE ---\n"
+            f"Old Section: IPC {clean_section} ({m['name']})\n"
+            f"New Section: BNS {m['bns']}\n"
+            f"Key Change: {m['change']}\n"
+            f"--------------------------------"
+        )
+    else:
+        return f"Section IPC {clean_section} not found in the immediate lookup table. Please consult the full BNS Schedule or ask me to search for the specific offense name."

backups/v1_production/tools/legal_tools.py

@@ -0,0 +1,52 @@
+def draft_rti_query(subject: str, department: str = "Relevant Public Authority"):
+    """
+    Drafts a formal RTI (Right to Information) query for a given subject.
+    """
+    template = (
+        f"Subject: Application under RTI Act, 2005 - Regarding {subject}\n\n"
+        f"To,\nThe Public Information Officer,\n{department}\n\n"
+        f"Sir/Madam,\n"
+        f"I request you to provide the following information under the RTI Act, 2005:\n"
+        f"1. Detailed records/notifications regarding {subject}.\n"
+        f"2. Any internal circulars or guidelines issued concerning this matter.\n"
+        f"3. Time period of the requested data: [Specify Period].\n\n"
+        f"I am a citizen of India. I have attached the prescribed fee of Rs. 10/-\n\n"
+        f"Regards,\n[Your Name]"
+    )
+    return template
+
+def draft_legal_notice(sender: str, receiver: str, grievance: str, relief: str):
+    """
+    Prepares a formal legal notice for a specific grievance.
+    
+    Args:
+        sender: Name and address of the sender.
+        receiver: Name and address of the receiver.
+        grievance: Detailed description of the legal grievance/dispute.
+        relief: The action or compensation demanded from the receiver.
+    """
+    notice = (
+        f"LEGAL NOTICE\n\n"
+        f"From: {sender}\n"
+        f"To: {receiver}\n\n"
+        f"Under instructions from my client {sender}, I hereby serve you with this legal notice:\n\n"
+        f"1. THE GRIEVANCE: {grievance}\n"
+        f"2. LEGAL VIOLATION: This act is a violation of relevant Indian laws.\n"
+        f"3. THE RELIEF: You are hereby called upon to {relief} within 15 days of receipt of this notice.\n\n"
+        f"Failing which, my client shall be constrained to initiate civil/criminal proceedings against you.\n\n"
+        f"Regards,\n[Advocate Name]"
+    )
+    return notice
+
+def analyze_case_summary(case_text: str):
+    """
+    Analyzes a legal case and provides a structured summary (Facts, Issues, Ratio).
+    
+    Args:
+        case_text: The full text or excerpt of the judgment to analyze.
+    """
+    return "Case Analysis Placeholder: I will provide Facts, Issues, Arguments, and the Ratio Decidendi here."
+
+def suggest_sc_followup():
+    """Provides a suggestion to check the latest Supreme Court filings."""
+    return "Would you like me to check the latest Supreme Court filings or judgments related to this matter?"

backups/v1_production/tools/search_tool.py

@@ -0,0 +1,79 @@
+import os
+import chromadb
+from chromadb.utils import embedding_functions
+
+# Configuration
+CHROMA_PATH = "data/chroma_db"
+COLLECTION_NAME = "nyaya_legal_docs"
+
+def get_vector_collection():
+    client = chromadb.PersistentClient(path=CHROMA_PATH)
+    embedding_func = embedding_functions.DefaultEmbeddingFunction()
+    return client.get_or_create_collection(
+        name=COLLECTION_NAME,
+        embedding_function=embedding_func
+    )
+
+def search_legal_documents(query: str):
+    """
+    Search across the indexed legal documents (Gazettes, Circulars, etc.) using semantic search.
+    
+    Args:
+        query: The natural language query to search for in local documents.
+    """
+    try:
+        collection = get_vector_collection()
+        results = collection.query(
+            query_texts=[query],
+            n_results=3
+        )
+        if results['documents']:
+            return "\n\n".join(results['documents'][0])
+        return "No relevant local documents found."
+    except Exception as e:
+        return f"Error searching local documents: {e}"
+
+import os
+import requests
+
+def search_indian_kanoon(query: str):
+    """
+    Search for legal cases, acts, and statutes on Indian Kanoon (Online).
+    """
+    token = os.getenv("INDIANKANOON_API_KEY")
+    if not token:
+        return f"Search result placeholder for: {query}. (API Token missing in .env. Using internal model knowledge.)"
+
+    try:
+        headers = {
+            "Authorization": f"Token {token}",
+            "Accept": "application/json"
+        }
+        params = {"formInput": query, "pagenum": 0}
+        # Searching for the query
+        response = requests.get(
+            f"https://api.indiankanoon.org/search/", 
+            headers=headers, 
+            params=params
+        )
+        data = response.json()
+        
+        if "results" in data:
+            results = data["results"][:3]
+            summary = "Found these cases on Indian Kanoon:\n"
+            for res in results:
+                summary += f"- {res['title']} ({res['docid']})\n"
+            return summary
+        return "No results found on Indian Kanoon."
+    except Exception as e:
+        return f"Error connecting to Indian Kanoon: {e}"
+
+def get_act_details(act_name: str, section: str = None):
+    """
+    Retrieve details of a specific Indian Act or Section.
+    
+    Args:
+        act_name: Name of the Act (e.g., 'IPC', 'DPDPA').
+        section: Specific section number (optional).
+    """
+    return f"Details for {act_name} Section {section if section else 'All'}. (Data retrieval pending)"

core/agent.py

@@ -0,0 +1,58 @@
+import os
+from google import genai
+from google.genai import types
+from dotenv import load_dotenv
+
+load_dotenv(override=True)
+
+class NyayaAgent:
+    def __init__(self, model_id='gemini-flash-lite-latest'):
+        self.client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+        self.model_id = model_id
+        self.system_instruction = (
+            "You are 'Nyaya Agent', a specialized Legal & Judiciary Assistant for the Indian Legal System. "
+            "You serve Citizens, Law Students, and Lawyers. Your tone and technicality must adapt based on the user's background.\n\n"
+            "ADAPTATION RULES:\n"
+            "1. CITIZEN/NON-PROFESSIONAL: Use simple language, avoid jargon, and explain concepts like 'Writ' or 'Affidavit'. Always suggest drafting an RTI query if they need data from a government body.\n"
+            "2. LAW STUDENT: Provide technical details, refer to specific case precedents (e.g., Kesavananda Bharati), and suggest checking the latest Supreme Court filings.\n"
+            "3. LAWYER/PROFESSIONAL: Provide high-technicality analysis, cite specific Acts/Sections (IPC, BNS, etc.), and discuss procedural nuances.\n\n"
+            "MANDATORY GUIDELINES:\n"
+            "- Be proactive: If a query involves government inaction or data, suggest an RTI.\n"
+            "- If a query involves a complex constitutional point, suggest checking Supreme Court updates.\n"
+            "- Be precise and refer to specific sections of Indian Acts/Statutes.\n"
+            "- BE GUIDED BY TRUTH: Your underlying principle is 'Satyameva Jayate' (Truth alone triumphs). Ensure your research is accurate and evidence-based.\n"
+            "- DO NOT provide definitive legal advice; clarify that you are a research assistant."
+        )
+
+    def generate_response(self, prompt, config=None):
+        import time
+        max_retries = 5
+        retry_delay = 10  # Increased delay for free tier stability
+
+        if config is None:
+            config = types.GenerateContentConfig(
+                system_instruction=self.system_instruction,
+                temperature=0.2,
+            )
+        
+        for attempt in range(max_retries):
+            try:
+                response = self.client.models.generate_content(
+                    model=self.model_id,
+                    contents=prompt,
+                    config=config
+                )
+                return response.text
+            except Exception as e:
+                if "503" in str(e) or "429" in str(e):
+                    if attempt < max_retries - 1:
+                        print(f"\n[System] Gemini is busy. Retrying in {retry_delay}s... (Attempt {attempt + 1}/{max_retries})", flush=True)
+                        time.sleep(retry_delay)
+                        retry_delay *= 2  # Exponential backoff
+                        continue
+                raise e
+
+if __name__ == "__main__":
+    # Quick test
+    agent = NyayaAgent()
+    print(agent.generate_response("Introduce yourself briefly."))

frontend/index.html

@@ -0,0 +1,133 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Nyaya Agent | Indian Legal Assistant</title>
+    <link rel="stylesheet" href="style.css">
+    <!-- Google Fonts -->
+    <link href="https://fonts.googleapis.com/css2?family=Playfair+Display:wght@700&family=Inter:wght@300;400;600&display=swap" rel="stylesheet">
+    <!-- Lucide Icons -->
+    <script src="https://unpkg.com/lucide@latest"></script>
+    <!-- Marked.js for Markdown -->
+    <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
+</head>
+<body>
+    <!-- Onboarding Overlay -->
+    <div id="onboarding-overlay" class="onboarding-container glass">
+        <div class="onboarding-card">
+            <div class="judicial-emblem">
+                <i data-lucide="scale"></i>
+            </div>
+            <h2 class="playfair">Welcome to Nyaya</h2>
+            <p class="onboarding-sub">सत्यमेव जयते</p>
+            <p class="instruction">To provide the most accurate legal assistance, please select your profile:</p>
+            
+            <div class="persona-grid">
+                <button class="persona-btn" data-persona="lawyer">
+                    <i data-lucide="briefcase"></i>
+                    <span>Lawyer / Judge</span>
+                    <small>High Technicality & Procedural Analysis</small>
+                </button>
+                <button class="persona-btn" data-persona="student">
+                    <i data-lucide="graduation-cap"></i>
+                    <span>Law Student</span>
+                    <small>Landmark Cases & Legal Theory</small>
+                </button>
+                <button class="persona-btn" data-persona="citizen">
+                    <i data-lucide="user"></i>
+                    <span>Citizen</span>
+                    <small>Simple Language & RTI Support</small>
+                </button>
+                <button class="persona-btn" data-persona="explorer">
+                    <i data-lucide="compass"></i>
+                    <span>Explorer</span>
+                    <small>History & System Overview</small>
+                </button>
+            </div>
+
+            <div class="onboarding-footer">
+                <p>⚠️ <strong>Legal Disclaimer:</strong> Nyaya Agent is an AI assistant and may occasionally provide inaccurate information. This tool is for research and drafting assistance only and is NOT a substitute for professional legal advice.</p>
+            </div>
+        </div>
+    </div>
+
+    <div class="app-container blur-background">
+        <!-- Sidebar -->
+        <aside class="sidebar glass">
+            <div class="logo-section">
+                <div class="legal-tech-logo">
+                    <i data-lucide="scale" class="gold-icon scales"></i>
+                    <i data-lucide="pen-tool" class="gold-icon pen"></i>
+                </div>
+                <h1 class="playfair">Nyaya</h1>
+            </div>
+            <nav class="nav-menu">
+                <button class="nav-item active" data-tool="chat"><i data-lucide="message-square"></i> Legal Chat</button>
+                <button class="nav-item" data-tool="search"><i data-lucide="search"></i> Case Search</button>
+                <button class="nav-item" data-tool="notice"><i data-lucide="file-text"></i> Draft Notice</button>
+                <button class="nav-item" data-tool="rti"><i data-lucide="scroll"></i> RTI Portal</button>
+                <button class="nav-item" data-tool="compliance"><i data-lucide="shield-check"></i> Compliance Audit</button>
+                <button class="nav-item" data-tool="bridge"><i data-lucide="git-branch"></i> Law Bridge</button>
+                <button class="nav-item switch-profile" id="switch-btn"><i data-lucide="user-cog"></i> Switch Profile</button>
+            </nav>
+            <div class="motto-section">
+                <div id="motto-container">
+                    <p class="motto" id="motto-text">सत्यमेव जयते</p>
+                    <span class="motto-en" id="motto-lang">HINDI</span>
+                </div>
+            </div>
+            <div class="sidebar-footer">
+                <p>⚠️ AI Research Assistant. Always verify legal drafts.</p>
+                <div class="footer-icons">
+                    <i data-lucide="shield-check"></i>
+                    <i data-lucide="landmark"></i>
+                </div>
+            </div>
+        </aside>
+
+        <!-- Main Content -->
+        <main class="main-content">
+            <header class="top-bar glass">
+                <div class="status-indicator">
+                    <span class="dot online"></span>
+                    <span>Judicial Network Online</span>
+                </div>
+                <div class="user-profile">
+                    <i data-lucide="user" class="profile-icon"></i>
+                    <span id="profile-status">Professional Dashboard</span>
+                    <div class="info-trigger" id="info-trigger">
+                        <i data-lucide="info"></i>
+                        <div class="persona-tooltip glass" id="persona-tooltip">
+                            Select a profile to see capabilities.
+                        </div>
+                    </div>
+                </div>
+            </header>
+
+            <div class="chat-wrapper">
+                <div id="chat-messages" class="chat-container">
+                </div>
+                
+                <div class="input-area glass">
+                    <input type="file" id="file-upload" style="display: none;" accept=".pdf,.txt">
+                    <button id="attach-btn" class="icon-btn" title="Attach Document">
+                        <i data-lucide="paperclip"></i>
+                    </button>
+                    <input type="text" id="user-input" placeholder="Enter your legal query here..." autocomplete="off">
+                    <button id="send-btn" class="gold-btn">
+                        <i data-lucide="gavel"></i>
+                        <span>Execute</span>
+                    </button>
+                </div>
+            </div>
+        </main>
+    </div>
+
+    <script src="script.js"></script>
+    <script>
+        // Initialize icons
+        lucide.createIcons();
+    </script>
+</body>
+</html>

frontend/script.js

@@ -0,0 +1,347 @@
+// API Configuration
+const API_BASE_URL = '/api';
+let userPersona = 'explorer'; // Default
+
+const chatMessages = document.getElementById('chat-messages');
+const userInput = document.getElementById('user-input');
+const sendBtn = document.getElementById('send-btn');
+const onboardingOverlay = document.getElementById('onboarding-overlay');
+const appContainer = document.querySelector('.app-container');
+const switchBtn = document.getElementById('switch-btn');
+const profileStatus = document.getElementById('profile-status');
+const attachBtn = document.getElementById('attach-btn');
+const fileUpload = document.getElementById('file-upload');
+const personaTooltip = document.getElementById('persona-tooltip');
+const infoTrigger = document.getElementById('info-trigger');
+
+let isRequestInProgress = false;
+
+// Persona Capability Data
+const personaCapabilities = {
+    lawyer: "<strong>Legal Professional Mode</strong><br>Optimized for deep statutory analysis, procedural advice, and drafting formal legal documents with precise terminology.",
+    student: "<strong>Academic Mode</strong><br>Focuses on legal theory, landmark Supreme Court cases, and historical context of Indian laws.",
+    citizen: "<strong>Empowerment Mode</strong><br>Provides simple explanations of legal rights, RTI support, and step-by-step guidance for basic legal issues.",
+    explorer: "<strong>Orientation Mode</strong><br>General overview of the Indian judicial hierarchy, history, and fundamental legal concepts."
+};
+
+function setLockdown(active) {
+    isRequestInProgress = active;
+    if (active) {
+        document.body.classList.add('lockdown-active');
+        userInput.placeholder = "Judicial Lockdown: Analysis in progress...";
+        sendBtn.disabled = true;
+    } else {
+        document.body.classList.remove('lockdown-active');
+        userInput.placeholder = "Enter your legal query here...";
+        sendBtn.disabled = false;
+    }
+}
+
+// Handle File Upload
+attachBtn.addEventListener('click', () => fileUpload.click());
+
+fileUpload.addEventListener('change', async (e) => {
+    const file = e.target.files[0];
+    if (!file) return;
+
+    addMessage(`Uploading ${file.name}...`, 'user');
+    
+    const formData = new FormData();
+    formData.append('file', file);
+
+    try {
+        const response = await fetch(`${API_BASE_URL}/upload`, {
+            method: 'POST',
+            body: formData
+        });
+        
+        const data = await response.json();
+        if (data.text) {
+            userInput.value = `Audit this document: \n\n ${data.text.substring(0, 1000)}...`;
+            addMessage(`Document ${file.name} processed. I have extracted the text. Click Execute to start the audit.`, 'system');
+        }
+    } catch (err) {
+        addMessage(`Error uploading file: ${err.message}`, 'system');
+    }
+});
+document.querySelectorAll('.persona-btn').forEach(btn => {
+    btn.addEventListener('click', () => {
+        userPersona = btn.getAttribute('data-persona');
+        onboardingOverlay.classList.add('hidden');
+        appContainer.classList.remove('blur-background');
+        
+        // Update top bar status
+        profileStatus.innerText = userPersona.charAt(0).toUpperCase() + userPersona.slice(1) + " Dashboard";
+
+        // Apply Dynamic Theme
+        document.documentElement.setAttribute('data-theme', userPersona);
+        
+        // Update Wisdom Tooltip
+        personaTooltip.innerHTML = personaCapabilities[userPersona];
+        
+        addMessage(`Welcome! You are now connected as a **${userPersona.toUpperCase()}**. How can I help you navigate the legal system today?`, 'system');
+    });
+});
+
+function highlightCitations(text) {
+    // Regex to find common Indian legal citations (e.g., Section 302, Article 21, IPC 302)
+    const citationRegex = /\b(Section|Article|Act|Clause)\s+\d+[A-Z]?\b|\b(IPC|BNS|CRPC|CPC|DPDPA)\s+\d+[A-Z]?\b/gi;
+    return text.replace(citationRegex, (match) => {
+        return `<span class="legal-citation" title="Judicial Reference">${match}</span>`;
+    });
+}
+
+switchBtn.addEventListener('click', () => {
+    onboardingOverlay.classList.remove('hidden');
+    appContainer.classList.add('blur-background');
+});
+
+// Motto Carousel Data
+const mottos = [
+    { text: "सत्यमेव जयते", lang: "Hindi" },
+    { text: "সত্যমেব জয়তে", lang: "Bengali" },
+    { text: "सत्यमेव जयते", lang: "Marathi" },
+    { text: "సత్యమేవ జయతే", lang: "Telugu" },
+    { text: "சத்யமேவ ஜெயதே", lang: "Tamil" },
+    { text: "સત્યમેવ જયતે", lang: "Gujarati" },
+    { text: "ستیہ میو جیتے", lang: "Urdu" },
+    { text: "ಸತ್ಯಮೇವ ಜಯತೇ", lang: "Kannada" },
+    { text: "ସତ୍ୟମେବ ଜୟତେ", lang: "Odia" },
+    { text: "സത്യമേവ ജയതേ", lang: "Malayalam" }
+];
+
+let currentMottoIndex = 0;
+const mottoText = document.getElementById('motto-text');
+const mottoLang = document.getElementById('motto-lang');
+
+function updateMotto() {
+    mottoText.classList.add('fade-out');
+    mottoLang.classList.add('fade-out');
+
+    setTimeout(() => {
+        currentMottoIndex = (currentMottoIndex + 1) % mottos.length;
+        mottoText.innerText = mottos[currentMottoIndex].text;
+        mottoLang.innerText = mottos[currentMottoIndex].lang;
+        
+        mottoText.classList.remove('fade-out');
+        mottoLang.classList.remove('fade-out');
+    }, 500);
+}
+
+setInterval(updateMotto, 4000); // Change every 4 seconds
+
+function addMessage(text, sender) {
+    const msgDiv = document.createElement('div');
+    msgDiv.className = `message ${sender}`; // Explicitly set classes
+    
+    // Highlight Citations
+    let highlightedText = highlightCitations(text);
+    
+    // Add Spark Icon for System
+    if (sender === 'system') {
+        const spark = document.createElement('div');
+        spark.className = 'system-spark';
+        spark.innerHTML = '✨';
+        msgDiv.appendChild(spark);
+    }
+    
+    const contentDiv = document.createElement('div');
+    contentDiv.className = 'message-content';
+    contentDiv.innerHTML = marked.parse(highlightedText);
+    msgDiv.appendChild(contentDiv);
+
+    // Action Button Detection (Proactive Agent Feature)
+    if (sender === 'system') {
+        const actionArea = document.createElement('div');
+        actionArea.classList.add('action-area');
+
+        if (text.toLowerCase().includes('rti query') || text.toLowerCase().includes('rti act')) {
+            const btn = createActionButton('scroll', 'Draft RTI Query', () => {
+                userInput.value = "Draft an RTI query for this matter.";
+                handleSendMessage();
+            });
+            actionArea.appendChild(btn);
+        }
+
+        if (text.toLowerCase().includes('legal notice')) {
+            const btn = createActionButton('file-text', 'Prepare Legal Notice', () => {
+                userInput.value = "Help me prepare a formal legal notice for this.";
+                handleSendMessage();
+            });
+            actionArea.appendChild(btn);
+        }
+
+        // New: Copy Draft Detection
+        if (text.includes('LEGAL NOTICE') || text.includes('Subject: Application under RTI') || text.includes('Facts:')) {
+            const copyBtn = createActionButton('copy', 'Copy Draft', () => {
+                navigator.clipboard.writeText(text);
+                alert('Draft copied to clipboard! (Satyameva Jayate)');
+            });
+            actionArea.appendChild(copyBtn);
+
+            const docBtn = createActionButton('file-text', 'Download DOC', () => exportToDoc(text));
+            actionArea.appendChild(docBtn);
+
+            const pdfBtn = createActionButton('download', 'Download PDF', () => exportToPdf(text));
+            actionArea.appendChild(pdfBtn);
+        }
+
+        if (actionArea.children.length > 0) {
+            msgDiv.appendChild(actionArea);
+        }
+    }
+    
+    const time = new Date().toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' });
+    const timeSpan = document.createElement('span');
+    timeSpan.classList.add('message-time');
+    timeSpan.innerText = time;
+    msgDiv.appendChild(timeSpan);
+    
+    chatMessages.appendChild(msgDiv);
+    setTimeout(scrollToBottom, 50); // Small delay for smooth rendering
+    lucide.createIcons();
+}
+
+function scrollToBottom() {
+    chatMessages.scrollTo({
+        top: chatMessages.scrollHeight,
+        behavior: 'smooth'
+    });
+}
+
+function exportToDoc(text) {
+    const header = "<html xmlns:o='urn:schemas-microsoft-com:office:office' xmlns:w='urn:schemas-microsoft-com:office:word' xmlns='http://www.w3.org/TR/REC-html40'><head><meta charset='utf-8'><title>Legal Draft</title></head><body>";
+    const footer = "</body></html>";
+    const sourceHTML = header + marked.parse(text) + footer;
+    
+    const source = 'data:application/vnd.ms-word;charset=utf-8,' + encodeURIComponent(sourceHTML);
+    const fileDownload = document.createElement("a");
+    document.body.appendChild(fileDownload);
+    fileDownload.href = source;
+    fileDownload.download = 'Nyaya_Legal_Draft.doc';
+    fileDownload.click();
+    document.body.removeChild(fileDownload);
+}
+
+function exportToPdf(text) {
+    // Premium Print Mode for PDF
+    const printWindow = window.open('', '_blank');
+    printWindow.document.write(`
+        <html>
+            <head>
+                <title>Nyaya Legal Draft</title>
+                <style>
+                    body { font-family: 'Times New Roman', serif; padding: 50px; line-height: 1.6; }
+                    h1, h2 { color: #4a0e0e; border-bottom: 2px solid #c5a059; }
+                    .watermark { position: fixed; top: 50%; left: 50%; transform: translate(-50%, -50%) rotate(-45deg); opacity: 0.1; font-size: 80px; z-index: -1; }
+                </style>
+            </head>
+            <body>
+                <div class="watermark">NYAYA AGENT</div>
+                ${marked.parse(text)}
+                <script>setTimeout(() => { window.print(); window.close(); }, 500);</script>
+            </body>
+        </html>
+    `);
+    printWindow.document.close();
+}
+
+function createActionButton(icon, label, callback) {
+    const btn = document.createElement('button');
+    btn.classList.add('action-btn');
+    btn.innerHTML = `<i data-lucide="${icon}"></i> <span>${label}</span>`;
+    btn.onclick = callback;
+    return btn;
+}
+
+async function handleSendMessage() {
+    if (isRequestInProgress) return;
+    const text = userInput.value.trim();
+    if (!text) return;
+
+    setLockdown(true);
+
+    // Add user message to UI
+    addMessage(text, 'user');
+    userInput.value = '';
+
+    // Show animated loading indicator
+    const loadingId = 'loading-' + Date.now();
+    const loadingDiv = document.createElement('div');
+    loadingDiv.id = loadingId;
+    loadingDiv.classList.add('message', 'system', 'glass', 'legal-loader');
+    loadingDiv.innerHTML = `
+        <div class="gavel-animate"><i data-lucide="gavel"></i></div>
+        <div class="loader-text">Order in the Court... Analyzing</div>
+    `;
+    chatMessages.appendChild(loadingDiv);
+    lucide.createIcons();
+    scrollToBottom();
+
+    try {
+        const response = await fetch(`${API_BASE_URL}/chat`, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify({ 
+                message: text,
+                persona: userPersona
+            })
+        });
+
+        const data = await response.json();
+        
+        // Remove loading indicator
+        document.getElementById(loadingId).remove();
+
+        if (data.response) {
+            addMessage(data.response, 'system');
+        } else if (data.detail) {
+            addMessage(`Error: ${data.detail}`, 'system');
+        }
+    } catch (error) {
+        if (document.getElementById(loadingId)) document.getElementById(loadingId).remove();
+        addMessage(`Connection Error: Could not reach the Judicial Network. Ensure the server is running.`, 'system');
+        console.error('API Error:', error);
+    } finally {
+        setLockdown(false);
+    }
+}
+
+sendBtn.addEventListener('click', handleSendMessage);
+
+userInput.addEventListener('keypress', (e) => {
+    if (e.key === 'Enter') {
+        handleSendMessage();
+    }
+});
+
+addMessage("Greetings. I am **Nyaya Agent**. I am ready to assist with your legal research, drafting, or compliance queries.", "system");
+
+// Sidebar Navigation
+const toolExamples = {
+    search: "Search for landmark cases. e.g., 'Find Supreme Court judgments on Right to Privacy (Justice K.S. Puttaswamy case)'",
+    notice: "Prepare a formal notice. e.g., 'Help me draft a legal notice for non-payment of rent by a tenant.'",
+    rti: "Draft an RTI application. e.g., 'Draft an RTI to find out the status of road repairs in Ward 12.'",
+    compliance: "Audit documents for DPDPA 2023. e.g., 'Audit this privacy policy for compliance with the new data protection act.'",
+    bridge: "Map old laws to new. e.g., 'What is the new BNS section for IPC 302 (Murder)?'"
+};
+
+document.querySelectorAll('.nav-item').forEach(item => {
+    item.addEventListener('click', () => {
+        if (isRequestInProgress) {
+            alert('Judicial Lockdown Active: Please wait for the current analysis to complete.');
+            return;
+        }
+
+        document.querySelectorAll('.nav-item').forEach(btn => btn.classList.remove('active'));
+        item.classList.add('active');
+        
+        const tool = item.getAttribute('data-tool');
+        if (tool !== 'chat') {
+            const example = toolExamples[tool];
+            addMessage(`Switched to **${tool.toUpperCase()}** module.\n\n**Example:** ${example}`, 'system');
+        }
+    });
+});

frontend/style.css

@@ -0,0 +1,805 @@
+:root {
+    --advocate-black: #0d0d0d;
+    --sidebar-black: #141414;
+    --judicial-gold: #c5a059;
+    --gold-glow: rgba(197, 160, 89, 0.2);
+    --courtroom-maroon: #4a0e0e;
+    --parchment: #f5f5dc;
+    --glass: rgba(255, 255, 255, 0.05);
+    --glass-border: rgba(255, 255, 255, 0.1);
+
+    /* Persona Themes */
+    --theme-lawyer: #c5a059;
+    --theme-student: #4ade80;
+    --theme-citizen: #38bdf8;
+    --theme-explorer: #a855f7;
+    
+    --accent-color: var(--theme-lawyer);
+}
+
+/* Dynamic Theme Application */
+:root[data-theme="lawyer"] { --accent-color: var(--theme-lawyer); --judicial-gold: var(--theme-lawyer); }
+:root[data-theme="student"] { --accent-color: var(--theme-student); --judicial-gold: var(--theme-student); }
+:root[data-theme="citizen"] { --accent-color: var(--theme-citizen); --judicial-gold: var(--theme-citizen); }
+:root[data-theme="explorer"] { --accent-color: var(--theme-explorer); --judicial-gold: var(--theme-explorer); }
+
+/* Smart Citation Highlighting */
+.legal-citation {
+    color: var(--judicial-gold);
+    font-weight: 600;
+    text-decoration: underline dotted;
+    cursor: help;
+    background: rgba(255, 255, 255, 0.05);
+    padding: 0 4px;
+    border-radius: 4px;
+    transition: all 0.3s ease;
+}
+
+.legal-citation:hover {
+    background: var(--judicial-gold);
+    color: var(--advocate-black);
+}
+
+* {
+    margin: 0;
+    padding: 0;
+    box-sizing: border-box;
+}
+
+body {
+    font-family: 'Inter', sans-serif;
+    background-color: var(--advocate-black);
+    color: var(--parchment);
+    height: 100vh;
+    overflow: hidden;
+}
+
+.app-container {
+    display: flex;
+    height: 100%;
+}
+
+/* Glassmorphism Utility */
+.glass {
+    background: var(--glass);
+    backdrop-filter: blur(12px);
+    -webkit-backdrop-filter: blur(12px);
+    border: 1px solid var(--glass-border);
+}
+
+/* Sidebar */
+.sidebar {
+    width: 280px;
+    height: 100%;
+    background: var(--sidebar-black);
+    padding: 2rem;
+    display: flex;
+    flex-direction: column;
+    border-right: 1px solid var(--glass-border);
+}
+
+.logo-section {
+    display: flex;
+    align-items: center;
+    gap: 1.2rem;
+    margin-bottom: 3rem;
+}
+
+.legal-tech-logo {
+    position: relative;
+    width: 45px;
+    height: 45px;
+}
+
+.legal-tech-logo .scales {
+    position: absolute;
+    top: 0;
+    left: 0;
+    z-index: 2;
+    animation: balance 4s infinite ease-in-out;
+}
+
+.legal-tech-logo .pen {
+    position: absolute;
+    bottom: -5px;
+    right: -5px;
+    width: 20px;
+    height: 20px;
+    opacity: 0.8;
+    animation: write-glow 3s infinite ease-in-out;
+}
+
+@keyframes balance {
+    0%, 100% { transform: rotate(-5deg); }
+    50% { transform: rotate(5deg); }
+}
+
+@keyframes write-glow {
+    0%, 100% { filter: drop-shadow(0 0 2px var(--judicial-gold)); opacity: 0.6; }
+    50% { filter: drop-shadow(0 0 10px var(--judicial-gold)); opacity: 1; }
+}
+
+.playfair {
+    font-family: 'Playfair Display', serif;
+    font-size: 1.8rem;
+    color: var(--judicial-gold);
+    letter-spacing: 1px;
+    background: linear-gradient(
+        to right, 
+        var(--judicial-gold) 20%, 
+        #fff 40%, 
+        var(--judicial-gold) 60%
+    );
+    background-size: 200% auto;
+    background-clip: text;
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    animation: shine 5s linear infinite;
+}
+
+@keyframes shine {
+    to { background-position: 200% center; }
+}
+
+.gold-icon {
+    color: var(--judicial-gold);
+    width: 32px;
+    height: 32px;
+    filter: drop-shadow(0 0 5px var(--gold-glow));
+}
+
+.nav-menu {
+    display: flex;
+    flex-direction: column;
+    gap: 0.5rem;
+    flex-grow: 1;
+}
+
+.nav-item {
+    background: transparent;
+    border: none;
+    color: rgba(245, 245, 220, 0.6);
+    padding: 1rem;
+    display: flex;
+    align-items: center;
+    gap: 1rem;
+    cursor: pointer;
+    border-radius: 8px;
+    transition: all 0.3s ease;
+    font-size: 0.95rem;
+}
+
+.nav-item i {
+    width: 18px;
+}
+
+.nav-item:hover, .nav-item.active {
+    background: var(--glass);
+    color: var(--judicial-gold);
+    border-left: 3px solid var(--judicial-gold);
+}
+
+.motto-section {
+    margin-top: auto;
+    text-align: center;
+    padding-top: 2rem;
+    border-top: 1px solid var(--glass-border);
+}
+
+.motto {
+    font-family: 'Playfair Display', serif;
+    font-size: 1.2rem;
+    color: var(--judicial-gold);
+    transition: opacity 0.5s ease-in-out;
+}
+
+.motto.fade-out {
+    opacity: 0;
+}
+
+.motto-en {
+    font-size: 0.7rem;
+    text-transform: uppercase;
+    letter-spacing: 2px;
+    opacity: 0.5;
+    transition: opacity 0.5s ease-in-out;
+}
+
+.motto-en.fade-out {
+    opacity: 0;
+}
+
+/* Main Content */
+.main-content {
+    flex-grow: 1;
+    display: flex;
+    flex-direction: column;
+    min-height: 0; /* Ensures child flex containers can scroll */
+    background: radial-gradient(circle at top right, #1a0a0a, var(--advocate-black));
+}
+
+.top-bar {
+    padding: 1rem 2rem;
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    border-bottom: 1px solid var(--glass-border);
+}
+
+.status-indicator {
+    display: flex;
+    align-items: center;
+    gap: 0.8rem;
+    font-size: 0.8rem;
+    opacity: 0.8;
+}
+
+.dot {
+    width: 8px;
+    height: 8px;
+    border-radius: 50%;
+}
+
+.dot.online {
+    background-color: #4caf50;
+    box-shadow: 0 0 10px #4caf50;
+}
+
+.chat-wrapper {
+    flex-grow: 1;
+    display: flex;
+    flex-direction: column;
+    padding: 2rem;
+    max-width: 1000px;
+    margin: 0 auto;
+    width: 100%;
+    min-height: 0; /* Crucial for flex scroll */
+}
+
+.chat-container {
+    flex-grow: 1;
+    overflow-y: auto;
+    display: flex;
+    flex-direction: column;
+    gap: 1.5rem;
+    padding-bottom: 2rem;
+    padding-right: 1rem; /* Space for scrollbar */
+}
+
+.message {
+    padding: 1.5rem;
+    border-radius: 12px;
+    max-width: 85%;
+    line-height: 1.7;
+    margin-bottom: 2rem;
+    position: relative;
+    box-shadow: 0 4px 15px rgba(0,0,0,0.2);
+    animation: fadeIn 0.4s ease;
+}
+
+@keyframes fadeIn {
+    from { opacity: 0; transform: translateY(10px); }
+    to { opacity: 1; transform: translateY(0); }
+}
+
+.message.system {
+    align-self: flex-start;
+    background: rgba(255, 255, 255, 0.03);
+    border-left: 4px solid var(--judicial-gold);
+    border-top-left-radius: 0;
+}
+
+.message.user {
+    align-self: flex-end;
+    background: var(--judicial-gold);
+    color: var(--advocate-black);
+    font-weight: 600;
+    border-bottom-right-radius: 0;
+    box-shadow: 0 8px 25px var(--gold-glow);
+}
+
+.message.user::before {
+    content: '⚖️ QUERY';
+    position: absolute;
+    top: -22px;
+    right: 0;
+    font-size: 0.65rem;
+    letter-spacing: 1px;
+    color: var(--judicial-gold);
+    opacity: 0.8;
+}
+
+.message.system::before {
+    content: '🏛️ RESPONSE';
+    position: absolute;
+    top: -22px;
+    left: 0;
+    font-size: 0.65rem;
+    letter-spacing: 1px;
+    color: var(--judicial-gold);
+    opacity: 0.8;
+}
+
+/* Markdown Styling */
+.message h1, .message h2, .message h3 {
+    color: var(--judicial-gold);
+    margin: 1.5rem 0 1rem 0;
+    font-family: 'Playfair Display', serif;
+    border-bottom: 1px solid rgba(197, 160, 89, 0.2);
+    padding-bottom: 0.5rem;
+}
+
+.message ul, .message ol {
+    margin-left: 1.5rem;
+    margin-bottom: 1rem;
+}
+
+.message p {
+    margin-bottom: 0.8rem;
+}
+
+/* Action Buttons */
+.action-area {
+    display: flex;
+    gap: 0.8rem;
+    margin-top: 1rem;
+    padding-top: 1rem;
+    border-top: 1px solid var(--glass-border);
+}
+
+.action-btn {
+    background: rgba(197, 160, 89, 0.1);
+    border: 1px solid var(--judicial-gold);
+    color: var(--judicial-gold);
+    padding: 0.5rem 1rem;
+    border-radius: 6px;
+    display: flex;
+    align-items: center;
+    gap: 0.5rem;
+    font-size: 0.85rem;
+    cursor: pointer;
+    transition: all 0.2s ease;
+}
+
+.action-btn:hover {
+    background: var(--judicial-gold);
+    color: var(--advocate-black);
+}
+
+#chat-messages {
+    flex: 1;
+    overflow-y: auto;
+    padding: 20px 20px 100px 20px; /* Large bottom padding for loader space */
+    display: flex;
+    flex-direction: column;
+    gap: 15px;
+    scroll-behavior: smooth;
+}
+
+.chat-wrapper {
+    flex-grow: 1;
+    display: flex;
+    flex-direction: column;
+    overflow: hidden;
+    position: relative;
+}
+
+.input-area {
+    display: flex;
+    gap: 1rem;
+    padding: 1rem;
+    border-radius: 15px;
+    margin: 10px 20px 20px 20px;
+    align-items: center;
+    background: rgba(255, 255, 255, 0.05);
+    backdrop-filter: blur(10px);
+    border: 1px solid var(--glass-border);
+    z-index: 10;
+}
+
+.icon-btn {
+    background: none;
+    border: none;
+    color: var(--judicial-gold);
+    cursor: pointer;
+    opacity: 0.7;
+    transition: all 0.2s;
+    padding: 0.5rem;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+}
+
+.icon-btn:hover {
+    opacity: 1;
+    transform: scale(1.1);
+}
+
+#user-input {
+    flex-grow: 1;
+    background: transparent;
+    border: none;
+    color: var(--parchment);
+    font-size: 1rem;
+    outline: none;
+}
+
+.gold-btn {
+    background: var(--judicial-gold);
+    color: var(--advocate-black);
+    border: none;
+    padding: 0.8rem 1.5rem;
+    border-radius: 8px;
+    display: flex;
+    align-items: center;
+    gap: 0.8rem;
+    font-weight: 600;
+    cursor: pointer;
+    transition: transform 0.2s ease, box-shadow 0.2s ease;
+}
+
+.gold-btn:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 5px 15px var(--gold-glow);
+}
+
+/* Legal Loader Animations */
+.legal-loader {
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    gap: 1rem;
+    padding: 1rem;
+    position: relative;
+}
+
+.legal-loader::after {
+    content: '';
+    position: absolute;
+    top: 50%;
+    left: 50%;
+    width: 20px;
+    height: 20px;
+    background: var(--judicial-gold);
+    border-radius: 50%;
+    transform: translate(-50%, -50%);
+    animation: ripple 2s infinite;
+    z-index: -1;
+    opacity: 0;
+}
+
+@keyframes ripple {
+    0% { width: 0; height: 0; opacity: 0.5; }
+    100% { width: 200px; height: 200px; opacity: 0; }
+}
+
+.gavel-animate {
+    font-size: 2.5rem;
+    color: var(--judicial-gold);
+    animation: strike 1.2s infinite ease-in-out;
+    display: inline-block;
+}
+
+.loader-text {
+    font-size: 0.8rem;
+    text-transform: uppercase;
+    letter-spacing: 2px;
+    color: var(--judicial-gold);
+    animation: pulse 1.5s infinite;
+}
+
+@keyframes strike {
+    0% { transform: rotate(0deg); }
+    30% { transform: rotate(-45deg); }
+    50% { transform: rotate(10deg); }
+    100% { transform: rotate(0deg); }
+}
+
+@keyframes pulse {
+    0%, 100% { opacity: 0.4; }
+    50% { opacity: 1; }
+}
+
+/* Weighing the Evidence Animation */
+.lockdown-active .scales {
+    animation: weighing 2s infinite ease-in-out;
+}
+
+@keyframes weighing {
+    0%, 100% { transform: rotate(-5deg); }
+    50% { transform: rotate(5deg); }
+}
+
+/* Onboarding Enhancements */
+.judicial-emblem {
+    margin-bottom: 1rem;
+    opacity: 0.15;
+    transform: scale(3);
+    color: var(--judicial-gold);
+}
+
+.onboarding-footer {
+    margin-top: 2rem;
+    padding-top: 1.5rem;
+    border-top: 1px solid var(--glass-border);
+    font-size: 0.75rem;
+    line-height: 1.5;
+    opacity: 0.7;
+    text-align: justify;
+}
+
+.onboarding-footer strong {
+    color: var(--judicial-gold);
+}
+
+.sidebar-footer {
+    padding: 1rem;
+    border-top: 1px solid var(--glass-border);
+    margin-top: auto;
+    font-size: 0.7rem;
+    color: var(--judicial-gold);
+    opacity: 0.8;
+    text-align: center;
+}
+
+.footer-icons {
+    display: flex;
+    justify-content: center;
+    gap: 1rem;
+    margin-top: 0.5rem;
+}
+
+.footer-icons i {
+    width: 14px;
+    height: 14px;
+}
+
+.message {
+    padding: 1.2rem 1.8rem;
+    border-radius: 20px;
+    line-height: 1.8;
+    position: relative;
+    box-shadow: 0 4px 15px rgba(0,0,0,0.3);
+    animation: fadeIn 0.4s ease;
+    word-wrap: break-word;
+}
+
+.message.user {
+    align-self: flex-end;
+    background: rgba(197, 160, 89, 0.9); /* Gold Bubble */
+    color: #0d0d0d;
+    font-weight: 500;
+    max-width: 70%;
+    border-bottom-right-radius: 4px;
+}
+
+.message.system {
+    align-self: flex-start;
+    background: rgba(255, 255, 255, 0.05); /* Glass Parchment */
+    color: var(--parchment);
+    max-width: 85%;
+    border: 1px solid var(--glass-border);
+    border-bottom-left-radius: 4px;
+}
+
+.system-spark {
+    position: absolute;
+    left: -35px;
+    top: 5px;
+    font-size: 1.2rem;
+    opacity: 0.8;
+}
+
+.message-content {
+    flex: 1;
+}
+
+.message-content p {
+    margin-bottom: 0.8rem;
+}
+
+.message-content p:last-child {
+    margin-bottom: 0;
+}
+
+/* Onboarding Overlay */
+.onboarding-container {
+    position: fixed;
+    top: 0;
+    left: 0;
+    width: 100%;
+    height: 100%;
+    z-index: 9999;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    background: rgba(10, 10, 10, 0.98);
+    backdrop-filter: blur(25px);
+    transition: opacity 0.5s ease;
+}
+
+.onboarding-container.hidden {
+    display: none !important;
+}
+
+.onboarding-card {
+    background: #141414;
+    padding: 3rem;
+    border-radius: 24px;
+    border: 1px solid var(--judicial-gold);
+    text-align: center;
+    max-width: 650px;
+    width: 90%;
+    box-shadow: 0 20px 60px rgba(0,0,0,0.8);
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    position: relative;
+    z-index: 10000; /* Higher than container */
+    pointer-events: auto !important;
+}
+
+.onboarding-sub {
+    color: var(--judicial-gold);
+    font-size: 1.5rem;
+    margin: 1rem 0;
+    font-family: 'Playfair Display', serif;
+}
+
+.instruction {
+    opacity: 0.7;
+    margin-bottom: 2rem;
+}
+
+.persona-grid {
+    display: grid;
+    grid-template-columns: 1fr 1fr;
+    gap: 1.5rem;
+}
+
+.persona-btn {
+    background: var(--glass);
+    border: 1px solid var(--glass-border);
+    color: var(--parchment);
+    padding: 1.5rem;
+    border-radius: 12px;
+    cursor: pointer;
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    gap: 0.8rem;
+    transition: all 0.3s ease;
+}
+
+.persona-btn i {
+    width: 24px;
+    height: 24px;
+    color: var(--judicial-gold);
+}
+
+.persona-btn:hover {
+    background: var(--gold-glow);
+    border-color: var(--judicial-gold);
+    transform: translateY(-5px);
+}
+
+.persona-btn small {
+    font-size: 0.7rem;
+    opacity: 0.6;
+    margin-top: 0.3rem;
+}
+
+.switch-profile {
+    margin-top: 2rem;
+    border-top: 1px solid var(--glass-border) !important;
+    border-radius: 0 !important;
+}
+
+.blur-background {
+    filter: blur(10px);
+}
+
+/* Custom Scrollbar */
+::-webkit-scrollbar {
+    width: 6px;
+}
+
+::-webkit-scrollbar-track {
+    background: transparent;
+}
+
+::-webkit-scrollbar-thumb {
+    background: var(--glass-border);
+    border-radius: 10px;
+}
+
+::-webkit-scrollbar-thumb:hover {
+    background: var(--judicial-gold);
+}
+
+/* Persona Info Icon & Tooltip */
+.info-trigger {
+    margin-left: 10px;
+    cursor: pointer;
+    position: relative;
+    display: flex;
+    align-items: center;
+    color: var(--judicial-gold);
+    opacity: 0.7;
+    transition: opacity 0.3s ease;
+}
+
+.info-trigger:hover {
+    opacity: 1;
+}
+
+.persona-tooltip {
+    position: absolute;
+    top: 40px;
+    right: 0;
+    width: 250px;
+    padding: 15px;
+    border-radius: 12px;
+    font-size: 0.85rem;
+    line-height: 1.4;
+    color: var(--parchment);
+    opacity: 0;
+    visibility: hidden;
+    transform: translateY(-10px);
+    transition: all 0.3s ease;
+    z-index: 1000;
+    border: 1px solid rgba(197, 160, 89, 0.2);
+    box-shadow: 0 10px 30px rgba(0,0,0,0.5);
+    background: rgba(13, 13, 13, 0.95);
+    backdrop-filter: blur(20px);
+}
+
+.info-trigger:hover .persona-tooltip {
+    opacity: 1;
+    visibility: visible;
+    transform: translateY(0);
+}
+
+.persona-tooltip strong {
+    color: var(--judicial-gold);
+    display: block;
+    margin-bottom: 5px;
+    font-family: 'Playfair Display', serif;
+}
+
+/* Judicial Lockdown State */
+.lockdown-active {
+    pointer-events: none;
+    cursor: not-allowed !important;
+}
+
+.lockdown-active aside, 
+.lockdown-active .top-bar, 
+.lockdown-active .input-area {
+    opacity: 0.6;
+    filter: grayscale(0.5);
+    transition: all 0.3s ease;
+}
+
+.lockdown-active .input-area {
+    border-color: rgba(255, 0, 0, 0.2);
+}
+
+.lockdown-active #user-input::placeholder {
+    color: rgba(255, 255, 255, 0.2);
+}
+
+.message-time {
+    position: absolute;
+    bottom: 5px;
+    right: 12px;
+    font-size: 0.6rem;
+    opacity: 0.5;
+    font-family: 'Inter', sans-serif;
+    font-weight: 400;
+}
+
+.message.user .message-time {
+    color: var(--advocate-black);
+    opacity: 0.7;
+}

list_models.py

@@ -0,0 +1,17 @@
+import os
+from google import genai
+from dotenv import load_dotenv
+
+load_dotenv(override=True)
+
+def list_available_models():
+    client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+    print("--- Available Models ---")
+    try:
+        for model in client.models.list():
+            print(f"- {model.name}")
+    except Exception as e:
+        print(f"Error: {e}")
+
+if __name__ == "__main__":
+    list_available_models()

nyaya_project_brief.md

@@ -0,0 +1,64 @@
+# 🏛️ Nyaya Agent: Technical & Functional Overview
+
+**Nyaya Agent** is a premium, AI-powered Judicial Assistant designed to navigate the complexities of the Indian Legal System. It leverages state-of-the-art Generative AI to provide persona-aware legal research, statutory mapping, and compliance auditing.
+
+---
+
+## 🚀 Tech Stack
+| Component | Technology |
+| :--- | :--- |
+| **AI Engine** | Google Gemini 2.0 Flash / Flash Lite |
+| **Backend** | FastAPI (Python 3.11) |
+| **Database** | ChromaDB (Vector Database for RAG) |
+| **Frontend** | Vanilla HTML5, CSS3 (Glassmorphism), Lucide Icons |
+| **Deployment** | Docker / Podman (Containerized) |
+| **Hosting** | Hugging Face Spaces (Cloud) |
+
+---
+
+## 🏛️ Core Features
+
+### ⚖️ BNS-IPC Law Bridge
+A specialized tool designed for the 2023 criminal law transition.
+- **Function**: Instantly maps old **Indian Penal Code (IPC)** sections to the new **Bharatiya Nyaya Sanhita (BNS)**.
+- **Value**: Provides lawyers with immediate statutory references and highlights key changes in definitions and punishments.
+
+### 🛡️ DPDPA Compliance Audit
+Automated auditing for the **Digital Personal Data Protection Act (DPDPA), 2023**.
+- **Function**: Analyzes privacy policies (PDF/Text) to identify legal gaps and compliance risks.
+- **Value**: Simplifies regulatory adherence for businesses and legal consultants.
+
+### 🔍 Judicial Network (Live Search)
+Integrated with the **Indian Kanoon API**.
+- **Function**: Real-time retrieval of landmark judgments, statutes, and case laws.
+- **Value**: Provides evidence-based legal research directly from official judicial records.
+
+---
+
+## 👤 User Personas
+The system dynamically adapts its language, technical depth, and output format based on the selected profile:
+
+1.  **Lawyer**: Focuses on procedural details, specific sections, and judicial precedents.
+2.  **Student**: Emphasizes legal theory, landmark cases, and academic analysis.
+3.  **Citizen**: Provides simple, jargon-free explanations of fundamental legal rights.
+4.  **Explorer**: Offers a general overview and historical context of the Indian legal system.
+
+---
+
+## ⚙️ System Workflow
+1.  **Input**: User interacts with the **Glassmorphism UI** or uploads a document.
+2.  **Routing**: The **FastAPI Backend** processes the request and identifies the active **Persona**.
+3.  **Reasoning**: The **Gemini AI Engine** determines if it needs to call specialized **Tools** (Kanoon Search, BNS Bridge, etc.).
+4.  **Synthesis**: The agent combines statutory data with its generative capabilities to provide a persona-tailored response.
+5.  **Output**: Results are displayed in the interactive chat with support for one-click **Copy-to-Clipboard** and **Drafting**.
+
+---
+
+## 🔐 Security & Deployment
+- **Containerization**: The entire ecosystem is packaged as a Docker image for consistent performance across environments.
+- **Secrets Management**: API Keys are never hardcoded; they are managed via **Space Secrets** (Hugging Face) or `.env` files (Local).
+- **Sub-Path Namespacing**: The application is architected to be cloud-ready and easily deployable to platforms like Hugging Face.
+
+---
+**Satyameva Jayate**  
+*Truth Alone Triumphs*

requirements.txt

@@ -0,0 +1,10 @@
+google-genai
+python-dotenv
+fastapi
+uvicorn
+pydantic-settings
+python-multipart
+pypdf
+chromadb
+requests
+markdown

server.py

@@ -0,0 +1,95 @@
+from fastapi import FastAPI, HTTPException, File, UploadFile
+from pydantic import BaseModel
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
+import io
+import os
+import pypdf
+import traceback
+from core.agent import NyayaAgent
+from tools.search_tool import search_indian_kanoon, get_act_details, search_legal_documents
+from tools.legal_tools import draft_rti_query, draft_legal_notice, analyze_case_summary
+from tools.compliance import audit_privacy_policy
+from tools.law_bridge import ipc_to_bns_lookup
+from google.genai import types
+
+# Initialize Agent and App
+agent = NyayaAgent()
+app = FastAPI(title="Nyaya Agent")
+
+# CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+class ChatRequest(BaseModel):
+    message: str
+    persona: str = "explorer"
+
+# API ROUTES MUST COME BEFORE STATIC MOUNT
+@app.post("/api/chat")
+async def chat(request: ChatRequest):
+    try:
+        custom_instruction = f"{agent.system_instruction}\n\nCURRENT USER PERSONA: {request.persona.upper()}. "
+        if request.persona == "lawyer":
+            custom_instruction += "Provide highly technical legal analysis with specific sections and procedural details."
+        elif request.persona == "student":
+            custom_instruction += "Provide academic analysis, citing landmark cases and legal theory."
+        else:
+            custom_instruction += "Use simple language, avoid jargon, and explain basic legal concepts clearly."
+
+        tools = [
+            search_indian_kanoon, 
+            get_act_details, 
+            search_legal_documents,
+            draft_rti_query,
+            draft_legal_notice,
+            analyze_case_summary,
+            audit_privacy_policy,
+            ipc_to_bns_lookup
+        ]
+        
+        config = types.GenerateContentConfig(
+            system_instruction=custom_instruction,
+            tools=tools,
+            temperature=0.2,
+        )
+        
+        response_text = agent.generate_response(request.message, config=config)
+        return {"response": response_text}
+    except Exception as e:
+        print(f"ERROR IN CHAT: {str(e)}", flush=True)
+        traceback.print_exc()
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/api/upload")
+async def upload_file(file: UploadFile = File(...)):
+    try:
+        content = await file.read()
+        text = ""
+        if file.filename.endswith(".pdf"):
+            reader = pypdf.PdfReader(io.BytesIO(content))
+            for page in reader.pages:
+                text += page.extract_text() + "\n"
+        else:
+            text = content.decode("utf-8")
+        return {"filename": file.filename, "text": text}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error processing file: {str(e)}")
+
+@app.get("/api/status")
+async def status():
+    return {"status": "Satyameva Jayate - Online"}
+
+# MOUNT STATIC FILES LAST
+app.mount("/", StaticFiles(directory="frontend", html=True), name="frontend")
+
+if __name__ == "__main__":
+    import uvicorn
+    # Hugging Face uses 7860
+    port = int(os.getenv("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)

session_snapshot.md

@@ -0,0 +1,36 @@
+# 🏛️ Nyaya Agent: Session Snapshot
+
+**Date**: May 12, 2026
+**Status**: Successfully Deployed & Verified 🚀
+
+---
+
+## ✅ Completed Milestones
+1.  **AI Engine**: Configured to use `gemini-flash-lite-latest` for maximum free-tier stability and high quota.
+2.  **Architecture**: FastAPI backend with a Root-Level routing system optimized for Hugging Face Spaces.
+3.  **UI/UX**: Full Glassmorphism judicial dashboard with persona-adaptive interactions.
+4.  **Legal Tools**: 
+    - **BNS-IPC Bridge** (Criminal Law mapping)
+    - **DPDPA 2023 Audit** (Privacy Policy analysis)
+    - **Indian Kanoon Search** (Case law retrieval)
+5.  **Cloud Deployment**: 100% functional Docker deployment on Hugging Face Spaces.
+
+---
+
+## 🔑 Active Configuration
+- **App URL**: `http://localhost:8000` (Local) | `huggingface.co/spaces/SeriousSam07/nyaya-agent` (Cloud)
+- **Container Port**: `7860` (Internal) -> `8000` (External)
+- **Base Path**: Root (`/`)
+- **API Endpoint**: `/api`
+
+---
+
+## 🛠️ Next Steps / Ideas for Future Sessions
+1.  **Legal History Sidebar**: Implement `localStorage` to save and display previous chat sessions.
+2.  **Judicial News Ticker**: Add a live RSS feed for the latest Supreme Court and High Court judgments.
+3.  **Advanced PDF Parsing**: Improve table extraction for complex legal documents.
+4.  **Audio Interface**: Add "Speech-to-Text" for lawyers to dictate notes directly.
+
+---
+**Current State**: All systems operational. 
+**Satyameva Jayate.**

test_drive.py

@@ -0,0 +1,54 @@
+import requests
+import json
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+# Updated for root-level deployment
+BASE_URL = "http://localhost:8000/api"
+
+def test_endpoint(path, payload, method="POST"):
+    print(f"Testing {path}...")
+    try:
+        if method == "POST":
+            response = requests.post(f"{BASE_URL}{path}", json=payload)
+        else:
+            response = requests.get(f"{BASE_URL}{path}")
+            
+        print(f"Status: {response.status_code}")
+        # Only print first 500 chars of response to keep logs clean
+        resp_json = response.json()
+        resp_str = json.dumps(resp_json, indent=2)
+        print(f"Response: {resp_str[:500]}..." if len(resp_str) > 500 else f"Response: {resp_str}")
+        print("-" * 30)
+        return response.status_code == 200
+    except Exception as e:
+        print(f"Error: {e}")
+        return False
+
+if __name__ == "__main__":
+    print("=== NYAYA AGENT FINAL ACCEPTANCE TEST ===\n")
+    
+    # 1. Test Status
+    test_endpoint("/status", {}, method="GET")
+
+    # 2. Test Lawyer Persona + BNS Bridge
+    test_endpoint("/chat", {
+        "message": "What is the new BNS section for IPC 302?",
+        "persona": "lawyer"
+    })
+
+    # 3. Test Citizen Persona + General Advice
+    test_endpoint("/chat", {
+        "message": "How do I file an RTI for my property records?",
+        "persona": "citizen"
+    })
+
+    # 4. Test Compliance Audit Tool
+    test_endpoint("/chat", {
+        "message": "Audit this: We collect your email for marketing. You can contact us at privacy@example.com.",
+        "persona": "lawyer"
+    })
+
+    print("\nTests complete. If all statuses are 200, the Judicial Hub is ready for the team!")

test_kanoon.py

@@ -0,0 +1,16 @@
+import os
+from dotenv import load_dotenv
+from tools.search_tool import search_indian_kanoon
+
+load_dotenv(override=True)
+
+def run_test():
+    print("--- Indian Kanoon API Connection Test ---")
+    query = "Constitution"
+    result = search_indian_kanoon(query)
+    print(f"Query: {query}")
+    print(f"Result:\n{result}")
+    print("-" * 40)
+
+if __name__ == "__main__":
+    run_test()

test_quota.py

@@ -0,0 +1,19 @@
+import os
+from google import genai
+from dotenv import load_dotenv
+
+load_dotenv()
+client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+
+models_to_test = ['gemini-flash-latest', 'gemini-pro-latest', 'gemini-1.5-flash-8b', 'gemini-1.5-flash']
+
+for model in models_to_test:
+    print(f"Testing model: {model}...")
+    try:
+        response = client.models.generate_content(
+            model=model,
+            contents="Say hi"
+        )
+        print(f"SUCCESS: {model} works! Response: {response.text}")
+    except Exception as e:
+        print(f"FAILED: {model} - Error: {e}")

tools/compliance.py

@@ -0,0 +1,17 @@
+def audit_privacy_policy(policy_text: str):
+    """
+    Analyzes a Privacy Policy against the Digital Personal Data Protection Act (DPDPA) 2023.
+    
+    Args:
+        policy_text: The full text of the privacy policy to audit.
+    """
+    # This tool will be called by the agent to perform a structured audit.
+    # The agent will use its internal knowledge of DPDPA 2023 to identify gaps.
+    return (
+        "DPDPA 2023 AUDIT REPORT\n"
+        "-----------------------\n"
+        "1. NOTICE (Section 5): Analyzing if purpose and data types are specified...\n"
+        "2. CONSENT (Section 6): Checking for 'Clear and Affirmative' consent language...\n"
+        "3. GRIEVANCE REDRESSAL (Section 12): Searching for Grievance Officer contact details...\n\n"
+        "Please provide the policy text, and I will highlight the specific compliance gaps."
+    )

tools/ingest.py

@@ -0,0 +1,66 @@
+import os
+import chromadb
+from chromadb.utils import embedding_functions
+from pypdf import PdfReader
+from dotenv import load_dotenv
+
+load_dotenv(override=True)
+
+# Configuration
+CHROMA_PATH = "data/chroma_db"
+COLLECTION_NAME = "nyaya_legal_docs"
+
+# Initialize ChromaDB
+client = chromadb.PersistentClient(path=CHROMA_PATH)
+
+# Use Gemini embeddings via a custom wrapper or OpenAI-compatible one if available
+# For simplicity in this workshop, we'll use a default sentence-transformer if available, 
+# or a simple placeholder. 
+# Better yet: Use Chroma's built-in default for now.
+embedding_func = embedding_functions.DefaultEmbeddingFunction()
+
+collection = client.get_or_create_collection(
+    name=COLLECTION_NAME,
+    embedding_function=embedding_func
+)
+
+def ingest_pdf(file_path):
+    """Reads a PDF, chunks it, and adds it to the vector database."""
+    print(f"Ingesting: {file_path}")
+    reader = PdfReader(file_path)
+    
+    doc_id = os.path.basename(file_path)
+    chunks = []
+    metadata = []
+    ids = []
+    
+    for i, page in enumerate(reader.pages):
+        text = page.extract_text()
+        if text.strip():
+            # Basic chunking by page for now
+            chunks.append(text)
+            metadata.append({"source": doc_id, "page": i + 1})
+            ids.append(f"{doc_id}_page_{i+1}")
+            
+    if chunks:
+        collection.add(
+            documents=chunks,
+            metadatas=metadata,
+            ids=ids
+        )
+        print(f"Successfully added {len(chunks)} pages from {doc_id}")
+
+def search_docs(query, n_results=3):
+    """Searches the collection for relevant snippets."""
+    results = collection.query(
+        query_texts=[query],
+        n_results=n_results
+    )
+    return results
+
+if __name__ == "__main__":
+    # Test with any PDFs in the data folder
+    data_dir = "data"
+    for file in os.listdir(data_dir):
+        if file.endswith(".pdf"):
+            ingest_pdf(os.path.join(data_dir, file))

tools/law_bridge.py

@@ -0,0 +1,29 @@
+def ipc_to_bns_lookup(ipc_section: str):
+    """
+    Maps an Indian Penal Code (IPC) section to the corresponding Bharatiya Nyaya Sanhita (BNS) 2023 section.
+    
+    Args:
+        ipc_section: The IPC section number (e.g., '302', '420').
+    """
+    # Key Mappings (Sample for MVP)
+    mappings = {
+        "302": {"bns": "101", "name": "Punishment for Murder", "change": "Punishment remains life/death; restructuring of clauses for clarity."},
+        "420": {"bns": "318", "name": "Cheating and dishonestly inducing delivery of property", "change": "Merged with other cheating sections; broader definition of 'property'."},
+        "498A": {"bns": "85/86", "name": "Cruelty by husband or relatives", "change": "Split into two sections; Section 85 defines cruelty, Section 86 defines punishment."},
+        "376": {"bns": "64-70", "name": "Punishment for Rape", "change": "Extensively expanded; separate sections for aggravated cases and gang rape."},
+        "124A": {"bns": "152", "name": "Sedition (now 'Acts endangering sovereignty')", "change": "The word 'Sedition' is removed; focuses on acts against the state/sovereignty."}
+    }
+    
+    clean_section = ipc_section.upper().replace("IPC", "").strip()
+    
+    if clean_section in mappings:
+        m = mappings[clean_section]
+        return (
+            f"--- BNS-IPC CROSS-REFERENCE ---\n"
+            f"Old Section: IPC {clean_section} ({m['name']})\n"
+            f"New Section: BNS {m['bns']}\n"
+            f"Key Change: {m['change']}\n"
+            f"--------------------------------"
+        )
+    else:
+        return f"Section IPC {clean_section} not found in the immediate lookup table. Please consult the full BNS Schedule or ask me to search for the specific offense name."

tools/legal_tools.py

@@ -0,0 +1,52 @@
+def draft_rti_query(subject: str, department: str = "Relevant Public Authority"):
+    """
+    Drafts a formal RTI (Right to Information) query for a given subject.
+    """
+    template = (
+        f"Subject: Application under RTI Act, 2005 - Regarding {subject}\n\n"
+        f"To,\nThe Public Information Officer,\n{department}\n\n"
+        f"Sir/Madam,\n"
+        f"I request you to provide the following information under the RTI Act, 2005:\n"
+        f"1. Detailed records/notifications regarding {subject}.\n"
+        f"2. Any internal circulars or guidelines issued concerning this matter.\n"
+        f"3. Time period of the requested data: [Specify Period].\n\n"
+        f"I am a citizen of India. I have attached the prescribed fee of Rs. 10/-\n\n"
+        f"Regards,\n[Your Name]"
+    )
+    return template
+
+def draft_legal_notice(sender: str, receiver: str, grievance: str, relief: str):
+    """
+    Prepares a formal legal notice for a specific grievance.
+    
+    Args:
+        sender: Name and address of the sender.
+        receiver: Name and address of the receiver.
+        grievance: Detailed description of the legal grievance/dispute.
+        relief: The action or compensation demanded from the receiver.
+    """
+    notice = (
+        f"LEGAL NOTICE\n\n"
+        f"From: {sender}\n"
+        f"To: {receiver}\n\n"
+        f"Under instructions from my client {sender}, I hereby serve you with this legal notice:\n\n"
+        f"1. THE GRIEVANCE: {grievance}\n"
+        f"2. LEGAL VIOLATION: This act is a violation of relevant Indian laws.\n"
+        f"3. THE RELIEF: You are hereby called upon to {relief} within 15 days of receipt of this notice.\n\n"
+        f"Failing which, my client shall be constrained to initiate civil/criminal proceedings against you.\n\n"
+        f"Regards,\n[Advocate Name]"
+    )
+    return notice
+
+def analyze_case_summary(case_text: str):
+    """
+    Analyzes a legal case and provides a structured summary (Facts, Issues, Ratio).
+    
+    Args:
+        case_text: The full text or excerpt of the judgment to analyze.
+    """
+    return "Case Analysis Placeholder: I will provide Facts, Issues, Arguments, and the Ratio Decidendi here."
+
+def suggest_sc_followup():
+    """Provides a suggestion to check the latest Supreme Court filings."""
+    return "Would you like me to check the latest Supreme Court filings or judgments related to this matter?"

tools/search_tool.py

@@ -0,0 +1,79 @@
+import os
+import chromadb
+from chromadb.utils import embedding_functions
+
+# Configuration
+CHROMA_PATH = "data/chroma_db"
+COLLECTION_NAME = "nyaya_legal_docs"
+
+def get_vector_collection():
+    client = chromadb.PersistentClient(path=CHROMA_PATH)
+    embedding_func = embedding_functions.DefaultEmbeddingFunction()
+    return client.get_or_create_collection(
+        name=COLLECTION_NAME,
+        embedding_function=embedding_func
+    )
+
+def search_legal_documents(query: str):
+    """
+    Search across the indexed legal documents (Gazettes, Circulars, etc.) using semantic search.
+    
+    Args:
+        query: The natural language query to search for in local documents.
+    """
+    try:
+        collection = get_vector_collection()
+        results = collection.query(
+            query_texts=[query],
+            n_results=3
+        )
+        if results['documents']:
+            return "\n\n".join(results['documents'][0])
+        return "No relevant local documents found."
+    except Exception as e:
+        return f"Error searching local documents: {e}"
+
+import os
+import requests
+
+def search_indian_kanoon(query: str):
+    """
+    Search for legal cases, acts, and statutes on Indian Kanoon (Online).
+    """
+    token = os.getenv("INDIANKANOON_API_KEY")
+    if not token:
+        return f"Search result placeholder for: {query}. (API Token missing in .env. Using internal model knowledge.)"
+
+    try:
+        headers = {
+            "Authorization": f"Token {token}",
+            "Accept": "application/json"
+        }
+        params = {"formInput": query, "pagenum": 0}
+        # Searching for the query - Indian Kanoon requires POST with params in the URL
+        response = requests.post(
+            f"https://api.indiankanoon.org/search/", 
+            headers=headers, 
+            params=params
+        )
+        data = response.json()
+        
+        if "docs" in data:
+            results = data["docs"][:3]
+            summary = "Found these cases on Indian Kanoon:\n"
+            for res in results:
+                summary += f"- {res['title']} (Source: {res.get('docsource', 'Unknown')}, ID: {res['tid']})\n"
+            return summary
+        return "No results found on Indian Kanoon."
+    except Exception as e:
+        return f"Error connecting to Indian Kanoon: {e}"
+
+def get_act_details(act_name: str, section: str = None):
+    """
+    Retrieve details of a specific Indian Act or Section.
+    
+    Args:
+        act_name: Name of the Act (e.g., 'IPC', 'DPDPA').
+        section: Specific section number (optional).
+    """
+    return f"Details for {act_name} Section {section if section else 'All'}. (Data retrieval pending)"