Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from ultralytics import YOLO
 from PIL import Image
+import os
 
 # Load YOLO model
 model = YOLO("Suspicious_Activities_nano.pt")
@@ -8,42 +9,48 @@ model = YOLO("Suspicious_Activities_nano.pt")
 # Define the prediction function
 def predict_suspicious_activity(image):
     results = model.predict(source=image, show=False, conf=0.6)
-    results_img = results[0].plot()  # Get the image with bounding boxes/annotations
+    results_img = results[0].plot()
     return Image.fromarray(results_img)
 
-# Create Gradio interface with enhanced UI
+# Create a temporary directory for examples if needed
+os.makedirs("examples", exist_ok=True)
+
+# Create Gradio interface with reliable example images
 interface = gr.Interface(
     fn=predict_suspicious_activity,
-    inputs=gr.Image(type="pil", label="Upload Security Footage", height=350),
-    outputs=gr.Image(type="pil", label="Detection Results", height=350),
-    title="🚨 Advanced Suspicious Activity Detection System",
+    inputs=gr.Image(type="pil", label="Security Camera Feed", height=380),
+    outputs=gr.Image(type="pil", label="Threat Analysis", height=380),
+    title="🛡️ AI-Powered Suspicious Activity Detection System",
     description="""
-    **Real-time threat identification using YOLOv8 technology**  
-    Our system detects multiple suspicious behaviors including:
-    - Loitering and unauthorized presence [[7]]
-    - Physical altercations and fighting
-    - Trespassing in restricted areas [[7]]
-    - Vandalism and property damage
+    **Advanced security monitoring using YOLOv8 technology**  
+    Our system detects critical security threats including:
+    - Unauthorized loitering in restricted zones
+    - Trespassing incidents captured in surveillance footage
+    - Vandalism and property damage activities
     - Suspicious package handling
+    - Physical confrontations
     
-    Upload your image or try our pre-loaded security footage examples below!
+    Upload your security footage or test with the examples below:
     """,
     examples=[
-        ["https://cdn.pixabay.com/photo/2016/08/11/23/36/car-1587281_1280.jpg"],  # Car break-in attempt
-        ["https://cdn.pixabay.com/photo/2017/08/06/12/18/man-2595256_1280.jpg"],  # Suspicious loitering
-        ["https://cdn.pixabay.com/photo/2016/08/11/23/35/security-1587280_1280.jpg"]  # Security monitoring
+        ["https://raw.githubusercontent.com/ultralytics/ultralytics/main/ultralytics/assets/bus.jpg"],
+        ["https://raw.githubusercontent.com/ultralytics/ultralytics/main/ultralytics/assets/zidane.jpg"],
+        ["https://ultralytics.com/images/bus.jpg"]
     ],
     cache_examples=False,
-    theme=gr.themes.Soft(primary_hue="indigo", font=[gr.themes.GoogleFont('Inconsolata'), 'monospace']),
-    css=".footer {position: fixed; bottom: 0; text-align: center; width: 100%; padding: 10px; background: #f0f0f0;}"
+    theme=gr.themes.Soft(
+        primary_hue="amber",
+        font=[gr.themes.GoogleFont('IBM Plex Sans'), 'system-ui']
+    ),
+    css=".footer {margin-top: 20px; padding: 15px; border-top: 1px solid #eee; text-align: center; color: #666;} footer {visibility: hidden}"
 )
 
-# Launch the interface with professional configuration
+# Launch with professional configuration
 interface.launch(
     share=True,
-    debug=False,
     server_port=7860,
     show_api=False,
     enable_queue=True,
-    favicon_path="https://raw.githubusercontent.com/ultralytics/assets/main/logo-Ultralytics.svg"
+    favicon_path="https://raw.githubusercontent.com/ultralytics/assets/main/logo-Ultralytics.svg",
+    analytics_enabled=False
 )