src/run.py

#!/usr/bin/env python3
"""
Startup script for FinSolve RAG System
Runs both FastAPI and Streamlit simultaneously
"""

import subprocess
import sys
import time
import os
import signal
import threading
from pathlib import Path

def run_fastapi():
    """Run FastAPI server"""
    print("🚀 Starting FastAPI server on http://localhost:8000")
    try:
        # Run FastAPI with uvicorn
        subprocess.run([
            sys.executable, "-m", "uvicorn", 
            "api:app", 
            "--host", "0.0.0.0", 
            "--port", "8000",
            "--reload"
        ], check=True)
    except KeyboardInterrupt:
        print("🛑 FastAPI server stopped")
    except Exception as e:
        print(f"❌ Error running FastAPI: {e}")

def run_streamlit():
    """Run Streamlit app"""
    print("🚀 Starting Streamlit app on http://localhost:8501")
    try:
        # Wait a moment for FastAPI to start
        time.sleep(3)
        
        # Run Streamlit
        subprocess.run([
            sys.executable, "-m", "streamlit", "run", 
            "streamlit_app.py",
            "--server.port", "8501",
            "--server.address", "0.0.0.0"
        ], check=True)
    except KeyboardInterrupt:
        print("🛑 Streamlit app stopped")
    except Exception as e:
        print(f"❌ Error running Streamlit: {e}")

def check_dependencies():
    """Check if all required dependencies are installed"""
    required_packages = [
        'fastapi', 'uvicorn', 'streamlit', 'chromadb', 
        'sentence-transformers', 'openai', 'pandas', 'plotly'
    ]
    
    missing_packages = []
    for package in required_packages:
        try:
            __import__(package.replace('-', '_'))
        except ImportError:
            missing_packages.append(package)
    
    if missing_packages:
        print(f"❌ Missing packages: {', '.join(missing_packages)}")
        print("📦 Install with: pip install -r requirements.txt")
        return False
    
    print("✅ All dependencies satisfied")
    return True

def signal_handler(signum, frame):
    """Handle shutdown signals"""
    print("\n🛑 Shutting down servers...")
    sys.exit(0)

def main():
    """Main function to start both servers"""
    print("🤖 FinSolve RAG System Startup")
    print("="*50)
    
    # Check dependencies
    if not check_dependencies():
        sys.exit(1)
    
    # Set up signal handlers for graceful shutdown
    signal.signal(signal.SIGINT, signal_handler)
    signal.signal(signal.SIGTERM, signal_handler)
    
    print("\n🏗️ Starting Complete Tech Stack:")
    print("📡 FastAPI: http://localhost:8000")
    print("🌐 Streamlit: http://localhost:8501")
    print("📖 API Docs: http://localhost:8000/docs")
    print("\n💡 Press Ctrl+C to stop both servers\n")
    
    try:
        # Start FastAPI in a separate thread
        fastapi_thread = threading.Thread(target=run_fastapi, daemon=True)
        fastapi_thread.start()
        
        # Run Streamlit in main thread
        run_streamlit()
        
    except KeyboardInterrupt:
        print("\n🛑 Received shutdown signal")
    except Exception as e:
        print(f"❌ Startup error: {e}")
    finally:
        print("👋 Goodbye!")

if __name__ == "__main__":
    main()