Update main.py

main.py

@@ -1,130 +1,64 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
+import pyrebase
 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")
+# Firebase config (client SDK)
+const firebaseConfig = {
+    apiKey: "AIzaSyAPgYJabQqcaKKCn6rob3AZd54s4hAnaAs",
+    authDomain: "sareesite-4aef4.firebaseapp.com",
+    projectId: "sareesite-4aef4",
+    storageBucket: "sareesite-4aef4.firebasestorage.app",
+    messagingSenderId: "504565105645",
+    appId: "1:504565105645:web:466ed0fc5267dc6550dde4"
+};
 
-# Add CORS middleware to allow cross-origin requests
+# Initialize app
+app = FastAPI()
+
+# CORS
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Allows all origins
+    allow_origins=["*"],
     allow_credentials=True,
-    allow_methods=["*"],  # Allows all methods
-    allow_headers=["*"],  # Allows all headers
+    allow_methods=["*"],
+    allow_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
+# Initialize pyrebase
+firebase = pyrebase.initialize_app(firebase_config)
+auth = firebase.auth()
+firestore = firebase.database()  # Note: this works only for Realtime DB, not Firestore
 
 @app.get("/")
-def read_root():
-    return {"message": "IoT Data API is running"}
+def root():
+    return {"message": "API is working"}
 
-@app.get("/api/latest", response_model=IoTData)
-async def get_latest_data():
-    """
-    Get the latest IoT data based on timestamp
-    """
+@app.get("/api/latest")
+def get_latest():
     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)}")
+        # Firestore REST API call (simulate client behavior)
+        import requests
+        url = f"https://firestore.googleapis.com/v1/projects/{firebase_config['projectId']}/databases/(default)/documents/IOT?pageSize=1&orderBy=timestamp desc"
 
-@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)}")
+        res = requests.get(url)
+        if res.status_code != 200:
+            raise HTTPException(status_code=500, detail="Failed to fetch from Firestore")
 
-@app.get("/api/health")
-async def health_check():
-    """
-    Health check endpoint to verify API is running
-    """
-    return {
-        "status": "healthy",
-        "timestamp": datetime.datetime.now().isoformat()
-    }
+        data = res.json()
 
-# 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)
+        if "documents" not in data or len(data["documents"]) == 0:
+            raise HTTPException(status_code=404, detail="No data found")
+
+        doc = data["documents"][0]
+        fields = doc["fields"]
+
+        # Convert Firestore field types to Python types
+        def parse_value(field):
+            return list(field.values())[0]
+
+        result = {k: parse_value(v) for k, v in fields.items()}
+        return JSONResponse(content=result)
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error: {str(e)}")