Update app.py

app.py

@@ -1,38 +1,62 @@
 import gradio as gr
 from ultralytics import YOLO
 import cv2
+import tempfile
+import os
 
-# Model load (automatic download hoga)
+# Model load
 model = YOLO('yolov8n.pt')
 
 def count_people(video):
-    # Video process karo
-    cap = cv2.VideoCapture(video)
-    entry_count = 0
+    if video is None:
+        return "Please upload a video!"
     
-    while cap.isOpened():
-        ret, frame = cap.read()
-        if not ret:
-            break
+    try:
+        # Video process karo
+        cap = cv2.VideoCapture(video)
+        total_detections = 0
+        frame_count = 0
         
-        # Person detect karo
-        results = model.track(frame, classes=[0], persist=True)
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret:
+                break
+            
+            frame_count += 1
+            
+            # Har 5th frame process karo (speed ke liye)
+            if frame_count % 5 == 0:
+                # Person detect karo
+                results = model(frame, classes=[0], verbose=False)
+                
+                # Count detections
+                if len(results[0].boxes) > 0:
+                    total_detections += len(results[0].boxes)
         
-        # Counting logic (simple version)
-        if results[0].boxes:
-            entry_count = len(results[0].boxes)
+        cap.release()
+        
+        avg_people = total_detections / (frame_count / 5) if frame_count > 0 else 0
+        
+        return f"""
+        📊 Detection Results:
+        - Total Frames Processed: {frame_count}
+        - Total Detections: {total_detections}
+        - Average People per Frame: {avg_people:.2f}
+        """
     
-    cap.release()
-    return f"Total People Detected: {entry_count}"
+    except Exception as e:
+        return f"Error processing video: {str(e)}"
 
 # Gradio interface
 demo = gr.Interface(
     fn=count_people,
-    inputs=gr.Video(),
-    outputs="text",
-    title="Guest Entry/Exit Management System",
-    description="Upload a video to count people"
+    inputs=gr.Video(label="Upload Video"),
+    outputs=gr.Textbox(label="Results"),
+    title="🚪 Guest Entry/Exit Management System",
+    description="Upload a video to detect and count people using YOLOv8",
+    examples=[],
+    theme=gr.themes.Soft()
 )
 
-demo.launch()
-```
+if __name__ == "__main__":
+    demo.launch()