SuriRaja commited on
Commit
9c4a4fe
·
1 Parent(s): 796b5c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -1
app.py CHANGED
@@ -1,3 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
1
  frame_gen = get_video_frame("data/drone_day.mp4")
2
 
3
  def monitor_feed():
@@ -5,11 +16,12 @@ def monitor_feed():
5
  try:
6
  frame = next(frame_gen)
7
  except StopIteration:
8
- # Auto-reload video when finished
9
  frame_gen = get_video_frame("data/drone_day.mp4")
10
  frame = next(frame_gen)
11
 
12
  cv2.imwrite("temp.jpg", frame)
 
13
  detections = detect_objects("temp.jpg")
14
  thermal = detect_thermal_anomalies("temp.jpg")
15
  shadow_flag = detect_shadow_coverage("temp.jpg")
@@ -21,3 +33,14 @@ def monitor_feed():
21
  }
22
  send_to_salesforce(alert_payload)
23
  return frame
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import cv2
3
+
4
+ # ✅ Import all services before calling anything
5
+ from services.video_service import get_video_frame
6
+ from services.detection_service import detect_objects
7
+ from services.thermal_service import detect_thermal_anomalies
8
+ from services.shadow_detection import detect_shadow_coverage
9
+ from services.salesforce_dispatcher import send_to_salesforce
10
+
11
+ # ✅ Initialize frame generator only after imports
12
  frame_gen = get_video_frame("data/drone_day.mp4")
13
 
14
  def monitor_feed():
 
16
  try:
17
  frame = next(frame_gen)
18
  except StopIteration:
19
+ # Auto-reload video when it ends
20
  frame_gen = get_video_frame("data/drone_day.mp4")
21
  frame = next(frame_gen)
22
 
23
  cv2.imwrite("temp.jpg", frame)
24
+
25
  detections = detect_objects("temp.jpg")
26
  thermal = detect_thermal_anomalies("temp.jpg")
27
  shadow_flag = detect_shadow_coverage("temp.jpg")
 
33
  }
34
  send_to_salesforce(alert_payload)
35
  return frame
36
+
37
+ # ✅ Setup Gradio interface
38
+ iface = gr.Interface(
39
+ fn=monitor_feed,
40
+ inputs=[],
41
+ outputs="image",
42
+ live=True,
43
+ title="Solar Surveillance Feed Simulation"
44
+ )
45
+
46
+ iface.launch()