"""
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("""
""")
# 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("""
#### âī¸ 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
""", elem_classes=["feature-card"])
with gr.Column():
gr.Markdown("""
#### đ 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
""", elem_classes=["feature-card"])
with gr.Row():
with gr.Column():
gr.Markdown("""
#### đ Legal Research
Comprehensive legal research tools and databases
- Multiple legal databases
- Case law research
- Statutory analysis
- Historical documents
- Citation tools
""", elem_classes=["feature-card"])
with gr.Column():
gr.Markdown("""
#### đ§ Communications
Integrated communication system
- Twilio integration
- SMS messaging
- Email notifications
- Phone capabilities
- Legal alerts & reminders
""", 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("""
---
âī¸ ProVerBs Legal AI Platform | Version 1.0.0
Hugging Face |
GitHub |
Documentation
â ī¸ Disclaimer: This AI provides general legal information only.
Consult with a licensed attorney for specific legal matters.
Š 2024 ProVerBs Legal AI. Built with â¤ī¸ for legal professionals worldwide.
""")
if __name__ == "__main__":
demo.queue(max_size=20)
demo.launch(
server_name="0.0.0.0",
server_port=7860,
share=False,
show_error=True
)