Create app.py

app.py

@@ -0,0 +1,42 @@
+from fastapi import FastAPI, UploadFile, File
+from models.object_detection import detect_faults
+from models.thermal_anomaly_detection import detect_thermal_anomalies
+from models.energy_anomaly_detection import detect_energy_anomalies
+import cv2
+import numpy as np
+
+app = FastAPI()
+
+@app.post("/detect_faults/")
+async def detect_faults_from_video(file: UploadFile = File(...)):
+    # Save the uploaded file temporarily
+    file_location = f"data/{file.filename}"
+    with open(file_location, "wb") as buffer:
+        buffer.write(file.file.read())
+    
+    # Read video or image for object detection
+    video = cv2.VideoCapture(file_location)
+    result = detect_faults(video)
+    return {"faults_detected": result}
+
+@app.post("/detect_thermal_anomalies/")
+async def detect_thermal_anomalies_from_image(file: UploadFile = File(...)):
+    # Save the uploaded thermal image
+    file_location = f"data/{file.filename}"
+    with open(file_location, "wb") as buffer:
+        buffer.write(file.file.read())
+    
+    # Load image and run thermal anomaly detection
+    result = detect_thermal_anomalies(file_location)
+    return {"thermal_anomalies": result}
+
+@app.post("/detect_energy_anomalies/")
+async def detect_energy_anomalies_from_data(file: UploadFile = File(...)):
+    # Save energy data file (CSV, JSON, etc.)
+    file_location = f"data/{file.filename}"
+    with open(file_location, "wb") as buffer:
+        buffer.write(file.file.read())
+    
+    # Run energy anomaly detection
+    result = detect_energy_anomalies(file_location)
+    return {"energy_anomalies": result}