Update app.py

app.py

@@ -368,7 +368,8 @@ def process_image(image):
         processed_frame = preprocess_frame(frame)
         start_time = time.time()
         results = yolo_model.predict(processed_frame)
-        annotated_frame = draw_bounding_boxes(frame.copy(), results)
+        # Skip drawing bounding boxes as per request; return the original frame
+        # annotated_frame = draw_bounding_boxes(frame.copy(), results)
         processing_time = time.time() - start_time
         logger.info(f"Image processing completed in {processing_time:.2f} seconds")
         
@@ -424,7 +425,7 @@ def process_image(image):
         if salesforce_errors:
             status_message += f" Salesforce logging errors: {', '.join(salesforce_errors)}"
         
-        return annotated_frame, status_message, generate_pdf_report(), recent_violations
+        return frame, status_message, generate_pdf_report(), recent_violations
     except Exception as e:
         logger.error(f"Image processing error: {e}")
         return None, f"Error processing image: {str(e)}", None, recent_violations
@@ -447,8 +448,9 @@ def process_rtsp_stream():
         for frame, timestamp in capture_rtsp_frames(RTSP_URL, max_frames=10):
             processed_frame = preprocess_frame(frame)
             results = yolo_model.predict(processed_frame)
-            annotated_frame = draw_bounding_boxes(frame.copy(), results)
-            frames.append(annotated_frame)
+            # Skip drawing bounding boxes as per request; append the original frame
+            # annotated_frame = draw_bounding_boxes(frame.copy(), results)
+            frames.append(frame)
             
             # Log violations if detected
             for result in results:
@@ -514,7 +516,7 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         with gr.Column():
-            image_output = gr.Image(label="Processed Image (with Bounding Boxes)")
+            image_output = gr.Image(label="Processed Image")
         with gr.Column():
             image_status = gr.Textbox(label="Image Processing Status")