diving into adding connection to database ,simple test

app.py

@@ -1,25 +1,46 @@
-from fastapi import FastAPI
-import ast 
-from typing import List,Union
+from fastapi import FastAPI, HTTPException
+from typing import List, Union
 import requests
+import ast
 
-app = FastAPI()
+from sqlalchemy import create_engine, text
+from sqlalchemy.orm import sessionmaker, Session
 
-@app.get("/")
-def read_root():
-    return {"message": "Hello from FastAPI on Hugging Face Spaces!"}
+# -------------------------
+# App Initialization
+# -------------------------
+app = FastAPI()
 
+# -------------------------
+# Database Setup (SQLAlchemy + PyMySQL)
+# -------------------------
+DB_URL = (
+    "mysql+pymysql://avnadmin:AVNS_aT0RGFafs6_34WFegSF"
+    "@mysql-19285eb2-tahaelshrif1-7999.h.aivencloud.com:11520/trafficManagerSignals"
+)
+
+engine = create_engine(DB_URL, pool_size=10, max_overflow=20, pool_pre_ping=True)
+SessionLocal = sessionmaker(bind=engine)
+
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+# -------------------------
+# Utilities
+# -------------------------
 def get_rectangle_container(coordinates):
     lons = [i for i, _ in coordinates]
     lats = [i for _, i in coordinates]
-
     min_lat, max_lat = min(lats), max(lats)
     min_lon, max_lon = min(lons), max(lons)
-    return (min_lat,max_lat,min_lon,max_lon)
-
+    return min_lat, max_lat, min_lon, max_lon
 
 def get_traffic_in_container(coordinates):
-    (min_lat,max_lat,min_lon,max_lon)= get_rectangle_container(coordinates)
+    (min_lat, max_lat, min_lon, max_lon) = get_rectangle_container(coordinates)
     overpass_url = "http://overpass-api.de/api/interpreter"
     query = f"""
     [out:json];
@@ -28,67 +49,47 @@ def get_traffic_in_container(coordinates):
     """
     response = requests.get(overpass_url, params={'data': query})
     data = response.json()
-    traffic_lights = [(el['lat'], el['lon']) for el in data['elements']]
-    return traffic_lights
+    return [(el['lat'], el['lon']) for el in data['elements']]
 
-import asyncio
-async def fetch_data():
-    await asyncio.sleep(1)  # Simulate an asynchronous operation
-    return {"data": "Fetched data"}
-
-@app.post("/tmp/")
-async def read_root(s:str):
-    data = await fetch_data()
-    return {"data": data}
+# -------------------------
+# Routes
+# -------------------------
+@app.get("/")
+def read_root():
+    return {"message": "Hello from FastAPI on Hugging Face Spaces!"}
 
 @app.post("/send-coordinates")
 def receive_coordinates(coords: Union[str, List[float]]):
-    """
-    Receive a list of coordinates from the mobile app.
-    Pass these coordinates to a function that checks which traffic signals exist in these areas.
-    """
-    if isinstance(coords,str):
-        coords = ast.literal_eval(coords) 
-    traffic_lights = get_traffic_in_container(coords)
-    tl_id = check_signals(traffic_lights)
-     
-    if tl_id:
-        return {"message": f"Coordinates {traffic_lights}"}
-    else:
-        return {"Error": "No Traffic Light found"}
-
-
-
-# -----------------------
-# 2. Check traffic signals
-# -----------------------
-
-def check_signals(coords):
-    """
-    Check which coordinates have traffic signals by querying the database.
-    Call a function to verify if it's allowed to open the signal.
-    """
-    return "6082411793"
-
-
-# -----------------------
-# 3. Verify signal status
-# -----------------------
-
-def can_open_signal(signal_data):
-    """
-    Determine whether the signal can be opened based on returned data.
-    If yes, call the IoT function to open the signal.
-    """
-    pass
+    if isinstance(coords, str):
+        coords = ast.literal_eval(coords)
 
+    traffic_lights = get_traffic_in_container(coords)
+    tl_ids = check_signals(traffic_lights)
 
-# -----------------------
-# 4. IoT trigger
-# -----------------------
+    if tl_ids:
+        return {"found_signals": tl_ids}
+    else:
+        return {"error": "No traffic signals found in database"}
 
-def trigger_iot_open(signal_id: str):
+# -------------------------
+# Signal Database Check
+# -------------------------
+def check_signals(coords, db: Session = next(get_db())):
     """
-    Send a GET request to the IoT system to open the signal with the given ID.
+    Check if lat/lon exists in traffic_signals DB table.
+    Returns list of tl_id_sumo.
     """
-    pass
+    found_signals = []
+    for lat, lon in coords:
+        result = db.execute(
+            text("""
+                SELECT tl_id_sumo FROM traffic_signals
+                WHERE lat = :lat AND lon = :lon
+            """),
+            {"lat": lat, "lon": lon}
+        ).fetchone()
+
+        if result:
+            found_signals.append(result.tl_id_sumo)
+
+    return found_signals

requirements.txt

@@ -1,3 +1,4 @@
 fastapi
 uvicorn[standard]
 requests
+pymysql