File size: 21,029 Bytes
6c914fc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
"""
ProVerBs Legal AI - Integrated Landing Page with AI Legal Chatbot
Incorporates your specialized AI Legal Chatbot with multiple modes
"""

import gradio as gr
from huggingface_hub import InferenceClient
import json
import os
from datetime import datetime
from typing import Dict, List, Optional

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"
        }
        
        # Application features from your original chatbot
        self.app_features = {
            "Legal Action Advisor": {
                "description": "Get personalized recommendations for seeking justice and remedy",
                "features": ["AI-powered action recommendations", "Case type analysis", "Timeline planning"]
            },
            "Document Analysis": {
                "description": "Upload and analyze legal documents with AI-powered insights",
                "features": ["OCR text extraction", "Multi-format support", "AI analysis"]
            },
            "Legal Research": {
                "description": "Comprehensive legal research tools and databases",
                "features": ["Multiple legal databases", "Case law research", "Statutory analysis"]
            },
            "Communications": {
                "description": "Integrated communication system with SMS, email, and phone",
                "features": ["Twilio integration", "Legal alerts", "Case reminders"]
            },
            "Document Generation": {
                "description": "AI-powered legal document creation and templates",
                "features": ["Smart templates", "Custom document builder", "Legal form generation"]
            }
        }
    
    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()
        
        # Feature recommendations based on query
        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()
    
    # Get mode-specific system prompt
    system_message = chatbot_integration.get_mode_system_prompt(mode)
    
    # Check for navigation-specific responses
    if mode == "navigation":
        nav_response = chatbot_integration.format_navigation_response(message)
        if nav_response:
            yield nav_response
            return
    
    # Use HF Inference API
    token = hf_token.token if hf_token else None
    client = InferenceClient(token=token, model="meta-llama/Llama-3.3-70B-Instruct")
    
    # Prepare messages
    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})
    
    # Stream response
    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
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;
}

.header-section h1 {
    font-size: 3rem;
    margin-bottom: 10px;
    font-weight: 700;
}

.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;
}
"""

# Create the main application
with gr.Blocks(
    theme=gr.themes.Soft(
        primary_hue="purple",
        secondary_hue="blue",
    ),
    css=custom_css,
    title="ProVerBs Legal AI Platform"
) as demo:
    
    # Header
    gr.HTML("""
    <div class="header-section">
        <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>
    """)
    
    # Login Section
    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 (Your specialized 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
            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("---")
            
            # Chat interface
            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),
                type="messages",
                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
            - **General Mode** answers broad legal questions
            - 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>
                    """, elem_classes=["feature-card"])
                
                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>
                    """, elem_classes=["feature-card"])
            
            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>
                    """, elem_classes=["feature-card"])
                
                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>
                    """, elem_classes=["feature-card"])
            
            gr.Markdown("""
            ---
            
            ### πŸ€– AI Assistant Modes
            
            Our chatbot includes **7 specialized modes**:
            
            1. **Navigation Guide** - Find features in the platform
            2. **General Assistant** - Broad legal questions
            3. **Document Validator** - Analyze document quality
            4. **Research Assistant** - Find legal precedents
            5. **Etymology Expert** - Learn term origins
            6. **Case Manager** - Organize your cases
            7. **Regulatory Monitor** - Stay updated on changes
            """)
        
        # 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:
            
            - **Navigation Guide** - Helps you find and use platform features
            - **General Legal Assistant** - Answers broad legal questions
            - **Document Validator** - Analyzes legal documents for completeness
            - **Legal Research Assistant** - Finds relevant case law and statutes
            - **Etymology Expert** - Explains origins of legal terms
            - **Case Management Helper** - Assists with organizing cases
            - **Regulatory Update Monitor** - Keeps you informed of changes
            
            ### πŸ‘₯ 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
            
            ### πŸ“š Resources
            
            - [Documentation](https://huggingface.co/Solomon7890)
            - [GitHub](https://github.com/Solomon7890)
            - [Community](#)
            - [Support](#)
            
            ### ⚠️ 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> | 
            <a href="#" target="_blank">Documentation</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
    )