Update app.py

app.py

@@ -1,42 +1,32 @@
-from flask import Flask, render_template, Response
-import cv2
-from ultralytics import YOLO
-
-app = Flask(__name__)
-
-# Load YOLOv8 model
-model = YOLO("yolov8n.pt")  # Make sure this file is present
-
-# Start webcam
-cap = cv2.VideoCapture(0)
-
-def generate_frames():
-    while True:
-        success, frame = cap.read()
-        if not success:
-            break
-
-        # Object detection
-        results = model.predict(frame, conf=0.5)
-        annotated_frame = results[0].plot()
-
-        # Encode frame to JPEG
-        ret, buffer = cv2.imencode('.jpg', annotated_frame)
-        frame = buffer.tobytes()
-
-        # Stream frame
-        yield (b'--frame\r\n'
-               b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
-
-# Main route with updated project name
-@app.route('/')
-def index():
-    project_name = "✨ Real Time Object Detection"
-    return render_template("index.html", project_name=project_name)
-
-@app.route('/video_feed')
-def video_feed():
-    return Response(generate_frames(), mimetype='multipart/x-mixed-replace; boundary=frame')
-
-if __name__ == '__main__':
-    app.run(debug=False)
+from flask import Flask, render_template, request, jsonify
+import cv2
+import numpy as np
+from ultralytics import YOLO
+import base64
+
+app = Flask(__name__)
+model = YOLO("yolov8n.pt")
+
+def readb64(base64_string):
+    nparr = np.frombuffer(base64.b64decode(base64_string.split(',')[1]), np.uint8)
+    return cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/detect', methods=['POST'])
+def detect():
+    data = request.get_json()
+    frame_data = data['image']
+
+    frame = readb64(frame_data)
+    results = model.predict(frame, conf=0.5)
+    annotated = results[0].plot()
+
+    _, buffer = cv2.imencode('.jpg', annotated)
+    annotated_b64 = base64.b64encode(buffer).decode('utf-8')
+    return jsonify({'image': f'data:image/jpeg;base64,{annotated_b64}'})
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)