Create app.py

app.py

@@ -0,0 +1,130 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+import firebase_admin
+from firebase_admin import credentials, firestore
+import os
+from pydantic import BaseModel
+from typing import Optional
+import datetime
+
+# Initialize FastAPI app
+app = FastAPI(title="IoT Data API", description="API for retrieving IoT device data from Firebase")
+
+# Add CORS middleware to allow cross-origin requests
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
+
+# Firebase configuration
+firebase_config = {
+    "apiKey": "AIzaSyAPgYJabQqcaKKCn6rob3AZd54s4hAnaAs",
+    "authDomain": "sareesite-4aef4.firebaseapp.com",
+    "projectId": "sareesite-4aef4",
+    "storageBucket": "sareesite-4aef4.firebasestorage.app",
+    "messagingSenderId": "504565105645",
+    "appId": "1:504565105645:web:466ed0fc5267dc6550dde4"
+}
+
+# Initialize Firebase
+try:
+    # For Firebase Admin SDK with configuration
+    firebase_admin.initialize_app(options={
+        'projectId': firebase_config['projectId'],
+    })
+    db = firestore.client()
+    print("Firebase connection established successfully")
+except Exception as e:
+    print(f"Error initializing Firebase: {e}")
+
+# Define the IoT data model
+class IoTData(BaseModel):
+    device_id: Optional[str] = None
+    timestamp: Optional[str] = None
+    detected_count: Optional[int] = None
+    has_detections: Optional[bool] = None
+    image_name: Optional[str] = None
+    imagebase64: Optional[str] = None
+
+@app.get("/")
+def read_root():
+    return {"message": "IoT Data API is running"}
+
+@app.get("/api/latest", response_model=IoTData)
+async def get_latest_data():
+    """
+    Get the latest IoT data based on timestamp
+    """
+    try:
+        # Query the IOT collection, order by timestamp in descending order, and limit to 1 document
+        iot_ref = db.collection("IOT")
+        query = iot_ref.order_by("timestamp", direction=firestore.Query.DESCENDING).limit(1)
+        docs = query.get()
+        
+        if not docs:
+            raise HTTPException(status_code=404, detail="No IoT data found")
+        
+        # Get the first (and only) document from the query result
+        latest_doc = docs[0]
+        data = latest_doc.to_dict()
+        
+        # Return the data as JSON
+        return JSONResponse(content=data)
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error retrieving data: {str(e)}")
+
+@app.get("/api/traffic-status")
+async def get_traffic_status():
+    """
+    Get traffic status based on the latest detected count
+    """
+    try:
+        # Query the latest document
+        iot_ref = db.collection("IOT")
+        query = iot_ref.order_by("timestamp", direction=firestore.Query.DESCENDING).limit(1)
+        docs = query.get()
+        
+        if not docs:
+            raise HTTPException(status_code=404, detail="No IoT data found")
+        
+        # Get detected count from the latest document
+        latest_doc = docs[0]
+        data = latest_doc.to_dict()
+        detected_count = data.get("detected_count")
+        
+        if detected_count is None:
+            return {"status": "UNKNOWN", "count": None}
+        
+        # Determine traffic status based on detected count
+        if detected_count < 4:
+            status = "LIGHT TRAFFIC"
+        elif detected_count < 7:
+            status = "MEDIUM TRAFFIC"
+        else:
+            status = "HEAVY TRAFFIC"
+            
+        return {"status": status, "count": detected_count}
+    
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error retrieving traffic status: {str(e)}")
+
+@app.get("/api/health")
+async def health_check():
+    """
+    Health check endpoint to verify API is running
+    """
+    return {
+        "status": "healthy",
+        "timestamp": datetime.datetime.now().isoformat()
+    }
+
+# Hugging Face Spaces typically expects this port
+if __name__ == "__main__":
+    import uvicorn
+    port = int(os.environ.get("PORT", 7860))
+    uvicorn.run("main:app", host="0.0.0.0", port=port, reload=True)