Update app.py

app.py

@@ -43,7 +43,7 @@ def live_feed_generator(video_type, confidence_threshold=0.9):
     video_path = VIDEO_PATHS.get(video_type)
     
     if not video_path or not os.path.exists(video_path):
-        yield gr.update(value="Video file not found."), None, None, None, None, None
+        yield gr.update(value="Video file not found."), None, None, None, None, None, None
         return
 
     cap = cv2.VideoCapture(video_path)
@@ -114,21 +114,26 @@ def live_feed_generator(video_type, confidence_threshold=0.9):
             box = detection["box"]
             coords = f"[{box['xmin']},{box['ymin']},{box['xmax']},{box['ymax']}]"
             metrics.append(coords)
-        metrics_str = f"anomalies: {metrics}, total_detected: {state.total_detected}"
+        metrics_str = f"Coordinates: {metrics}\nTotal Detected: {state.total_detected}"
 
-        # Generate detection trend plot
+        # Generate detection trend plot with dark theme
+        plt.style.use('dark_background')
         plt.figure(figsize=(4, 2))
-        plt.plot(list(state.anomaly_history), marker='o')
-        plt.title("Anomalies Over Time")
-        plt.xlabel("Frame")
-        plt.ylabel("Count")
-        plt.grid(True)
+        plt.plot(list(state.anomaly_history), marker='o', color='yellow')
+        plt.title("Anomalies Over Time", color='white')
+        plt.xlabel("Frame", color='white')
+        plt.ylabel("Count", color='white')
+        plt.grid(True, color='gray')
+        plt.tick_params(colors='white')
         trend_plot = plt.gcf()
         plt.close()
 
         # Generate anomaly types summary
         anomaly_types_str = "\n".join([f"{k}: {v}" for k, v in state.anomaly_types.items()])
 
+        # Update timestamp
+        timestamp_str = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
         # Yield updated UI components
         yield (
             gr.update(value=annotated_frame_rgb),           # Live Video Feed
@@ -136,7 +141,8 @@ def live_feed_generator(video_type, confidence_threshold=0.9):
             gr.update(value="\n".join(state.logs)),        # Live Logs
             gr.update(value=trend_plot),                    # Detection Trend
             gr.update(value=anomaly_types_str),            # Anomaly Types
-            gr.update(value=list(state.captured_events))    # Captured Events
+            gr.update(value=list(state.captured_events)),   # Captured Events
+            gr.update(value=timestamp_str)                 # Timestamp
         )
 
         # Simulate real-time by sleeping between frames
@@ -144,33 +150,86 @@ def live_feed_generator(video_type, confidence_threshold=0.9):
 
     cap.release()
 
+# Custom CSS for dark theme and styling
+custom_css = """
+body, .gradio-container {
+    background-color: #1a1a1a !important;
+    color: white !important;
+    font-family: Arial, sans-serif !important;
+}
+h1, h2, h3, label {
+    color: white !important;
+    font-weight: bold !important;
+}
+.gradio-row, .gradio-column {
+    background-color: #2b2b2b !important;
+    border-radius: 8px !important;
+    padding: 10px !important;
+    margin: 5px !important;
+}
+#live-feed {
+    border: 2px solid #444 !important;
+    border-radius: 8px !important;
+}
+#live-metrics, #live-logs, #anomaly-types {
+    background-color: #333 !important;
+    color: white !important;
+    border: 1px solid #555 !important;
+    border-radius: 8px !important;
+    padding: 10px !important;
+    height: 100px !important;
+    overflow-y: auto !important;
+}
+#detection-trend, #captured-events {
+    background-color: #333 !important;
+    border: 1px solid #555 !important;
+    border-radius: 8px !important;
+    padding: 10px !important;
+}
+#status-indicator {
+    color: #00ff00 !important;
+    font-size: 14px !important;
+}
+#timestamp {
+    font-size: 16px !important;
+    color: #cccccc !important;
+}
+"""
+
 # Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# Solar Panel Monitoring System")
+with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown("# Fault Detection")
+    timestamp = gr.Textbox(label="", value=datetime.now().strftime("%Y-%m-%d %H:%M:%S"), elem_id="timestamp")
+    
     with gr.Row():
-        with gr.Column():
-            video_type = gr.Dropdown(
-                choices=["Day Feed", "Night Feed", "Thermal Feed", "Shadow/Dust Feed"],
-                label="Select Drone Feed",
-                value="Thermal Feed"
+        # Left Panel: Live Feed and Controls
+        with gr.Column(scale=7):
+            with gr.Row():
+                video_type = gr.Dropdown(
+                    choices=["Day Feed", "Night Feed", "Thermal Feed", "Shadow/Dust Feed"],
+                    label="Select Drone Feed",
+                    value="Thermal Feed"
+                )
+                confidence_threshold = gr.Slider(0.5, 1.0, value=0.9, label="Confidence Threshold")
+                start_button = gr.Button("Start Live Feed")
+            live_feed = gr.Image(label="Live Video Feed", streaming=True, elem_id="live-feed")
+            status_indicator = gr.HTML(
+                '<p id="status-indicator">Status: <span style="color: green;">Running</span> •</p>',
+                label=""
             )
-            confidence_threshold = gr.Slider(0.5, 1.0, value=0.9, label="Confidence Threshold")
-            start_button = gr.Button("Start Live Feed")
-        with gr.Column():
-            live_feed = gr.Image(label="Live Video Feed", streaming=True)
-    with gr.Row():
-        with gr.Column():
-            live_metrics = gr.Textbox(label="Live Metrics")
-            live_logs = gr.Textbox(label="Live Logs")
-        with gr.Column():
-            detection_trend = gr.Plot(label="Detection Trend")
-            anomaly_types = gr.Textbox(label="Anomaly Types")
-    captured_events = gr.Gallery(label="Captured Events (Last 5)")
+        
+        # Right Panel: Analytics
+        with gr.Column(scale=3):
+            live_metrics = gr.Textbox(label="Live Metrics", elem_id="live-metrics")
+            live_logs = gr.Textbox(label="Live Logs", elem_id="live-logs")
+            detection_trend = gr.Plot(label="Detection Trend", elem_id="detection-trend")
+            anomaly_types = gr.Textbox(label="Anomaly Types", elem_id="anomaly-types")
+            captured_events = gr.Gallery(label="Captured Events (Last 5)", elem_id="captured-events")
 
     start_button.click(
         fn=live_feed_generator,
         inputs=[video_type, confidence_threshold],
-        outputs=[live_feed, live_metrics, live_logs, detection_trend, anomaly_types, captured_events]
+        outputs=[live_feed, live_metrics, live_logs, detection_trend, anomaly_types, captured_events, timestamp]
     )
 
 demo.launch()