Deploy ProVerBs LaW with logo support

app.py

@@ -1,866 +1,538 @@
-"""
-ProVerBs Legal AI - Complete AI Integration
-Features: 7 AI Modes + Rotating Logos + DeepSeek-OCR + ERNIE-4.5-VL + GDPVAL
-"""
-
-import gradio as gr
-from huggingface_hub import InferenceClient
-import json
-import os
-from datetime import datetime
-from typing import Dict, List, Optional
-import base64
-from pathlib import Path
-
-# AI Model Integration
-try:
-    from transformers import pipeline, AutoModel
-    from datasets import load_dataset
-    AI_MODELS_AVAILABLE = True
-except ImportError:
-    AI_MODELS_AVAILABLE = False
-    print("⚠️ AI models not available. Install transformers and datasets.")
-
-class EnhancedAILegalChatbot:
-    """
-    Enhanced AI Legal Chatbot with multiple AI models
-    - DeepSeek-OCR for document text extraction
-    - ERNIE-4.5-VL for vision-language understanding
-    - OpenAI GDPVAL dataset for legal knowledge
-    """
-    
-    def __init__(self):
-        self.specialized_modes = {
-            "navigation": "Application Navigation Guide",
-            "general": "General Legal Assistant",
-            "document_validation": "Document Validator with OCR & Vision AI",
-            "legal_research": "Legal Research Assistant with GDPVAL",
-            "etymology": "Legal Etymology Lookup",
-            "case_management": "Case Management Helper",
-            "regulatory_updates": "Regulatory Update Monitor"
-        }
-        
-        # Initialize AI models
-        self.ocr_pipeline = None
-        self.vision_pipeline = None
-        self.gdpval_dataset = None
-        self.minimax_pipeline = None
-        
-        if AI_MODELS_AVAILABLE:
-            self._initialize_models()
-    
-    def _initialize_models(self):
-        """Initialize all AI models"""
-        
-        # DeepSeek-OCR for text extraction
-        try:
-            print("📦 Loading DeepSeek-OCR...")
-            self.ocr_pipeline = pipeline(
-                "image-text-to-text", 
-                model="deepseek-ai/DeepSeek-OCR", 
-                trust_remote_code=True
-            )
-            print("✅ DeepSeek-OCR loaded!")
-        except Exception as e:
-            print(f"⚠️ DeepSeek-OCR not loaded: {e}")
-        
-        # ERNIE-4.5-VL for vision-language understanding
-        try:
-            print("📦 Loading ERNIE-4.5-VL (this may take a while)...")
-            self.vision_pipeline = pipeline(
-                "image-text-to-text",
-                model="baidu/ERNIE-4.5-VL-28B-A3B-Thinking",
-                trust_remote_code=True
-            )
-            print("✅ ERNIE-4.5-VL loaded!")
-        except Exception as e:
-            print(f"⚠️ ERNIE-4.5-VL not loaded: {e}")
-        
-        # OpenAI GDPVAL dataset for legal knowledge
-        try:
-            print("📦 Loading OpenAI GDPVAL dataset...")
-            self.gdpval_dataset = load_dataset("openai/gdpval", split="train")
-            print(f"✅ GDPVAL loaded! {len(self.gdpval_dataset)} entries")
-        except Exception as e:
-            print(f"⚠️ GDPVAL not loaded: {e}")
-        
-        # MiniMax-M2 for advanced text generation
-        try:
-            print("📦 Loading MiniMax-M2...")
-            self.minimax_pipeline = pipeline(
-                "text-generation",
-                model="MiniMaxAI/MiniMax-M2",
-                trust_remote_code=True
-            )
-            print("✅ MiniMax-M2 loaded!")
-        except Exception as e:
-            print(f"⚠️ MiniMax-M2 not loaded: {e}")
-    
-    def process_with_ocr(self, image_path: str) -> str:
-        """Extract text using DeepSeek-OCR"""
-        if not self.ocr_pipeline:
-            return "❌ OCR not available"
-        
-        try:
-            result = self.ocr_pipeline(image_path)
-            return result[0]['generated_text'] if result else ""
-        except Exception as e:
-            return f"❌ OCR error: {str(e)}"
-    
-    def process_with_vision(self, image_url: str, question: str) -> str:
-        """Analyze image using ERNIE-4.5-VL"""
-        if not self.vision_pipeline:
-            return "❌ Vision AI not available"
-        
-        try:
-            messages = [
-                {
-                    "role": "user",
-                    "content": [
-                        {"type": "image", "url": image_url},
-                        {"type": "text", "text": question}
-                    ]
-                }
-            ]
-            result = self.vision_pipeline(text=messages)
-            return result
-        except Exception as e:
-            return f"❌ Vision AI error: {str(e)}"
-    
-    def search_gdpval(self, query: str, limit: int = 5) -> str:
-        """Search OpenAI GDPVAL dataset for legal knowledge"""
-        if not self.gdpval_dataset:
-            return "❌ GDPVAL dataset not available"
-        
-        try:
-            # Simple keyword search (can be enhanced with embeddings)
-            results = []
-            query_lower = query.lower()
-            
-            for i, entry in enumerate(self.gdpval_dataset):
-                if i >= 100:  # Limit search for performance
-                    break
-                
-                # Check if query keywords appear in entry
-                entry_text = str(entry).lower()
-                if any(word in entry_text for word in query_lower.split()):
-                    results.append(entry)
-                    if len(results) >= limit:
-                        break
-            
-            if results:
-                formatted = "### 📚 GDPVAL Knowledge Base Results:\n\n"
-                for i, result in enumerate(results, 1):
-                    formatted += f"**Result {i}:**\n{result}\n\n"
-                return formatted
-            else:
-                return "No relevant results found in GDPVAL dataset."
-        except Exception as e:
-            return f"❌ GDPVAL search error: {str(e)}"
-    
-    def analyze_legal_document(self, image_path: str) -> str:
-        """
-        Complete document analysis using multiple AI models:
-        1. DeepSeek-OCR for text extraction
-        2. ERNIE-4.5-VL for visual understanding
-        3. Legal analysis with AI
-        """
-        analysis = "## 📄 Complete Legal Document Analysis\n\n"
-        
-        # Step 1: OCR Text Extraction
-        analysis += "### Step 1: Text Extraction (DeepSeek-OCR)\n"
-        extracted_text = self.process_with_ocr(image_path)
-        analysis += f"```\n{extracted_text[:500]}...\n```\n\n"
-        
-        # Step 2: Vision Analysis
-        analysis += "### Step 2: Visual Document Understanding (ERNIE-4.5-VL)\n"
-        vision_result = self.process_with_vision(
-            image_path,
-            "Analyze this legal document. Identify document type, key sections, and any notable elements."
-        )
-        analysis += f"{vision_result}\n\n"
-        
-        # Step 3: Legal Term Detection
-        analysis += "### Step 3: Legal Analysis\n"
-        legal_terms = self._detect_legal_terms(extracted_text)
-        analysis += f"**Legal Terms Found:** {legal_terms}\n\n"
-        
-        # Step 4: Document Quality Check
-        analysis += "### Step 4: Document Quality\n"
-        quality = self._assess_document_quality(extracted_text)
-        analysis += f"{quality}\n\n"
-        
-        return analysis
-    
-    def _detect_legal_terms(self, text: str) -> str:
-        """Detect legal terminology in text"""
-        legal_terms = [
-            'contract', 'agreement', 'party', 'clause', 'provision',
-            'whereas', 'hereby', 'herein', 'pursuant', 'consideration',
-            'liability', 'indemnify', 'warranty', 'breach', 'terminate',
-            'arbitration', 'jurisdiction', 'statute', 'regulation'
-        ]
-        
-        found = [term for term in legal_terms if term.lower() in text.lower()]
-        return ', '.join(found) if found else "None detected"
-    
-    def _assess_document_quality(self, text: str) -> str:
-        """Assess document completeness and quality"""
-        issues = []
-        
-        if len(text) < 100:
-            issues.append("⚠️ Document seems very short")
-        
-        if not any(term in text.lower() for term in ['party', 'parties', 'agreement']):
-            issues.append("⚠️ Missing party identification")
-        
-        if not any(term in text.lower() for term in ['date', 'dated', 'effective']):
-            issues.append("⚠️ No effective date found")
-        
-        if issues:
-            return "**Issues Found:**\n" + "\n".join(issues)
-        else:
-            return "✅ Document appears complete"
-    
-    def generate_with_minimax(self, messages: list) -> str:
-        """Generate response using MiniMax-M2"""
-        if not self.minimax_pipeline:
-            return "❌ MiniMax-M2 not available"
-        
-        try:
-            result = self.minimax_pipeline(messages)
-            return result[0]['generated_text'] if result else "No response"
-        except Exception as e:
-            return f"❌ MiniMax error: {str(e)}"
-    
-    def get_mode_system_prompt(self, mode: str) -> str:
-        """Get specialized system prompt based on mode"""
-        prompts = {
-            "navigation": """You are a ProVerBs Application Navigation Guide.
-
-**Enhanced Features:**
-- Document Analysis with OCR (DeepSeek-OCR)
-- Vision AI Analysis (ERNIE-4.5-VL)
-- Legal Knowledge Base (OpenAI GDPVAL)
-- 7 Specialized AI Modes
-
-Guide users to the right features.""",
-            
-            "general": """You are a General Legal Assistant for ProVerBs Legal AI Platform. 
-            
-**Available Tools:**
-- GDPVAL legal knowledge dataset
-- Vision AI for document understanding
-- OCR for text extraction
-
-Provide accurate legal information while noting you cannot provide legal advice.""",
-            
-            "document_validation": """You are an Advanced Document Validator.
-
-**AI-Powered Capabilities:**
-- **DeepSeek-OCR**: Extract text from scanned documents
-- **ERNIE-4.5-VL**: Understand document structure and layout visually
-- **Legal Analysis**: Validate completeness and legal terms
-
-**Process:**
-1. Extract text with OCR
-2. Analyze visually with ERNIE
-3. Check legal validity
-4. Provide detailed feedback""",
-            
-            "legal_research": """You are a Legal Research Assistant with GDPVAL access.
-
-**Enhanced Research Tools:**
-- OpenAI GDPVAL dataset for legal knowledge
-- Case law and precedent search
-- Statute and regulation analysis
-
-Provide comprehensive research with citations.""",
-            
-            "etymology": """You are a Legal Etymology Expert. Explain origins of legal terms.""",
-            
-            "case_management": """You are a Case Management Helper with AI document processing.""",
-            
-            "regulatory_updates": """You are a Regulatory Update Monitor."""
-        }
-        return prompts.get(mode, prompts["general"])
-
-def respond_with_mode(
-    message,
-    history: list,
-    mode: str,
-    model_choice: str,
-    max_tokens: int,
-    temperature: float,
-    top_p: float,
-):
-    """Generate AI response based on selected mode"""
-    chatbot = EnhancedAILegalChatbot()
-    
-    system_message = chatbot.get_mode_system_prompt(mode)
-    
-    # Enhanced responses for specific modes
-    if mode == "document_validation" and "analyze" in message.lower():
-        yield """
-## 📄 Advanced Document Validator
-
-**Multi-AI Analysis Available:**
-
-1. **DeepSeek-OCR**: Extract text from scanned documents
-2. **ERNIE-4.5-VL**: Understand document structure visually
-3. **Legal AI**: Validate and analyze content
-
-**Upload a document to get:**
-- ✅ Text extraction
-- ✅ Visual structure analysis
-- ✅ Legal term detection
-- ✅ Completeness check
-- ✅ Quality assessment
-"""
-        return
-    
-    if mode == "legal_research" and "search" in message.lower():
-        # Search GDPVAL dataset
-        gdpval_results = chatbot.search_gdpval(message)
-        yield gdpval_results
-        return
-    
-    # Choose AI model based on selection
-    if model_choice == "MiniMax-M2" and chatbot.minimax_pipeline:
-        # Use MiniMax-M2
-        try:
-            messages_minimax = [
-                {"role": "system", "content": system_message},
-                {"role": "user", "content": message}
-            ]
-            response = chatbot.generate_with_minimax(messages_minimax)
-            yield response
-        except Exception as e:
-            yield f"MiniMax Error: {str(e)}"
-    else:
-        # Use HF Inference API (Llama) as default
-        try:
-            client = InferenceClient(model="meta-llama/Llama-3.3-70B-Instruct")
-            
-            messages = [{"role": "system", "content": system_message}]
-            
-            for user_msg, assistant_msg in history:
-                if user_msg:
-                    messages.append({"role": "user", "content": user_msg})
-                if assistant_msg:
-                    messages.append({"role": "assistant", "content": assistant_msg})
-            
-            messages.append({"role": "user", "content": message})
-            
-            response = ""
-            for message_chunk in client.chat_completion(
-                messages,
-                max_tokens=max_tokens,
-                stream=True,
-                temperature=temperature,
-                top_p=top_p,
-            ):
-                if message_chunk.choices and message_chunk.choices[0].delta.content:
-                    token = message_chunk.choices[0].delta.content
-                    response += token
-                    yield response
-        except Exception as e:
-            # Provide demo response for local preview
-            demo_response = f"""
-## 🎯 Preview Mode Response
-
-**Your Question:** {message}
-
-**Mode:** {mode}
-
-**Note:** This is a demo response for local preview. The AI chat will work fully once deployed to Hugging Face Spaces with authentication.
-
-### What This Mode Does:
-
-{system_message[:200]}...
-
-### Features Available:
-- ✅ UI and navigation fully functional
-- ✅ All tabs working
-- ✅ Watermark backgrounds active
-- ✅ Logo rotation active
-- ⚠️ AI responses require HF deployment
-
-**To test AI chat:** Deploy to Hugging Face Spaces where authentication is automatic.
-
----
-
-**For now:** Test the visual design, tab navigation, dropdowns, and timing features!
-"""
-            yield demo_response
-
-
-# Custom CSS with watermark background logos
-custom_css = """
-.gradio-container {
-    max-width: 1200px !important;
-    position: relative;
-}
-
-/* Watermark background container */
-.watermark-container {
-    position: fixed;
-    top: 0;
-    left: 0;
-    width: 100%;
-    height: 100%;
-    z-index: 0;
-    pointer-events: none;
-    overflow: hidden;
-}
-
-/* Watermark logo styling */
-.watermark-logo {
-    position: absolute;
-    width: 300px;
-    height: 300px;
-    opacity: 0.08;
-    object-fit: contain;
-    transition: opacity 2s ease-in-out;
-}
-
-.watermark-logo.active {
-    opacity: 0.08;
-}
-
-.watermark-logo.hidden {
-    opacity: 0;
-}
-
-/* Position watermarks at different locations */
-.watermark-1 {
-    top: 10%;
-    left: 10%;
-    transform: rotate(-15deg);
-}
-
-.watermark-2 {
-    top: 15%;
-    right: 15%;
-    transform: rotate(20deg);
-}
-
-.watermark-3 {
-    bottom: 20%;
-    left: 15%;
-    transform: rotate(10deg);
-}
-
-.watermark-4 {
-    bottom: 15%;
-    right: 10%;
-    transform: rotate(-20deg);
-}
-
-.watermark-5 {
-    top: 50%;
-    left: 50%;
-    transform: translate(-50%, -50%) rotate(-5deg);
-}
-
-/* Ensure content is above watermarks */
-.gradio-container > * {
-    position: relative;
-    z-index: 1;
-}
-
-.header-section {
-    text-align: center;
-    padding: 40px 20px;
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-    color: white;
-    border-radius: 12px;
-    margin-bottom: 30px;
-    position: relative;
-    z-index: 2;
-}
-
-.logo-container {
-    margin-bottom: 20px;
-    display: flex;
-    justify-content: center;
-    align-items: center;
-}
-
-.rotating-logo {
-    width: 150px;
-    height: 150px;
-    border-radius: 50%;
-    object-fit: cover;
-    border: 4px solid rgba(255, 255, 255, 0.8);
-    box-shadow: 0 8px 32px rgba(0, 0, 0, 0.3);
-}
-
-.header-section h1 {
-    font-size: 3rem;
-    margin-bottom: 10px;
-    font-weight: 700;
-    text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.2);
-}
-
-/* Make sure tabs and content are above watermarks */
-.tabs, .tabitem {
-    position: relative;
-    z-index: 1;
-}
-"""
-
-# JavaScript for rotating logos in header and random watermarks
-rotating_logo_js = """
-<script>
-// Header logo rotation
-function rotateHeaderLogo() {
-    const logos = document.querySelectorAll('.rotating-logo');
-    let currentIndex = 0;
-    
-    function showNextLogo() {
-        logos.forEach((logo, index) => {
-            logo.style.display = 'none';
-        });
-        if (logos.length > 0) {
-            logos[currentIndex].style.display = 'block';
-            currentIndex = (currentIndex + 1) % logos.length;
-        }
-    }
-    
-    showNextLogo();
-    setInterval(showNextLogo, 60000);
-}
-
-// Random watermark rotation
-function rotateWatermarks() {
-    const watermarks = document.querySelectorAll('.watermark-logo');
-    const logoImages = [
-        'file/assets/logo_1.jpg',
-        'file/assets/logo_2.jpg',
-        'file/assets/logo_3.jpg'
-    ];
-    
-    function randomizeWatermarks() {
-        watermarks.forEach((watermark) => {
-            // Fade out
-            watermark.classList.remove('active');
-            watermark.classList.add('hidden');
-            
-            setTimeout(() => {
-                // Pick random logo
-                const randomLogo = logoImages[Math.floor(Math.random() * logoImages.length)];
-                watermark.src = randomLogo;
-                
-                // Fade in
-                watermark.classList.remove('hidden');
-                watermark.classList.add('active');
-            }, 2000);
-        });
-    }
-    
-    // Initial randomization
-    randomizeWatermarks();
-    
-    // Change watermarks every 30 seconds with random logos
-    setInterval(randomizeWatermarks, 30000);
-}
-
-// Initialize both rotations
-if (document.readyState === 'loading') {
-    document.addEventListener('DOMContentLoaded', function() {
-        rotateHeaderLogo();
-        rotateWatermarks();
-    });
-} else {
-    rotateHeaderLogo();
-    rotateWatermarks();
-}
-</script>
-"""
-
-# Create the main application
-demo = gr.Blocks(title="ProVerBs Legal AI Platform")
-
-with demo:
-    
-    # Watermark background logos
-    gr.HTML(f"""
-    <div class="watermark-container">
-        <img src="file/assets/logo_1.jpg" class="watermark-logo watermark-1 active" alt="Watermark 1">
-        <img src="file/assets/logo_2.jpg" class="watermark-logo watermark-2 active" alt="Watermark 2">
-        <img src="file/assets/logo_3.jpg" class="watermark-logo watermark-3 active" alt="Watermark 3">
-        <img src="file/assets/logo_1.jpg" class="watermark-logo watermark-4 active" alt="Watermark 4">
-        <img src="file/assets/logo_2.jpg" class="watermark-logo watermark-5 active" alt="Watermark 5">
-    </div>
-    
-    <style>{custom_css}</style>
-    {rotating_logo_js}
-    
-    <!-- Header with Rotating Logos -->
-    <div class="header-section">
-        <div class="logo-container">
-            <img src="file/assets/logo_1.jpg" class="rotating-logo" alt="Logo 1" style="display: block;">
-            <img src="file/assets/logo_2.jpg" class="rotating-logo" alt="Logo 2" style="display: none;">
-            <img src="file/assets/logo_3.jpg" class="rotating-logo" alt="Logo 3" style="display: none;">
-        </div>
-        <h1>⚖️ ProVerBs Legal AI Platform</h1>
-        <p>Lawful vs. Legal: Dual Analysis "Adappt'plication"</p>
-        <p style="font-size: 1rem; margin-top: 10px;">
-            Multi-AI Legal System | 5 AI Models Working Together 🚀
-        </p>
-    </div>
-    """)
-    
-    gr.Markdown("---")
-    
-    # Main Tabs
-    with gr.Tabs():
-        
-        # Tab 1: Welcome
-        with gr.Tab("🏠 Welcome"):
-            gr.Markdown("""
-            ## Welcome to ProVerBs Legal AI Platform
-            
-            ### 🤖 5 AI Models Integrated:
-            
-            1. **DeepSeek-OCR** - Extract text from scanned documents
-            2. **ERNIE-4.5-VL** - Advanced vision-language understanding
-            3. **OpenAI GDPVAL** - Legal knowledge dataset
-            4. **Meta Llama 3.3** - General AI assistance
-            5. **MiniMax-M2** - Advanced text generation
-            
-            ### ⚖️ 7 Specialized Modes:
-            
-            - 📍 Navigation Guide
-            - 💬 General Legal Assistant (with GDPVAL)
-            - 📄 Document Validator (OCR + Vision AI)
-            - 🔍 Legal Research (GDPVAL-powered)
-            - 📚 Etymology Expert
-            - 💼 Case Management
-            - 📋 Regulatory Updates
-            
-            ### ✨ Unique Features:
-            
-            - **Multi-AI Document Analysis**: Combines OCR + Vision AI
-            - **Legal Knowledge Base**: Access to GDPVAL dataset
-            - **Rotating Custom Logos**: Your professional branding
-            - **Complete Legal Solution**: All-in-one platform
-            """)
-        
-        # Tab 2: AI Legal Chatbot
-        with gr.Tab("🤖 AI Legal Chatbot"):
-            gr.Markdown("""
-            ## Multi-AI Legal Chatbot
-            
-            **Powered by 5 AI Models** for comprehensive legal assistance!
-            
-            **Choose Your AI Model:**
-            - **Meta Llama 3.3** - Fast, efficient, streaming responses
-            - **MiniMax-M2** - Advanced text generation capabilities
-            """)
-            
-            mode_selector = gr.Dropdown(
-                choices=[
-                    "navigation",
-                    "general",
-                    "document_validation",
-                    "legal_research",
-                    "etymology",
-                    "case_management",
-                    "regulatory_updates"
-                ],
-                value="navigation",
-                label="Select AI Assistant Mode"
-            )
-            
-            model_selector = gr.Dropdown(
-                choices=[
-                    "Meta Llama 3.3",
-                    "MiniMax-M2"
-                ],
-                value="Meta Llama 3.3",
-                label="Select AI Model"
-            )
-            
-            gr.Markdown("---")
-            
-            chatbot = gr.ChatInterface(
-                respond_with_mode,
-                chatbot=gr.Chatbot(height=500),
-                additional_inputs=[
-                    mode_selector,
-                    model_selector,
-                    gr.Slider(128, 4096, value=2048, label="Max Tokens"),
-                    gr.Slider(0.1, 2.0, value=0.7, label="Temperature"),
-                    gr.Slider(0.1, 1.0, value=0.95, label="Top-p"),
-                ],
-                examples=[
-                    ["How do I use the multi-AI document analysis?"],
-                    ["Search GDPVAL for contract law information"],
-                    ["Analyze a document with vision AI"],
-                ],
-            )
-        
-        # Tab 3: Document Analysis (NEW!)
-        with gr.Tab("📄 Document Analysis"):
-            gr.Markdown("""
-            ## 📄 Document Upload & AI Analysis
-            
-            Upload legal documents for comprehensive AI analysis!
-            """)
-            
-            with gr.Row():
-                with gr.Column(scale=1):
-                    doc_upload = gr.File(
-                        label="📎 Upload Document",
-                        file_types=[".pdf", ".jpg", ".jpeg", ".png", ".txt"]
-                    )
-                    
-                    analysis_type = gr.Radio(
-                        choices=[
-                            "Complete Analysis",
-                            "OCR Only",
-                            "Visual Only",
-                            "Legal Validation"
-                        ],
-                        value="Complete Analysis",
-                        label="Analysis Type"
-                    )
-                    
-                    analyze_btn = gr.Button("🔍 Analyze", variant="primary", size="lg")
-                
-                with gr.Column(scale=2):
-                    analysis_output = gr.Markdown("Upload a document to begin analysis.")
-            
-            def analyze_doc(file_path, analysis_type):
-                if not file_path:
-                    return "⚠️ Please upload a document first."
-                
-                filename = file_path.split('/')[-1] if '/' in file_path else file_path.split('\\')[-1]
-                
-                return f"""
-## 📄 Document Analysis Results
-
-**File:** {filename}  
-**Analysis Type:** {analysis_type}
-
-### ⚠️ Preview Mode
-
-**What happens in production:**
-- **DeepSeek-OCR** extracts text from images
-- **ERNIE-4.5-VL** analyzes document structure
-- **Legal AI** validates completeness
-
-**Demo Output:**
-- Document uploaded successfully ✅
-- File format supported ✅
-- Ready for AI analysis ✅
-
-**Note:** Full analysis available after HF deployment.
-
-**Try it:** Upload any legal document (PDF, image, text) to see the interface!
-"""
-            
-            analyze_btn.click(
-                fn=analyze_doc,
-                inputs=[doc_upload, analysis_type],
-                outputs=analysis_output
-            )
-        
-        # Tab 4: Features
-        with gr.Tab("✨ Features"):
-            gr.Markdown("""
-            ## Advanced AI Features
-            
-            ### 🤖 Integrated AI Models:
-            
-            1. **DeepSeek-OCR**
-               - Text extraction from images
-               - Document scanning
-               - High accuracy OCR
-            
-            2. **ERNIE-4.5-VL** (Baidu)
-               - Vision-language understanding
-               - Document layout analysis
-               - Visual question answering
-            
-            3. **OpenAI GDPVAL**
-               - Legal knowledge dataset
-               - Case law and precedents
-               - Regulatory information
-            
-            4. **Meta Llama 3.3**
-               - General AI assistance
-               - Legal reasoning
-               - Natural conversation
-            
-            5. **MiniMax-M2** ⭐ NEW!
-               - Advanced text generation
-               - High-quality responses
-               - Alternative AI model option
-            
-            ### 🎯 What This Means:
-            
-            - **Most Advanced**: 5 AI models working together
-            - **Best Analysis**: Multiple AI perspectives  
-            - **Model Choice**: Select the best AI for your task
-            - **Comprehensive**: Text + Vision + Knowledge
-            - **Professional**: Your custom branding with logos
-            """)
-        
-        # Tab 4: About
-        with gr.Tab("ℹ️ About"):
-            gr.Markdown("""
-            ## About ProVerBs Legal AI - Complete Edition
-            
-            ### 🚀 Version 2.1.0 - Complete Multi-AI Integration
-            
-            **This is the most advanced version** with all AI capabilities:
-            
-            - ✅ 7 Specialized AI Modes
-            - ✅ 3 Rotating Custom Logos
-            - ✅ DeepSeek-OCR Integration
-            - ✅ ERNIE-4.5-VL Vision AI
-            - ✅ OpenAI GDPVAL Dataset
-            - ✅ Meta Llama 3.3 AI
-            - ✅ MiniMax-M2 AI ⭐ NEW!
-            
-            ### 📊 Technical Stack:
-            
-            - **Frontend**: Gradio 4.x
-            - **OCR**: DeepSeek-OCR
-            - **Vision**: ERNIE-4.5-VL-28B
-            - **Dataset**: OpenAI GDPVAL
-            - **LLM 1**: Meta Llama 3.3-70B
-            - **LLM 2**: MiniMax-M2
-            - **Platform**: Hugging Face Spaces
-            
-            ### ⚠️ Disclaimer:
-            
-            This platform provides general legal information only. 
-            Always consult with a qualified attorney for specific legal matters.
-            
-            ---
-            
-            **Version 2.1.0** | Built by Solomon7890 | 5-Model AI Legal Platform
-            """)
-    
-    # Footer
-    gr.Markdown("""
-    ---
-    <div style="text-align: center; padding: 20px;">
-        <p><strong>⚖️ ProVerBs Legal AI Platform</strong> | Version 2.1.0 - 5-Model AI Edition</p>
-        <p>Powered by: DeepSeek-OCR • ERNIE-4.5-VL • GDPVAL • Llama 3.3 • MiniMax-M2</p>
-        <p>© 2024 ProVerBs Legal AI</p>
-    </div>
-    """)
-
-if __name__ == "__main__":
-    demo.queue(max_size=20)
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        show_error=True
-    )
+"""
+ProVerBs Legal AI - Integrated Landing Page with Rotating Logos
+Features your custom logos with 60-second rotation
+"""
+
+import gradio as gr
+from huggingface_hub import InferenceClient
+import json
+import os
+from datetime import datetime
+from typing import Dict, List, Optional
+import base64
+from pathlib import Path
+
+class AILegalChatbotIntegration:
+    """
+    Integration of your AI Legal Chatbot into Gradio
+    Supports all specialized modes from your original chatbot
+    """
+    
+    def __init__(self):
+        self.specialized_modes = {
+            "navigation": "Application Navigation Guide",
+            "general": "General Legal Assistant",
+            "document_validation": "Document Validator",
+            "legal_research": "Legal Research Assistant",
+            "etymology": "Legal Etymology Lookup",
+            "case_management": "Case Management Helper",
+            "regulatory_updates": "Regulatory Update Monitor"
+        }
+    
+    def get_mode_system_prompt(self, mode: str) -> str:
+        """Get specialized system prompt based on mode"""
+        prompts = {
+            "navigation": """You are a ProVerBs Application Navigation Guide. Help users navigate the application's features:
+            
+**Available Features:**
+- Legal Action Advisor: Get recommendations for seeking justice
+- Document Analysis: Upload and analyze legal documents
+- Legal Research: Access comprehensive legal databases
+- Communications: SMS, email, and phone integration
+- Document Generation: Create legal documents with AI
+
+Guide users to the right features and explain how to use them effectively.""",
+            
+            "general": """You are a General Legal Assistant for ProVerBs Legal AI Platform. Provide accurate legal information while noting that you cannot provide legal advice. Always recommend consulting with a licensed attorney for specific legal matters. Be professional, thorough, and cite relevant legal principles when possible.""",
+            
+            "document_validation": """You are a Document Validator. Analyze legal documents for:
+- Completeness and required elements
+- Legal terminology accuracy
+- Structural integrity
+- Common issues and red flags
+Provide specific feedback on document quality and validity.""",
+            
+            "legal_research": """You are a Legal Research Assistant. Help users:
+- Find relevant case law and precedents
+- Understand statutes and regulations
+- Research legal principles and concepts
+- Cite authoritative legal sources
+Provide comprehensive research guidance.""",
+            
+            "etymology": """You are a Legal Etymology Expert. Explain the origins and meanings of legal terms:
+- Latin and historical roots
+- Evolution of legal terminology
+- Modern usage and interpretation
+- Related legal concepts
+Make legal language accessible and understandable.""",
+            
+            "case_management": """You are a Case Management Helper. Assist with:
+- Organizing case information
+- Tracking deadlines and milestones
+- Managing documents and evidence
+- Coordinating case activities
+Provide practical case management advice.""",
+            
+            "regulatory_updates": """You are a Regulatory Update Monitor. Keep users informed about:
+- Recent legal and regulatory changes
+- Industry-specific compliance updates
+- Important legislative developments
+- Impact analysis of new regulations
+Provide timely and relevant regulatory information."""
+        }
+        return prompts.get(mode, prompts["general"])
+    
+    def format_navigation_response(self, query: str) -> str:
+        """Format response for navigation queries"""
+        query_lower = query.lower()
+        
+        recommendations = []
+        
+        if any(word in query_lower for word in ["document", "contract", "agreement", "analyze"]):
+            recommendations.append("📄 **Document Analysis** - Upload and analyze your documents")
+        
+        if any(word in query_lower for word in ["research", "case", "law", "statute"]):
+            recommendations.append("🔍 **Legal Research** - Access comprehensive legal databases")
+        
+        if any(word in query_lower for word in ["action", "remedy", "justice", "sue"]):
+            recommendations.append("⚖️ **Legal Action Advisor** - Get recommendations for your situation")
+        
+        if any(word in query_lower for word in ["create", "generate", "template", "form"]):
+            recommendations.append("📝 **Document Generation** - Create legal documents with AI")
+        
+        if any(word in query_lower for word in ["communicate", "message", "sms", "email"]):
+            recommendations.append("📧 **Communications** - Integrated messaging system")
+        
+        if recommendations:
+            return "### I can help you with these features:\n\n" + "\n".join(recommendations) + "\n\n**What would you like to explore?**"
+        
+        return None
+
+def respond_with_mode(
+    message,
+    history: list,
+    mode: str,
+    max_tokens: int,
+    temperature: float,
+    top_p: float,
+    hf_token: gr.OAuthToken | None = None
+):
+    """Generate AI response based on selected mode"""
+    chatbot_integration = AILegalChatbotIntegration()
+    
+    system_message = chatbot_integration.get_mode_system_prompt(mode)
+    
+    if mode == "navigation":
+        nav_response = chatbot_integration.format_navigation_response(message)
+        if nav_response:
+            yield nav_response
+            return
+    
+    token = hf_token.token if hf_token else None
+    client = InferenceClient(token=token, model="meta-llama/Llama-3.3-70B-Instruct")
+    
+    messages = [{"role": "system", "content": system_message}]
+    
+    for user_msg, assistant_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if assistant_msg:
+            messages.append({"role": "assistant", "content": assistant_msg})
+    
+    messages.append({"role": "user", "content": message})
+    
+    response = ""
+    try:
+        for message_chunk in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            if message_chunk.choices and message_chunk.choices[0].delta.content:
+                token = message_chunk.choices[0].delta.content
+                response += token
+                yield response
+    except Exception as e:
+        yield f"Error: {str(e)}"
+
+
+# Custom CSS with rotating logo animation
+custom_css = """
+.gradio-container {
+    max-width: 1200px !important;
+}
+
+.header-section {
+    text-align: center;
+    padding: 40px 20px;
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    color: white;
+    border-radius: 12px;
+    margin-bottom: 30px;
+    position: relative;
+}
+
+.logo-container {
+    margin-bottom: 20px;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+}
+
+.rotating-logo {
+    width: 150px;
+    height: 150px;
+    border-radius: 50%;
+    object-fit: cover;
+    border: 4px solid rgba(255, 255, 255, 0.8);
+    box-shadow: 0 8px 32px rgba(0, 0, 0, 0.3);
+    animation: fadeInOut 60s infinite;
+}
+
+@keyframes fadeInOut {
+    0%, 20% { opacity: 1; }
+    25%, 45% { opacity: 0; }
+    50%, 70% { opacity: 1; }
+    75%, 95% { opacity: 0; }
+    100% { opacity: 1; }
+}
+
+.logo-1 { animation-delay: 0s; }
+.logo-2 { animation-delay: 20s; }
+.logo-3 { animation-delay: 40s; }
+
+.header-section h1 {
+    font-size: 3rem;
+    margin-bottom: 10px;
+    font-weight: 700;
+    text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.2);
+}
+
+.mode-selector {
+    font-size: 1.1rem !important;
+    font-weight: 600 !important;
+    padding: 12px !important;
+}
+
+.tab-nav button {
+    font-size: 16px;
+    font-weight: 600;
+}
+
+.feature-card {
+    border: 2px solid #e0e0e0;
+    border-radius: 12px;
+    padding: 20px;
+    margin: 10px;
+    background: #f8f9fa;
+    transition: all 0.3s;
+}
+
+.feature-card:hover {
+    border-color: #667eea;
+    box-shadow: 0 4px 12px rgba(102, 126, 234, 0.3);
+    transform: translateY(-2px);
+}
+"""
+
+# JavaScript for rotating logos
+rotating_logo_js = """
+<script>
+function rotateLogo() {
+    const logos = document.querySelectorAll('.rotating-logo');
+    let currentIndex = 0;
+    
+    function showNextLogo() {
+        logos.forEach((logo, index) => {
+            logo.style.display = 'none';
+        });
+        
+        logos[currentIndex].style.display = 'block';
+        
+        currentIndex = (currentIndex + 1) % logos.length;
+    }
+    
+    // Show first logo initially
+    showNextLogo();
+    
+    // Rotate every 60 seconds
+    setInterval(showNextLogo, 60000);
+}
+
+// Start rotation when page loads
+if (document.readyState === 'loading') {
+    document.addEventListener('DOMContentLoaded', rotateLogo);
+} else {
+    rotateLogo();
+}
+</script>
+"""
+
+# Create the main application
+demo = gr.Blocks(title="ProVerBs Legal AI Platform")
+demo.css = custom_css
+
+with demo:
+    
+    # Header with Rotating Logos
+    gr.HTML(f"""
+    <div class="header-section">
+        <div class="logo-container">
+            <img src="file/assets/logo_1.jpg" class="rotating-logo logo-1" alt="ProVerBs Logo 1" style="display: block;">
+            <img src="file/assets/logo_2.jpg" class="rotating-logo logo-2" alt="ProVerBs Logo 2" style="display: none;">
+            <img src="file/assets/logo_3.jpg" class="rotating-logo logo-3" alt="ProVerBs Logo 3" style="display: none;">
+        </div>
+        <h1>⚖️ ProVerBs Legal AI Platform</h1>
+        <p>Lawful vs. Legal: Dual Analysis "Adappt'plication"</p>
+        <p style="font-size: 1rem; margin-top: 10px;">
+            Professional Legal AI System | Multi-Module Platform | Powered by Advanced AI
+        </p>
+    </div>
+    {rotating_logo_js}
+    """)
+    
+    # Login Section (commented out for local preview)
+    # with gr.Row():
+    #     with gr.Column(scale=1):
+    #         gr.LoginButton(size="lg")
+    #     with gr.Column(scale=5):
+    #         gr.Markdown("👈 **Login with your Hugging Face account** for full access")
+    
+    gr.Markdown("---")
+    
+    # Main Tabs
+    with gr.Tabs() as tabs:
+        
+        # Tab 1: Welcome
+        with gr.Tab("🏠 Welcome", id="welcome"):
+            gr.Markdown("""
+            ## Welcome to ProVerBs Legal AI Platform
+            
+            A comprehensive legal AI system with **multiple specialized assistants** to help you navigate legal matters.
+            
+            ### 🎯 Choose Your AI Assistant Mode
+            
+            Our platform features **7 specialized AI modes** to serve your specific needs:
+            
+            - **📍 Navigation Guide** - Help finding features in the platform
+            - **💬 General Legal Assistant** - Broad legal questions and guidance
+            - **📄 Document Validator** - Analyze and validate legal documents
+            - **🔍 Legal Research** - Case law and statutory research
+            - **📚 Etymology Expert** - Understanding legal terminology origins
+            - **💼 Case Management** - Organizing and tracking legal cases
+            - **📋 Regulatory Updates** - Stay informed about legal changes
+            
+            ### ⚖️ Platform Features
+            
+            - **Legal Action Advisor** - Get personalized recommendations
+            - **Document Analysis** - AI-powered document processing
+            - **Legal Research Tools** - Comprehensive databases
+            - **Communications** - Integrated SMS, email, phone
+            - **Document Generation** - Create legal documents with AI
+            
+            **Ready to start?** Click the "AI Legal Chatbot" tab to begin!
+            """)
+        
+        # Tab 2: AI Legal Chatbot
+        with gr.Tab("🤖 AI Legal Chatbot", id="chatbot"):
+            gr.Markdown("""
+            ## AI Legal Chatbot - Multiple Specialized Modes
+            
+            Select your assistant mode below and start chatting!
+            """)
+            
+            mode_selector = gr.Dropdown(
+                choices=list({
+                    "navigation": "📍 Navigation Guide - Find features in the app",
+                    "general": "💬 General Legal Assistant - Broad legal questions",
+                    "document_validation": "📄 Document Validator - Analyze documents",
+                    "legal_research": "🔍 Legal Research - Case law & statutes",
+                    "etymology": "📚 Etymology Expert - Legal term origins",
+                    "case_management": "💼 Case Management - Organize cases",
+                    "regulatory_updates": "📋 Regulatory Updates - Legal changes"
+                }.items()),
+                value="navigation",
+                label="Select AI Assistant Mode",
+                elem_classes=["mode-selector"]
+            )
+            
+            gr.Markdown("---")
+            
+            chatbot = gr.ChatInterface(
+                lambda message, history, mode, max_tokens, temperature, top_p, hf_token: 
+                    respond_with_mode(message, history, mode.split(":")[0], max_tokens, temperature, top_p, hf_token),
+                chatbot=gr.Chatbot(
+                    height=500,
+                    placeholder="💬 Select a mode above and ask your question...",
+                    show_label=False,
+                ),
+                textbox=gr.Textbox(
+                    placeholder="Type your question here...",
+                    container=False,
+                    scale=7
+                ),
+                additional_inputs=[
+                    mode_selector,
+                    gr.Slider(128, 4096, value=2048, step=128, label="Max Tokens"),
+                    gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature"),
+                    gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p"),
+                ],
+                examples=[
+                    ["How do I navigate to the document analysis feature?"],
+                    ["What is the difference between lawful and legal?"],
+                    ["Can you help me validate a contract?"],
+                    ["I need to research case law about contracts"],
+                    ["What does 'habeas corpus' mean?"],
+                    ["How do I organize my legal case documents?"],
+                    ["What are the latest regulatory changes in business law?"],
+                ],
+                cache_examples=False,
+            )
+            
+            gr.Markdown("""
+            ### 💡 Tips for Best Results
+            
+            - **Choose the right mode** for your question type
+            - **Be specific** with your questions
+            - **Navigation Mode** helps you find features in the app
+            - Each mode is specialized for different tasks!
+            """)
+        
+        # Tab 3: Features Overview
+        with gr.Tab("✨ Features", id="features"):
+            gr.Markdown("""
+            ## Platform Features
+            
+            ### 🎯 Core Capabilities
+            """)
+            
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("""
+                    <div class="feature-card">
+                    
+                    #### ⚖️ Legal Action Advisor
+                    Get personalized recommendations for seeking justice and remedy
+                    
+                    - AI-powered action recommendations
+                    - Case type analysis
+                    - Timeline planning
+                    - Free resource finder
+                    - Evidence collection guidance
+                    </div>
+                    """)
+                
+                with gr.Column():
+                    gr.Markdown("""
+                    <div class="feature-card">
+                    
+                    #### 📄 Document Analysis
+                    Upload and analyze legal documents with AI insights
+                    
+                    - OCR text extraction
+                    - Multi-format support (PDF, DOCX, TXT)
+                    - AI-powered analysis
+                    - Legal database cross-reference
+                    - Document validation
+                    </div>
+                    """)
+            
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("""
+                    <div class="feature-card">
+                    
+                    #### 🔍 Legal Research
+                    Comprehensive legal research tools and databases
+                    
+                    - Multiple legal databases
+                    - Case law research
+                    - Statutory analysis
+                    - Historical documents
+                    - Citation tools
+                    </div>
+                    """)
+                
+                with gr.Column():
+                    gr.Markdown("""
+                    <div class="feature-card">
+                    
+                    #### 📧 Communications
+                    Integrated communication system
+                    
+                    - Twilio integration
+                    - SMS messaging
+                    - Email notifications
+                    - Phone capabilities
+                    - Legal alerts & reminders
+                    </div>
+                    """)
+        
+        # Tab 4: About
+        with gr.Tab("ℹ️ About", id="about"):
+            gr.Markdown("""
+            ## About ProVerBs Legal AI
+            
+            ### 🎓 Our Platform
+            
+            ProVerBs Legal AI combines advanced artificial intelligence with comprehensive legal knowledge
+            to provide accessible, accurate legal information and tools.
+            
+            ### 🤖 Specialized AI Chatbot
+            
+            Our AI chatbot features **7 specialized modes**, each trained for specific legal tasks.
+            
+            ### 👥 Who We Serve
+            
+            - **Legal Professionals** - Enhance your practice with AI
+            - **Law Students** - Research and study assistance
+            - **Businesses** - Understand legal implications
+            - **Individuals** - Learn about your legal rights
+            
+            ### 🔒 Privacy & Security
+            
+            - End-to-end encryption
+            - No storage without consent
+            - GDPR and CCPA compliant
+            - Secure OAuth authentication
+            
+            ### ⚠️ Important Disclaimer
+            
+            This platform provides general legal information only. It does not constitute legal advice.
+            Always consult with a qualified attorney for specific legal matters.
+            
+            ---
+            
+            **Version 1.0.0** | Built by Solomon7890 | Powered by Hugging Face
+            """)
+    
+    # Footer
+    gr.Markdown("""
+    ---
+    
+    <div style="text-align: center; padding: 20px; color: #666;">
+        <p><strong>⚖️ ProVerBs Legal AI Platform</strong> | Version 1.0.0</p>
+        <p>
+            <a href="https://huggingface.co/Solomon7890" target="_blank">Hugging Face</a> | 
+            <a href="https://github.com/Solomon7890" target="_blank">GitHub</a>
+        </p>
+        <p style="font-size: 0.9rem; margin-top: 10px;">
+            ⚠️ <strong>Disclaimer</strong>: This AI provides general legal information only. 
+            Consult with a licensed attorney for specific legal matters.
+        </p>
+        <p style="font-size: 0.85rem; color: #999;">
+            © 2024 ProVerBs Legal AI. Built with ❤️ for legal professionals worldwide.
+        </p>
+    </div>
+    """)
+
+if __name__ == "__main__":
+    demo.queue(max_size=20)
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_error=True
+    )