Update app.py

app.py

@@ -33,11 +33,11 @@ def detect_and_draw_image(input_image):
 
     return input_image, df
 
-# دالة لاكتشاف الكائنات في الفيديو
 def detect_and_draw_video(video_path):
     cap = cv2.VideoCapture(video_path)
     frames = []
     overall_counts = {}
+    seen_objects = []  # قائمة لتتبع الكائنات التي تم اكتشافها
 
     while cap.isOpened():
         ret, frame = cap.read()
@@ -51,15 +51,24 @@ def detect_and_draw_video(video_path):
         for detection in detections:
             xmin, ymin, xmax, ymax, conf, class_id = detection
             label = model.names[int(class_id)]
-            overall_counts[label] = overall_counts.get(label, 0) + 1
-            
-            cv2.rectangle(frame, (int(xmin), int(ymin)), (int(xmax), int(ymax)), (255, 0, 0), 2)
-            cv2.putText(frame, f"{label}: {conf:.2f}", (int(xmin), int(ymin) - 10), 
-                        cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 2)
+            current_object = (label, int(xmin), int(ymin), int(xmax), int(ymax))
+
+            # التحقق من وجود الكائن في قائمة seen_objects
+            if not any(existing[0] == label and
+                       (existing[1] < xmax and existing[3] > xmin and
+                        existing[2] < ymax and existing[4] > ymin) for existing in seen_objects):
+                seen_objects.append(current_object)
+                overall_counts[label] = overall_counts.get(label, 0) + 1
+
+                # رسم المستطيل والكلمات على الإطار
+                cv2.rectangle(frame, (int(xmin), int(ymin)), (int(xmax), int(ymax)), (255, 0, 0), 2)
+                cv2.putText(frame, f"{label}: {conf:.2f}", (int(xmin), int(ymin) - 10), 
+                            cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 2)
 
         frames.append(frame)
 
     cap.release()
+
     output_path = 'output.mp4'
     out = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*'mp4v'), 20.0, (640, 480))
 
@@ -67,7 +76,10 @@ def detect_and_draw_video(video_path):
         out.write(frame)
     out.release()
 
+    # ترجمة التسميات إلى العربية
     translated_labels = translator(list(overall_counts.keys()))
+
+    # إنشاء DataFrame لتخزين النتائج
     df = pd.DataFrame({
         'Label (English)': list(overall_counts.keys()),
         'Label (Arabic)': [t['translation_text'] for t in translated_labels],
@@ -76,6 +88,7 @@ def detect_and_draw_video(video_path):
 
     return output_path, df
 
+
 # واجهة صورة
 image_interface = gr.Interface(
     fn=detect_and_draw_image,