Update app.py

app.py

@@ -1,25 +1,43 @@
-# Import necessary libraries
-import warnings
-from ultralytics import YOLO  # Ensure ultralytics is installed
-from transformers import pipeline, DetrImageProcessor
-
-# Suppress unnecessary warnings
-warnings.filterwarnings("ignore", message=".*copying from a non-meta parameter.*")
-
-# Setup the image processor
-processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50", use_fast=True)
-
-# Load the YOLO model (make sure ultralytics is installed)
-model = YOLO("yolov5s.pt")  # Change to the model you are using
-
-# Your model inference or other operations
-# Example detection function
-def detect_objects(image):
-    # Use your model for object detection
-    results = model(image)
-    return results
-
-# Call the function
-image = "path_to_image.jpg"  # Replace with your image path
-detection_results = detect_objects(image)
-print(detection_results)
+import gradio as gr
+import os, sys
+import cv2
+
+# Ensure services is importable
+sys.path.append(os.path.join(os.path.dirname(__file__), "services"))
+
+from video_service import get_video_frame
+from detection_service import detect_objects
+from thermal_service import detect_thermal_anomalies
+from shadow_detection import detect_shadow_coverage
+from salesforce_dispatcher import send_to_salesforce
+
+frame_generator = get_video_frame("data/sample_pole_video.mp4")
+
+def detect_frame():
+    try:
+        frame = next(frame_generator)
+        temp_path = "temp.jpg"
+        cv2.imwrite(temp_path, frame)
+
+        objects = detect_objects(temp_path)
+        thermal = detect_thermal_anomalies(temp_path)
+        shadow = detect_shadow_coverage(temp_path)
+
+        alert = {
+            "detections": objects,
+            "thermal": bool(thermal),
+            "shadow_issue": shadow,
+        }
+
+        send_to_salesforce(alert)
+        return frame
+    except StopIteration:
+        return None
+
+gr.Interface(
+    fn=detect_frame,
+    inputs=[],
+    outputs="image",
+    live=True,
+    title="VIEP Smart Pole Live Fault Detection"
+).launch()