debug

app.py

@@ -8,28 +8,40 @@ from mqtt_manager import *
 from traffic_utils import *
 from Utils import * 
 from threading import Thread
+import state
+from contextlib import asynccontextmanager
 
 # -------------------------
 # App Initialization
 # -------------------------
-app = FastAPI()
 
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Starting code
+    intialize_mqtt_brokers()
+    start_mqtt_publishers()
+    
+    yield  
+
+    # At End
+    print("🔻 Shutting down...")
+
+app = FastAPI(lifespan=lifespan)
 
 
 #--------------------------
 # Routes
 # -------------------------
 @app.get("/")
-def read_root():
+def main():
     return {"message": "Hello from FastAPI on Hugging Face Spaces!"}
 
 
 
-
 @app.post("/send-coordinates")
 def receive_coordinates(payload: CoordinatesPayload,background_tasks: BackgroundTasks):
     coords= payload.coords
-    state= payload.state
+    state_= payload.state
     time_= payload.time
 
     if isinstance(coords, str):
@@ -40,12 +52,7 @@ def receive_coordinates(payload: CoordinatesPayload,background_tasks: Background
     print("ok")
     if tl_ids:
         for tl_id in tl_ids:
-            start =time.time()
-            print(singlas_state[tl_id] )
-            Thread(target=open_signal, args=(tl_id, state, time_), daemon=True).start()
-            end =time.time()
-            print(end-start)
-            print(singlas_state[tl_id] )
+            background_tasks.add_task(open_signal, tl_id, state_, time_) #"EMR" # "ACC"
 
         return {"found_signals": tl_ids}
     else:

mqtt_manager.py

@@ -4,8 +4,7 @@ import time
 import threading
 from dotenv import load_dotenv
 import os
-from traffic_utils import status_acc,status_emr,status_free,singlas_state
-
+import state
 import paho.mqtt.client as mqtt
 import time
 
@@ -26,24 +25,22 @@ client = mqtt.Client()
 
 def intialize_mqtt_brokers():
     client.connect(mqtt_broker, mqtt_port)
-    for tl_id in singlas_state.keys():
+    for tl_id in state.singlas_state.keys():
         client.subscribe(topic_rep(tl_id))
 
 def start_mqtt_publishers():
-    for tl_id in singlas_state.keys():
+    for tl_id in state.singlas_state.keys():
         threading.Thread(target=mqtt_publisher_loop, args=(tl_id,), daemon=True).start()
 
 
 def get_State(tl_id):
-    if tl_id in singlas_state:
-        return singlas_state[tl_id]
+    if tl_id in state.singlas_state:
+        return state.singlas_state[tl_id]
     else:
         return "FREE"
     
 
 def mqtt_publisher_loop(tl_id):
-    client = mqtt.Client()
-    client.connect(mqtt_broker, mqtt_port)
     client.loop_start()
 
     last_state = None
@@ -56,7 +53,7 @@ def mqtt_publisher_loop(tl_id):
             last_state = current_state
         time.sleep(1)
 
-intialize_mqtt_brokers()
+
 
 
 

state.py

@@ -0,0 +1,7 @@
+status_free = "FREE"
+status_emr = "EMR"
+status_acc = "ACC"
+
+singlas_state = {
+    "1698478721" :status_free,
+    "6082411793":status_free }

traffic_utils.py

@@ -2,19 +2,15 @@ import requests
 from database_manager import *
 from dotenv import load_dotenv
 import os
-
+import state
 load_dotenv(dotenv_path="keys.env")
 
 overpass_url = os.getenv("URL_Traffic")
 
 import time
 
-status_free = "FREE"
-status_emr = "EMR"
-status_acc = "ACC"
-singlas_state = {
-    "1698478721" :status_free,
-    "6082411793":status_free}
+
+
 
 def get_traffic_in_container(coordinates):
     (min_lat, max_lat, min_lon, max_lon) = get_rectangle_container(coordinates)
@@ -35,12 +31,11 @@ def get_rectangle_container(coordinates):
     min_lon, max_lon = min(lons), max(lons)
     return min_lat, max_lat, min_lon, max_lon
 
-def open_signal(tl_id,state,time_):
-    singlas_state[tl_id] = state 
-    print(singlas_state[tl_id] )
+def open_signal(tl_id,state_,time_):
+    state.singlas_state[tl_id] = state_ 
     time.sleep(time_)
-    singlas_state[tl_id] = status_free 
-    print(singlas_state[tl_id] )
+    state.singlas_state[tl_id] = state.status_free 
+
 
 
 def check_signals(coords):