import os
import subprocess
import time
import signal
import sys

def run_services():
    """
    Run both API and Streamlit app as separate processes.
    """
    print("Starting NAFNet services...")
    
    # Set environment variables to suppress warnings
    os.environ["PYTHONWARNINGS"] = "ignore"
    
    # Define commands
    api_cmd = ["python", "api.py"]
    streamlit_cmd = ["streamlit", "run", "app.py", "--server.port=8501"]
    
    # Start API server
    print("Starting API server on port 8001...")
    api_process = subprocess.Popen(api_cmd)
    
    # Wait for API to start
    print("Waiting for API to initialize (5 seconds)...")
    time.sleep(5)
    
    # Start Streamlit app
    print("Starting Streamlit app on port 8501...")
    streamlit_process = subprocess.Popen(streamlit_cmd)
    
    print("\n" + "="*50)
    print("Services started successfully!")
    print("API running at: http://localhost:8001")
    print("Streamlit app running at: http://localhost:8501")
    print("="*50 + "\n")
    
    # Handle graceful shutdown on Ctrl+C
    def signal_handler(sig, frame):
        print("\nShutting down services...")
        streamlit_process.terminate()
        api_process.terminate()
        
        # Wait for processes to terminate
        streamlit_process.wait()
        api_process.wait()
        
        print("Services stopped.")
        sys.exit(0)
    
    # Register signal handler
    signal.signal(signal.SIGINT, signal_handler)
    
    try:
        # Keep the script running
        while True:
            time.sleep(1)
    except KeyboardInterrupt:
        # Handle keyboard interrupt (Ctrl+C)
        signal_handler(None, None)

if __name__ == "__main__":
    run_services()