Spaces:
Running
Running
File size: 1,967 Bytes
cce0ed9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
#!/usr/bin/env python3
import subprocess
import sys
import os
import time
import signal
import atexit
def start_api_server():
api_process = subprocess.Popen(
[sys.executable, "api.py"],
cwd=os.path.dirname(os.path.abspath(__file__))
)
print("API server started (PID:", api_process.pid, ")")
return api_process
def start_gradio_app():
gradio_process = subprocess.Popen(
[sys.executable, "-c", "import gradio as gr; import run; run.demo.launch()"],
cwd=os.path.dirname(os.path.abspath(__file__))
)
print("Gradio interface started (PID:", gradio_process.pid, ")")
return gradio_process
def cleanup_processes(api_process, gradio_process):
print("\nShutting down services...")
if api_process and api_process.poll() is None:
api_process.terminate()
print("API server terminated")
if gradio_process and gradio_process.poll() is None:
gradio_process.terminate()
print("Gradio interface terminated")
def main():
api_process = start_api_server()
# Give the API server a moment to start
time.sleep(2)
gradio_process = start_gradio_app()
# Register cleanup function to be called on exit
atexit.register(cleanup_processes, api_process, gradio_process)
# Handle keyboard interrupts
def signal_handler(sig, frame):
print("\nReceived termination signal")
cleanup_processes(api_process, gradio_process)
sys.exit(0)
signal.signal(signal.SIGINT, signal_handler)
signal.signal(signal.SIGTERM, signal_handler)
print("\nLeaderboard application started!")
print("- API server running at http://localhost:8000")
print("- Gradio interface running at http://localhost:7860")
print("\nPress Ctrl+C to stop all services")
# Keep the main process running
try:
while True:
time.sleep(1)
except KeyboardInterrupt:
pass
if __name__ == "__main__":
main()
|