added links for test videos

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Wild Fire Tracker
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

app.py

@@ -1,3 +1,7 @@
+# Wild Fire Tracker - Fire Detection MCP Server
+# Copyright (c) 2024 Wild Fire Tracker
+# Licensed under MIT License - see LICENSE file for details
+
 import gradio as gr
 import cv2
 import numpy as np
@@ -24,6 +28,7 @@ class FireDetectionMCP:
         self.status_color = "#808080"  # Gray
         self.last_analysis_time = 0
         self.frame_count = 0
+        self.last_detection_time = None
         
     def analyze_frame(self, frame):
         """Analyze frame for fire/smoke"""
@@ -43,8 +48,8 @@ class FireDetectionMCP:
             answer = vqa_processor.decode(outputs[0], skip_special_tokens=True).lower()
             
             # Check for fire/smoke
-            fire_keywords = ['fire', 'flame', 'burning', 'blaze']
-            smoke_keywords = ['smoke', 'smoky', 'vapor']
+            fire_keywords = ['fire', 'flame', 'burning', 'blaze', 'flames']
+            smoke_keywords = ['smoke', 'smoky', 'vapor', 'fog', 'mist', 'haze', 'cloud', 'grey', 'gray']
             
             if any(word in answer for word in fire_keywords):
                 return "🔥 FIRE DETECTED", "#FF4500"  # Orange
@@ -97,6 +102,7 @@ class FireDetectionMCP:
                 print(f"[{datetime.now().strftime('%H:%M:%S')}] Analyzing frame {self.frame_count}...")
                 self.status, self.status_color = self.analyze_frame(frame)
                 self.last_analysis_time = current_time
+                self.last_detection_time = datetime.now().strftime('%H:%M:%S')
                 print(f"[{datetime.now().strftime('%H:%M:%S')}] Result: {self.status}")
             
             # Add status overlay
@@ -115,8 +121,9 @@ class FireDetectionMCP:
             cv2.putText(display_frame, self.status, (10, 40), 
                        cv2.FONT_HERSHEY_SIMPLEX, 1.0, color, 2)
             
-            # Add frame counter
-            cv2.putText(display_frame, f"Frame: {self.frame_count}", (10, 470), 
+            # Add timestamp
+            timestamp = datetime.now().strftime('%H:%M:%S')
+            cv2.putText(display_frame, f"Time: {timestamp}", (10, 470), 
                        cv2.FONT_HERSHEY_SIMPLEX, 0.6, (255, 255, 255), 1)
             
             # Store frame
@@ -162,6 +169,8 @@ class FireDetectionMCP:
     
     def get_status(self):
         """Get current status"""
+        if self.last_detection_time:
+            return f"{self.status} (Last check: {self.last_detection_time})"
         return self.status
 
 # Initialize MCP server
@@ -173,8 +182,9 @@ def create_interface():
     with gr.Blocks(title="🔥 Fire Detection MCP Server", theme=gr.themes.Soft()) as interface:
         gr.Markdown("# 🔥 Fire Detection MCP Server")
         gr.Markdown("Real-time fire and smoke detection from video streams (analyzes every 10 seconds)")
-        gr.Markdown("⚠️ **Usage**: Upload your own video file or use live sources (webcam/RTSP). Webcam may not work on HF Spaces.")
-        
+        gr.Markdown("⚠️ **Usage**: Upload your own video file or use live sources (webcam/RTSP). It may take few seconds to load stream and show analysis. Webcam may not work on HF Spaces.")
+        gr.Markdown("🔗 **Sample Videos**: [Fire Test Video](https://www.pexels.com/video/a-man-carrying-gear-walking-away-from-a-controlled-fire-8552246/) | [Smoke Test Video](https://www.pexels.com/video/aerial-view-of-controlled-forest-fire-in-spring-31361444/)")
+
         with gr.Row():
             with gr.Column(scale=1):
                 gr.Markdown("### Video Source Options")
@@ -186,7 +196,7 @@ def create_interface():
                             file_types=[".mp4", ".avi", ".mov"],
                             type="filepath"
                         )
-                        upload_btn = gr.Button("🚀 Start from Upload", variant="primary")
+                        upload_btn = gr.Button("🚀 Start Monitoring", variant="primary")
                     
                     with gr.Tab("📹 Live Sources"):
                         video_input = gr.Textbox(
@@ -194,7 +204,7 @@ def create_interface():
                             placeholder="0 (webcam), rtsp://url, or path/to/video.mp4",
                             value="0"
                         )
-                        live_btn = gr.Button("🚀 Start Live Source", variant="primary")
+                        live_btn = gr.Button("🚀 Start Monitoring", variant="primary")
                 
                 stop_btn = gr.Button("🛑 Stop Monitoring", variant="secondary")