Update setup.py

setup.py

@@ -1,776 +1,222 @@
 #!/usr/bin/env python3
 """
-Complete Setup Script for Iran Legal Dashboard
-==============================================
-Enhanced setup with dependency management, environment setup, and comprehensive initialization
+Startup Script for Iran Legal Dashboard
+======================================
+Ensures proper initialization and debugging
 """
 
-import sqlite3
-import json
 import os
 import sys
+import sqlite3
 import logging
-import subprocess
-import shutil
-from datetime import datetime
-from typing import Dict, List, Optional
 from pathlib import Path
 
-# Configure logging
+# Setup logging
 logging.basicConfig(
     level=logging.INFO,
-    format='%(asctime)s - %(levelname)s - %(message)s',
-    handlers=[
-        logging.FileHandler('setup.log'),
-        logging.StreamHandler(sys.stdout)
-    ]
+    format='%(asctime)s - %(levelname)s - %(message)s'
 )
 logger = logging.getLogger(__name__)
 
-class SetupManager:
-    """Complete setup manager for the Iran Legal Dashboard"""
+def check_environment():
+    """Check if running environment is properly configured"""
+    logger.info("🔍 Checking environment...")
+    
+    # Check current directory
+    cwd = os.getcwd()
+    logger.info(f"📁 Current working directory: {cwd}")
+    
+    # Check for required files
+    required_files = ['app.py', 'requirements.txt']
+    src_required_files = ['src/streamlit_app.py'] if os.path.exists('src') else []
+    
+    all_required = required_files + src_required_files
+    
+    missing_files = []
+    for file in all_required:
+        if not os.path.exists(file):
+            missing_files.append(file)
+    
+    if missing_files:
+        logger.error(f"❌ Missing required files: {missing_files}")
+        return False
+    
+    logger.info("✅ All required files found")
+    return True
+
+def setup_database():
+    """Setup database with proper permissions"""
+    logger.info("🗄️ Setting up database...")
     
-    def __init__(self):
-        self.project_root = Path.cwd()
-        self.data_dir = self.project_root / "data"
-        self.logs_dir = self.project_root / "logs"
-        self.temp_dir = self.project_root / "temp"
-        self.config_file = self.project_root / "config.json"
-        
-    def run_complete_setup(self):
-        """Run the complete setup process"""
-        logger.info("🚀 Starting Iran Legal Dashboard Complete Setup...")
-        
+    # Try different database paths
+    db_paths = [
+        "/tmp/iran_legal.db",
+        "./iran_legal.db",
+        "data/iran_legal.db"
+    ]
+    
+    for db_path in db_paths:
         try:
-            # Step 1: Check system requirements
-            if not self.check_system_requirements():
-                return False
-            
-            # Step 2: Check and install dependencies
-            if not self.setup_dependencies():
-                return False
-            
-            # Step 3: Create directory structure
-            self.create_directory_structure()
-            
-            # Step 4: Create configuration
-            self.create_configuration()
-            
-            # Step 5: Initialize database
-            if not self.setup_database():
-                return False
-            
-            # Step 6: Create sample data
-            self.create_sample_data()
-            
-            # Step 7: Setup environment files
-            self.create_environment_files()
-            
-            # Step 8: Verify installation
-            if not self.verify_installation():
-                return False
-            
-            # Step 9: Create launcher scripts
-            self.create_launcher_scripts()
-            
-            logger.info("✅ Setup completed successfully!")
-            self.show_final_instructions()
-            return True
+            # Ensure directory exists
+            db_dir = os.path.dirname(db_path)
+            if db_dir and not os.path.exists(db_dir):
+                os.makedirs(db_dir, exist_ok=True)
             
+            # Test database connection
+            with sqlite3.connect(db_path, timeout=10) as conn:
+                conn.execute("SELECT 1")
+                logger.info(f"✅ Database accessible at: {db_path}")
+                return db_path
+                
         except Exception as e:
-            logger.error(f"❌ Setup failed: {e}")
-            return False
+            logger.warning(f"⚠️ Cannot access {db_path}: {e}")
+            continue
     
-    def check_system_requirements(self) -> bool:
-        """Check system requirements"""
-        logger.info("🔍 Checking system requirements...")
-        
-        # Check Python version
-        python_version = sys.version_info
-        if python_version.major < 3 or (python_version.major == 3 and python_version.minor < 7):
-            logger.error("❌ Python 3.7+ is required")
-            return False
-        
-        logger.info(f"✅ Python {python_version.major}.{python_version.minor}.{python_version.micro}")
-        
-        # Check available disk space (at least 100MB)
-        try:
-            free_space = shutil.disk_usage('.').free / (1024 * 1024)  # MB
-            if free_space < 100:
-                logger.warning(f"⚠️ Low disk space: {free_space:.1f}MB available")
-            else:
-                logger.info(f"✅ Disk space: {free_space:.1f}MB available")
-        except Exception as e:
-            logger.warning(f"⚠️ Could not check disk space: {e}")
-        
-        # Check write permissions
-        try:
-            test_file = self.project_root / "test_write.tmp"
-            test_file.write_text("test")
-            test_file.unlink()
-            logger.info("✅ Write permissions OK")
-        except Exception as e:
-            logger.error(f"❌ No write permissions: {e}")
-            return False
-        
-        return True
+    logger.error("❌ No accessible database path found")
+    return None
+
+def check_permissions():
+    """Check write permissions for critical directories"""
+    logger.info("🔐 Checking permissions...")
     
-    def setup_dependencies(self) -> bool:
-        """Check and install Python dependencies"""
-        logger.info("📦 Setting up dependencies...")
-        
-        required_packages = {
-            'streamlit': '>=1.28.0',
-            'pandas': '>=1.5.0',
-            'plotly': '>=5.0.0',
-            'requests': '>=2.25.0',
-            'beautifulsoup4': '>=4.9.0',
-            'Pillow': '>=8.0.0',
-            'python-dateutil': '>=2.8.0'
-        }
-        
-        missing_packages = []
-        
-        for package, version in required_packages.items():
-            try:
-                if package == 'beautifulsoup4':
-                    import bs4
-                    logger.info(f"✅ {package} is available")
-                elif package == 'Pillow':
-                    import PIL
-                    logger.info(f"✅ {package} is available")
-                elif package == 'python-dateutil':
-                    import dateutil
-                    logger.info(f"✅ {package} is available")
-                else:
-                    __import__(package)
-                    logger.info(f"✅ {package} is available")
-            except ImportError:
-                missing_packages.append(f"{package}{version}")
-                logger.warning(f"⚠️ {package} is missing")
-        
-        if missing_packages:
-            logger.info("📥 Installing missing packages...")
-            
-            # Create requirements.txt
-            requirements_content = "\n".join(missing_packages)
-            requirements_file = self.project_root / "requirements.txt"
-            requirements_file.write_text(requirements_content)
-            
-            try:
-                # Try to install packages
-                subprocess.check_call([
-                    sys.executable, "-m", "pip", "install", "-r", str(requirements_file)
-                ])
-                logger.info("✅ Dependencies installed successfully")
-            except subprocess.CalledProcessError as e:
-                logger.error(f"❌ Failed to install dependencies: {e}")
-                logger.info("Please run: pip install -r requirements.txt")
-                return False
-        
-        return True
+    test_dirs = ["/tmp", ".", "./data", "./src"]
     
-    def create_directory_structure(self):
-        """Create necessary directories"""
-        logger.info("📁 Creating directory structure...")
-        
-        directories = [
-            self.data_dir,
-            self.logs_dir,
-            self.temp_dir,
-            self.project_root / "assets",
-            self.project_root / "backups",
-            self.project_root / "exports"
-        ]
-        
-        for directory in directories:
+    results = {}
+    for test_dir in test_dirs:
+        if os.path.exists(test_dir):
             try:
-                directory.mkdir(exist_ok=True)
-                logger.info(f"✅ Created/verified directory: {directory}")
+                test_file = os.path.join(test_dir, "test_write.tmp")
+                with open(test_file, 'w') as f:
+                    f.write("test")
+                os.remove(test_file)
+                results[test_dir] = True
+                logger.info(f"✅ {test_dir} - Writable")
             except Exception as e:
-                logger.error(f"❌ Failed to create directory {directory}: {e}")
+                results[test_dir] = False
+                logger.warning(f"❌ {test_dir} - Not writable: {e}")
+        else:
+            results[test_dir] = None
+            logger.info(f"ℹ️ {test_dir} - Does not exist")
     
-    def create_configuration(self):
-        """Create configuration file"""
-        logger.info("⚙️ Creating configuration...")
-        
-        config = {
-            "app": {
-                "name": "Iran Legal Information Dashboard",
-                "version": "1.0.0",
-                "description": "Legal document management and analysis system",
-                "created": datetime.now().isoformat()
-            },
-            "database": {
-                "type": "sqlite",
-                "path": str(self.data_dir / "iran_legal.db"),
-                "backup_interval": "daily",
-                "max_backups": 7
-            },
-            "ai": {
-                "enabled": True,
-                "min_score_threshold": 0.3,
-                "keyword_extraction_limit": 10
-            },
-            "scraping": {
-                "enabled": True,
-                "default_timeout": 15,
-                "max_concurrent": 5,
-                "delay_between_requests": 1,
-                "trusted_domains": [
-                    "dastour.ir",
-                    "mizanonline.ir",
-                    "judiciary.ir",
-                    "majlis.ir",
-                    "rc.majlis.ir"
-                ]
-            },
-            "ui": {
-                "language": "fa",
-                "theme": "modern",
-                "items_per_page": 20,
-                "enable_animations": True
-            },
-            "security": {
-                "enable_logging": True,
-                "max_upload_size": "10MB",
-                "allowed_file_types": ["pdf", "txt", "docx"]
-            },
-            "paths": {
-                "data": str(self.data_dir),
-                "logs": str(self.logs_dir),
-                "temp": str(self.temp_dir),
-                "backups": str(self.project_root / "backups"),
-                "exports": str(self.project_root / "exports")
-            }
-        }
-        
-        try:
-            with open(self.config_file, 'w', encoding='utf-8') as f:
-                json.dump(config, f, indent=2, ensure_ascii=False)
-            logger.info(f"✅ Configuration created: {self.config_file}")
-        except Exception as e:
-            logger.error(f"❌ Failed to create configuration: {e}")
+    return results
+
+def verify_imports():
+    """Verify all required imports work"""
+    logger.info("📦 Verifying imports...")
+    
+    required_modules = [
+        'streamlit',
+        'pandas', 
+        'plotly',
+        'requests',
+        'bs4',
+        'sqlite3',
+        'json',
+        'hashlib',
+        'datetime'
+    ]
     
-    def setup_database(self) -> bool:
-        """Initialize database with tables and sample data"""
-        logger.info("🗄️ Setting up database...")
-        
-        db_path = self.data_dir / "iran_legal.db"
-        
+    failed_imports = []
+    
+    for module in required_modules:
         try:
-            with sqlite3.connect(str(db_path)) as conn:
-                # Enable WAL mode and foreign keys
-                conn.execute("PRAGMA journal_mode=WAL;")
-                conn.execute("PRAGMA foreign_keys = ON;")
-                
-                # Create tables
-                self._create_database_tables(conn)
-                
-                # Add sample data
-                self._add_sample_data(conn)
-                
-                logger.info(f"✅ Database initialized: {db_path}")
-                return True
-                
-        except Exception as e:
-            logger.error(f"❌ Database setup failed: {e}")
-            return False
+            __import__(module)
+            logger.info(f"✅ {module} - OK")
+        except ImportError as e:
+            failed_imports.append(module)
+            logger.error(f"❌ {module} - Failed: {e}")
     
-    def _create_database_tables(self, conn: sqlite3.Connection):
-        """Create database tables"""
-        
-        # Documents table
-        conn.execute("""
-            CREATE TABLE IF NOT EXISTS documents (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                title TEXT NOT NULL,
-                content TEXT NOT NULL,
-                source TEXT,
-                category TEXT,
-                ai_score REAL DEFAULT 0.0,
-                keywords TEXT,
-                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                file_size INTEGER DEFAULT 0,
-                language TEXT DEFAULT 'fa',
-                status TEXT DEFAULT 'active'
-            )
-        """)
-        
-        # Scraped items table
-        conn.execute("""
-            CREATE TABLE IF NOT EXISTS scraped_items (
-                id TEXT PRIMARY KEY,
-                url TEXT NOT NULL UNIQUE,
-                title TEXT,
-                content TEXT,
-                domain TEXT,
-                rating_score REAL DEFAULT 0.0,
-                word_count INTEGER DEFAULT 0,
-                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                status TEXT DEFAULT 'completed',
-                last_scraped TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-            )
-        """)
-        
-        # System logs table
-        conn.execute("""
-            CREATE TABLE IF NOT EXISTS system_logs (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                level TEXT NOT NULL,
-                component TEXT,
-                message TEXT NOT NULL,
-                details TEXT
-            )
-        """)
-        
-        # User sessions table (for future multi-user support)
-        conn.execute("""
-            CREATE TABLE IF NOT EXISTS user_sessions (
-                id TEXT PRIMARY KEY,
-                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                last_activity TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-                ip_address TEXT,
-                user_agent TEXT,
-                active BOOLEAN DEFAULT 1
-            )
-        """)
-        
-        # Performance indexes
-        indexes = [
-            "CREATE INDEX IF NOT EXISTS idx_documents_category ON documents(category);",
-            "CREATE INDEX IF NOT EXISTS idx_documents_ai_score ON documents(ai_score);",
-            "CREATE INDEX IF NOT EXISTS idx_documents_created_at ON documents(created_at);",
-            "CREATE INDEX IF NOT EXISTS idx_scraped_domain ON scraped_items(domain);",
-            "CREATE INDEX IF NOT EXISTS idx_scraped_rating ON scraped_items(rating_score);",
-            "CREATE INDEX IF NOT EXISTS idx_scraped_created_at ON scraped_items(created_at);",
-            "CREATE INDEX IF NOT EXISTS idx_logs_timestamp ON system_logs(timestamp);",
-            "CREATE INDEX IF NOT EXISTS idx_logs_level ON system_logs(level);"
-        ]
-        
-        for index in indexes:
-            conn.execute(index)
-        
-        conn.commit()
-        logger.info("✅ Database tables created")
+    if failed_imports:
+        logger.error(f"❌ Failed imports: {failed_imports}")
+        logger.info("💡 Run: pip install -r requirements.txt")
+        return False
     
-    def _add_sample_data(self, conn: sqlite3.Connection):
-        """Add comprehensive sample data"""
-        
-        # Sample documents
-        sample_documents = [
-            {
-                'title': 'قانون اساسی جمهوری اسلامی ایران - فصل اول',
-                'content': '''فصل اول - اصول کلی
-
-اصل یکم: حکومت ایران، جمهوری اسلامی است که ملت ایران بر اساس ایمان دیرینه‌اش به حاکمیت حق و عدالت قرآن، در انقلاب اسلامی سال ۱۳۵۷ تحت رهبری امام خمینی به آن رأی مثبت داد.
-
-اصل دوم: جمهوری اسلامی، نظامی است مبتنی بر ایمان به خدای یکتا، حاکمیت و قانون الهی، عدل، ایمان و تقوا، و در حکومت، دین مبین اسلام مذهب جعفری اثنی عشری.
-
-اصل سوم: برای نیل به اهداف مندرج در اصل دوم، دولت جمهوری اسلامی ایران موظف است تمام امکانات خود را برای دستیابی به مقاصد ذیل به کار گیرد:
-۱- ایجاد محیط مساعد برای رشد فضایل اخلاقی بر پایه ایمان و تقوا و مبارزه با تمام مظاهر فساد و تباهی
-۲- ارتقاء سطح آگاهی عمومی در همه زمینه‌ها با بهره‌گیری درست از مطبوعات، رسانه‌های گروهی و سایر وسایل
-۳- آموزش و پرورش و تربیت بدنی رایگان برای همه در همه سطوح و تسهیل و تعمیم آموزش عالی''',
-                'source': 'قانون اساسی ج.ا.ایران',
-                'category': 'قانون',
-                'ai_score': 0.95,
-                'keywords': json.dumps(['قانون اساسی', 'جمهوری اسلامی', 'حاکمیت', 'عدالت', 'اسلام']),
-                'file_size': 2450,
-                'language': 'fa'
-            },
-            {
-                'title': 'قانون مجازات اسلامی - احکام عمومی',
-                'content': '''باب اول - احکام عمومی
-
-ماده ۱- مجازات‌ها به اعتبار کیفیت به سه دسته تقسیم می‌شوند:
-الف) حدود
-ب) قصاص
-ج) تعزیرات
-
-ماده ۲- حدود عبارت است از مجازات‌هایی که نوع و میزان آنها در شرع مقدس تعیین شده و حق العبد محسوب نمی‌شود.
-
-ماده ۳- قصاص عبارت است از مجازاتی که در شرع مقدس تعیین شده و حق العبد محسوب می‌شود.
-
-ماده ۴- تعزیرات عبارت است از مجازات‌هایی غیر از حدود و قصاص که برای تأدیب مرتکب یا جلوگیری از تکرار جرم یا ارتداع دیگران تعیین می‌شود.
-
-ماده ۵- هرگاه برای یک عمل مجرمانه، مجازات حد یا قصاص و نیز تعزیر تعیین شده باشد، مجازات حد یا قصاص مانع از اجرای تعزیر نیست مگر آنکه قانون طور دیگری مقرر داشته باشد.''',
-                'source': 'قانون مجازات اسلام��',
-                'category': 'قانون',
-                'ai_score': 0.88,
-                'keywords': json.dumps(['مجازات', 'حدود', 'قصاص', 'تعزیرات', 'شرع']),
-                'file_size': 1850,
-                'language': 'fa'
-            },
-            {
-                'title': 'قانون مدنی - قراردادها',
-                'content': '''فصل اول - احکام عمومی قراردادها
-
-ماده ۱۰- قراردادها برای متعاقدینی که آنها را منعقد نمده‌اند لازم‌الاجرا و قابل فسخ نیست مگر به رضایت طرفین یا در مواردی که قانون تجویز کرده است.
+    return True
 
-ماده ۱۱- در قراردادها نه تنها به الفاظ بلکه به قصد و نیت طرفین نیز اعتبار است.
-
-ماده ۱۲- مجهول‌بودن مبیع یا ثمن مانع از صحت بیع است مگر آنکه مجهول بودن آن خفیف و متعارف باشد.
-
-ماده ۱۳- در عقود و ایقاعات، عبرت به قصد و معنی است نه به الفاظ.
-
-ماده ۱۴- تصرف در مال غیر بدون اذن مالک موجب ضمان است.''',
-                'source': 'قانون مدنی',
-                'category': 'قرارداد',
-                'ai_score': 0.82,
-                'keywords': json.dumps(['قرارداد', 'متعاقدین', 'بیع', 'ثمن', 'مبیع']),
-                'file_size': 1620,
-                'language': 'fa'
-            },
-            {
-                'title': 'قانون آیین دادرسی کیفری - کلیات',
-                'content': '''باب اول - کلیات
-
-ماده ۱- این قانون آیین تعقیب و محاکمه جرایم و اجرای احکام کیفری را تعیین می‌کند.
-
-ماده ۲- دادرسی کیفری بر اساس احترام به کرامت انسانی، حفظ حقوق فردی و اجتماعی، حق دفاع، برائت ذمه، آزادی و امنیت اشخاص استوار است.
-
-ماده ۳- هیچ کس را نمی‌توان بدون حکم مقام قضایی ذی‌صلاح دستگیر، بازداشت یا زندانی کرد.
-
-ماده ۴- متهم حق دارد در تمام مراحل دادرسی وکیل انتخاب کند.
-
-ماده ۵- تفتیش مسکن، محل کسب و کار، وسایل نقلیه یا اماکن دیگر جز در موارد مستثنی قانونی منوط به حکم مقام قضایی است.''',
-                'source': 'قانون آیین دادرسی کیفری',
-                'category': 'حکم',
-                'ai_score': 0.90,
-                'keywords': json.dumps(['دادرسی', 'کیفری', 'متهم', 'دادگاه', 'قاضی']),
-                'file_size': 2100,
-                'language': 'fa'
-            }
-        ]
-        
-        # Insert sample documents
-        for doc in sample_documents:
+def create_sample_data(db_path):
+    """Create sample data for testing"""
+    logger.info("📝 Creating sample data...")
+    
+    try:
+        with sqlite3.connect(db_path) as conn:
+            # Check if data already exists
+            cursor = conn.execute("SELECT COUNT(*) FROM documents")
+            doc_count = cursor.fetchone()[0]
+            
+            if doc_count > 0:
+                logger.info(f"ℹ️ Database already has {doc_count} documents")
+                return True
+            
+            # Create sample document
             conn.execute("""
                 INSERT INTO documents (title, content, source, category, ai_score, keywords, file_size, language)
                 VALUES (?, ?, ?, ?, ?, ?, ?, ?)
             """, (
-                doc['title'], doc['content'], doc['source'], doc['category'],
-                doc['ai_score'], doc['keywords'], doc['file_size'], doc['language']
+                "نمونه سند حقوقی",
+                "این یک نمونه سند حقوقی برای تست سیستم است که شامل محتوای آزمایشی می‌باشد.",
+                "سیستم تست",
+                "قانون",
+                0.75,
+                '["نمونه", "حقوق", "تست"]',
+                150,
+                "fa"
             ))
-        
-        # Sample scraped items
-        sample_scraped = [
-            {
-                'id': 'sample_dastour_001',
-                'url': 'https://dastour.ir/bakhshodare/ghanoon-asasi',
-                'title': 'متن کامل قانون اساسی جمهوری اسلامی ایران',
-                'content': 'قانون اساسی جمهوری اسلامی ایران که در سال ۱۳۵۸ تصویب و در سال ۱۳۶۸ بازنگری شد، مشتمل بر ۱۷۷ اصل است که در ۱۴ فصل تنظیم شده است. این قانون بنیادی‌ترین و مهم‌ترین قانون کشور محسوب می‌شود.',
-                'domain': 'dastour.ir',
-                'rating_score': 0.95,
-                'word_count': 15420,
-                'status': 'completed'
-            },
-            {
-                'id': 'sample_mizan_001',
-                'url': 'https://mizanonline.ir/fa/news/legal-reforms-2024',
-                'title': 'اصلاحات جدید در نظام حقوقی کشور - ۱۴۰۳',
-                'content': 'در راستای بهبود نظام حقوقی کشور و همسو شدن با تحولات جهانی، مجموعه‌ای از اصلاحات قانونی در دست بررسی قرار گرفته که شامل تغییرات در قوانین مدنی، جزایی و اداری می‌شود. این اصلاحات با هدف تسریع روند دادرسی و حفظ حقوق شهروندان طراحی شده‌اند.',
-                'domain': 'mizanonline.ir',
-                'rating_score': 0.78,
-                'word_count': 852,
-                'status': 'completed'
-            },
-            {
-                'id': 'sample_judiciary_001',
-                'url': 'https://judiciary.ir/news/supreme-court-decisions-2024',
-                'title': 'آرای مهم دیوان عالی کشور در سال ۱۴۰۳',
-                'content': 'دیوان عالی کشور در سال جاری مجموعه‌ای از آرای مهم و تأثیرگذار در خصوص تفسیر قوانین مختلف صادر کرده که نقش به‌سزایی در وحدت رویه قضایی و شفاف‌سازی مسائل حقوقی خواهد داشت. این آرا شامل موضوعات مدنی، کیفری و اداری می‌شود.',
-                'domain': 'judiciary.ir',
-                'rating_score': 0.87,
-                'word_count': 1240,
-                'status': 'completed'
-            }
-        ]
-        
-        # Insert sample scraped items
-        for item in sample_scraped:
+            
+            # Create sample scraped item
             conn.execute("""
                 INSERT INTO scraped_items (id, url, title, content, domain, rating_score, word_count, status)
                 VALUES (?, ?, ?, ?, ?, ?, ?, ?)
             """, (
-                item['id'], item['url'], item['title'], item['content'],
-                item['domain'], item['rating_score'], item['word_count'], item['status']
+                "sample_001",
+                "https://dastour.ir/sample",
+                "نمونه محتوای حقوقی",
+                "این یک نمونه محتوای حقوقی جمع‌آوری شده از وب است.",
+                "dastour.ir",
+                0.80,
+                25,
+                "completed"
             ))
-        
-        # Add system log entry
-        conn.execute("""
-            INSERT INTO system_logs (level, component, message, details)
-            VALUES (?, ?, ?, ?)
-        """, ('INFO', 'setup', 'Database initialized with sample data', 
-              json.dumps({'documents': len(sample_documents), 'scraped_items': len(sample_scraped)})))
-        
-        conn.commit()
-        logger.info(f"✅ Added {len(sample_documents)} sample documents and {len(sample_scraped)} scraped items")
-    
-    def create_sample_data(self):
-        """Create additional sample data files"""
-        logger.info("📝 Creating sample data files...")
-        
-        # Create sample upload file
-        sample_text = """نمونه متن حقوقی
-
-این یک نمونه متن حقوقی است که می‌توانید برای تست سیستم استفاده کنید.
-
-ماده ۱- این متن به عنوان نمونه تهیه شده است.
-ماده ۲- محتوای این متن صرفاً جنبه آموزشی دارد.
+            
+            conn.commit()
+            logger.info("✅ Sample data created successfully")
+            return True
+            
+    except Exception as e:
+        logger.error(f"❌ Failed to create sample data: {e}")
+        return False
 
-تبصره: این متن قابلیت آپلود و تحلیل توسط سیستم هوش مصنوعی را دارد.
-"""
-        
-        sample_file = self.temp_dir / "sample_legal_text.txt"
-        sample_file.write_text(sample_text, encoding='utf-8')
-        logger.info(f"✅ Sample file created: {sample_file}")
+def main():
+    """Main startup function"""
+    logger.info("🚀 Starting Iran Legal Dashboard initialization...")
     
-    def create_environment_files(self):
-        """Create environment and configuration files"""
-        logger.info("🔧 Creating environment files...")
-        
-        # Create .streamlit directory and config
-        streamlit_dir = self.project_root / ".streamlit"
-        streamlit_dir.mkdir(exist_ok=True)
-        
-        # Streamlit config
-        streamlit_config = """[theme]
-primaryColor = "#667eea"
-backgroundColor = "#ffffff"
-secondaryBackgroundColor = "#f0f2f6"
-textColor = "#262730"
-font = "sans serif"
-
-[server]
-headless = true
-port = 8501
-enableCORS = false
-enableXsrfProtection = false
-
-[browser]
-gatherUsageStats = false
-"""
-        
-        (streamlit_dir / "config.toml").write_text(streamlit_config)
-        
-        # Create .env file
-        env_content = """# Iran Legal Dashboard Environment Variables
-
-# Application Settings
-APP_NAME="Iran Legal Information Dashboard"
-APP_VERSION="1.0.0"
-DEBUG=false
-
-# Database Settings
-DATABASE_PATH="./data/iran_legal.db"
-DATABASE_BACKUP_ENABLED=true
-
-# Security Settings
-SECRET_KEY="your-secret-key-here"
-MAX_UPLOAD_SIZE=10485760
-
-# Feature Flags
-ENABLE_AI_ANALYSIS=true
-ENABLE_WEB_SCRAPING=true
-ENABLE_LOGGING=true
-
-# External Services (if needed)
-# API_KEY=""
-# EXTERNAL_SERVICE_URL=""
-"""
-        
-        env_file = self.project_root / ".env"
-        env_file.write_text(env_content)
-        
-        logger.info("✅ Environment files created")
+    # Check environment
+    if not check_environment():
+        logger.error("❌ Environment check failed")
+        return False
     
-    def verify_installation(self) -> bool:
-        """Verify the installation"""
-        logger.info("🔍 Verifying installation...")
-        
-        checks = [
-            (self.data_dir / "iran_legal.db", "Database file"),
-            (self.config_file, "Configuration file"),
-            (self.project_root / ".streamlit" / "config.toml", "Streamlit config"),
-            (self.project_root / ".env", "Environment file")
-        ]
-        
-        all_good = True
-        for file_path, description in checks:
-            if file_path.exists():
-                logger.info(f"✅ {description}: OK")
-            else:
-                logger.error(f"❌ {description}: Missing")
-                all_good = False
-        
-        # Test database connection
-        try:
-            db_path = self.data_dir / "iran_legal.db"
-            with sqlite3.connect(str(db_path)) as conn:
-                cursor = conn.execute("SELECT COUNT(*) FROM documents")
-                doc_count = cursor.fetchone()[0]
-                cursor = conn.execute("SELECT COUNT(*) FROM scraped_items")
-                scraped_count = cursor.fetchone()[0]
-                logger.info(f"✅ Database connectivity: OK ({doc_count} documents, {scraped_count} scraped items)")
-        except Exception as e:
-            logger.error(f"❌ Database connectivity: Failed - {e}")
-            all_good = False
-        
-        return all_good
+    # Check permissions
+    check_permissions()
     
-    def create_launcher_scripts(self):
-        """Create launcher scripts for different platforms"""
-        logger.info("🚀 Creating launcher scripts...")
-        
-        # Windows batch file
-        windows_script = """@echo off
-echo Starting Iran Legal Dashboard...
-echo.
-
-:: Check if Python is available
-python --version >nul 2>&1
-if errorlevel 1 (
-    echo Python is not installed or not in PATH
-    pause
-    exit /b 1
-)
-
-:: Check if Streamlit is available
-python -c "import streamlit" >nul 2>&1
-if errorlevel 1 (
-    echo Installing required packages...
-    pip install -r requirements.txt
-)
-
-:: Start the application
-echo Starting application on http://localhost:8501
-echo Press Ctrl+C to stop
-echo.
-streamlit run streamlit_app.py
-
-pause
-"""
-        
-        windows_file = self.project_root / "start_dashboard.bat"
-        windows_file.write_text(windows_script)
-        
-        # Unix shell script
-        unix_script = """#!/bin/bash
-echo "Starting Iran Legal Dashboard..."
-echo
-
-# Check if Python is available
-if ! command -v python3 &> /dev/null; then
-    echo "Python 3 is not installed or not in PATH"
-    exit 1
-fi
-
-# Check if Streamlit is available
-python3 -c "import streamlit" 2>/dev/null
-if [ $? -ne 0 ]; then
-    echo "Installing required packages..."
-    pip3 install -r requirements.txt
-fi
-
-# Start the application
-echo "Starting application on http://localhost:8501"
-echo "Press Ctrl+C to stop"
-echo
-streamlit run streamlit_app.py
-"""
-        
-        unix_file = self.project_root / "start_dashboard.sh"
-        unix_file.write_text(unix_script)
-        
-        # Make shell script executable
-        try:
-            unix_file.chmod(0o755)
-        except Exception:
-            pass  # Windows doesn't support chmod
-        
-        # Create requirements.txt
-        requirements = """streamlit>=1.28.0
-pandas>=1.5.0
-plotly>=5.0.0
-requests>=2.25.0
-beautifulsoup4>=4.9.0
-Pillow>=8.0.0
-python-dateutil>=2.8.0
-"""
-        
-        requirements_file = self.project_root / "requirements.txt"
-        requirements_file.write_text(requirements)
-        
-        logger.info("✅ Launcher scripts created")
+    # Verify imports
+    if not verify_imports():
+        logger.error("❌ Import verification failed")
+        return False
     
-    def show_final_instructions(self):
-        """Show final setup instructions"""
-        logger.info("\n" + "="*60)
-        logger.info("🎉 SETUP COMPLETED SUCCESSFULLY!")
-        logger.info("="*60)
-        logger.info("")
-        logger.info("📋 NEXT STEPS:")
-        logger.info("")
-        logger.info("1. To start the application:")
-        
-        if os.name == 'nt':  # Windows
-            logger.info("   Windows: Double-click 'start_dashboard.bat'")
-            logger.info("   Or run: streamlit run streamlit_app.py")
-        else:  # Unix-like
-            logger.info("   Unix/Linux/macOS: ./start_dashboard.sh")
-            logger.info("   Or run: streamlit run streamlit_app.py")
-        
-        logger.info("")
-        logger.info("2. Open your browser and go to: http://localhost:8501")
-        logger.info("")
-        logger.info("📁 IMPORTANT FILES:")
-        logger.info(f"   • Main app: streamlit_app.py")
-        logger.info(f"   • Database: {self.data_dir}/iran_legal.db")
-        logger.info(f"   • Config: {self.config_file}")
-        logger.info(f"   • Logs: {self.logs_dir}/")
-        logger.info("")
-        logger.info("🔧 TROUBLESHOOTING:")
-        logger.info("   • Check setup.log for detailed logs")
-        logger.info("   • Ensure Python 3.7+ is installed")
-        logger.info("   • Run 'pip install -r requirements.txt' if needed")
-        logger.info("")
-        logger.info("📚 FEATURES READY:")
-        logger.info("   ✅ Document management")
-        logger.info("   ✅ AI analysis engine")
-        logger.info("   ✅ Web scraping service")
-        logger.info("   ✅ Search functionality")
-        logger.info("   ✅ Analytics dashboard")
-        logger.info("")
-        logger.info("="*60)
-
-def main():
-    """Main setup function"""
-    print("🚀 Iran Legal Dashboard Setup")
-    print("=" * 40)
+    # Setup database
+    db_path = setup_database()
+    if not db_path:
+        logger.error("❌ Database setup failed")
+        return False
     
-    setup_manager = SetupManager()
+    # Create sample data
+    create_sample_data(db_path)
     
-    try:
-        success = setup_manager.run_complete_setup()
-        
-        if success:
-            print("\n✅ Setup completed successfully!")
-            print("Run 'streamlit run streamlit_app.py' to start the application.")
-        else:
-            print("\n❌ Setup failed. Check setup.log for details.")
-            sys.exit(1)
-            
-    except KeyboardInterrupt:
-        print("\n\n⚠️ Setup interrupted by user")
-        sys.exit(1)
-    except Exception as e:
-        logger.error(f"Unexpected error: {e}")
-        print(f"\n❌ Unexpected error: {e}")
-        sys.exit(1)
+    logger.info("✅ Initialization completed successfully!")
+    logger.info("🎉 Ready to run: streamlit run app.py")
+    
+    return True
 
 if __name__ == "__main__":
-    main()
+    success = main()
+    if not success:
+        sys.exit(1)