start_system.py

#!/usr/bin/env python3
"""
Startup script for the complete RAG Chatbot system.
"""
import subprocess
import time
import sys
import signal
import os
from pathlib import Path

def start_api():
    """Start the API server."""
    print("🚀 Starting API server...")
    return subprocess.Popen([
        sys.executable, "start_api.py"
    ], cwd=Path(__file__).parent)

def start_ui():
    """Start the UI server."""
    print("🎨 Starting UI server...")
    time.sleep(3)  # Wait for API to start
    return subprocess.Popen([
        sys.executable, "start_ui.py"
    ], cwd=Path(__file__).parent)

def main():
    """Main startup function."""
    print("🤖 RAG Chatbot System Startup")
    print("=" * 40)
    
    # Check if virtual environment is activated
    if not hasattr(sys, 'real_prefix') and not (hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix):
        print("⚠️  Warning: Virtual environment not detected")
        print("   Please run: source venv/bin/activate")
        print("   Then run this script again")
        return
    
    api_process = None
    ui_process = None
    
    try:
        # Start API
        api_process = start_api()
        
        # Start UI
        ui_process = start_ui()
        
        print("\n✅ System started successfully!")
        print("📊 API Documentation: http://localhost:8000/docs")
        print("🎨 Web Interface: http://localhost:7860")
        print("\nPress Ctrl+C to stop both services")
        
        # Wait for processes
        while True:
            time.sleep(1)
            
            # Check if processes are still running
            if api_process.poll() is not None:
                print("❌ API process stopped unexpectedly")
                break
            
            if ui_process.poll() is not None:
                print("❌ UI process stopped unexpectedly")
                break
    
    except KeyboardInterrupt:
        print("\n🛑 Shutting down system...")
        
    finally:
        # Clean shutdown
        if ui_process:
            ui_process.terminate()
            ui_process.wait()
            print("✅ UI server stopped")
        
        if api_process:
            api_process.terminate()
            api_process.wait()
            print("✅ API server stopped")
        
        print("👋 System shutdown complete")

if __name__ == "__main__":
    main()