Update app.py

app.py

@@ -26,8 +26,7 @@ def process_video(video_path):
     fourcc = cv2.VideoWriter_fourcc(*"mp4v")
     out = cv2.VideoWriter(output_path, fourcc, fps, (frame_width, frame_height))
 
-    total_person_count = 0
-
+    # Proses video
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
@@ -41,7 +40,6 @@ def process_video(video_path):
         # Filter hanya label 'person'
         person_detections = [d for d in detections if names[int(d[-1])] == "person"]
         person_count = len(person_detections)
-        total_person_count += person_count
 
         # Render frame dan buat salinan eksplisit
         annotated_frame = results.render()[0]
@@ -58,20 +56,21 @@ def process_video(video_path):
     cap.release()
     out.release()
 
-    return output_path, total_person_count
+    # Mengembalikan video yang telah diproses (tidak menjumlahkan seluruh frame)
+    return output_path
 
 # Fungsi Gradio untuk antarmuka
 def gradio_interface(video_file):
-    output_path, total_person_count = process_video(video_file)
-    return output_path, f"Total people detected: {total_person_count}"
+    output_path = process_video(video_file)
+    return output_path
 
 # Antarmuka Gradio
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=gr.File(type="filepath"),  # Input berupa file video
-    outputs=[gr.File(label="Processed Video"), gr.Text(label="Total People Count")],
+    outputs=gr.File(label="Processed Video"),
     title="Person Counter using YOLOv5",
-    description="Upload a video file to detect and count the number of people using YOLOv5."
+    description="Upload a video file to detect and count the number of people in each frame using YOLOv5."
 )
 
 # Menjalankan aplikasi