test

Dockerfile

@@ -0,0 +1,14 @@
+# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]

app.py

@@ -0,0 +1,163 @@
+from flask import Flask, Response, request, jsonify
+import cv2
+from flask_cors import CORS 
+import pytesseract
+import numpy as np
+import os
+import uuid
+
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
+
+uri = "mongodb+srv://cse443Prudhvi:pEjVbWv6oJHaHSJA@cluster0.7ournot.mongodb.net/"
+# Create a new client and connect to the server
+
+
+cluster = MongoClient(uri)
+db = cluster['cseData']  # Replace '<dbname>' with your actual database name
+collection = db['numberPlates'] 
+
+
+
+FRAME_WIDTH = 640
+FRAME_HEIGHT = 480
+PLATE_CASCADE = cv2.CascadeClassifier('./indian_license_plate.xml')
+MIN_AREA = 300
+COLOR = (255, 0, 255) 
+
+
+# Declare a variable with the type ndarray[Any, dtype[generic]]
+img_roi: np.ndarray = np.array([1, 2, 3])
+
+app = Flask(__name__)
+CORS(app)
+counter = 0  # Counter for image filenames
+ 
+def generate_frames(n):
+    cap = cv2.VideoCapture(0) 
+    
+    global img_roi
+    while cap.isOpened():
+        # read the camera frame
+        success, frame = cap.read()
+        
+        if not success:
+            break
+        else:
+            img_gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+            number_plates = PLATE_CASCADE.detectMultiScale(img_gray, 1.3, 7)
+
+            # Iterate through detected plates
+            
+            for (x, y, w, h) in number_plates:
+                area = (w * h)+100
+                if area > MIN_AREA:
+                    # Draw rectangle around the plate
+                    cv2.rectangle(frame, (x, y), (x + w, y + h), COLOR, 2)
+                    # Add text label
+                    cv2.putText(frame, "License Plate", (x, y - 5), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, COLOR, 2)
+                    # Show region of interest (ROI)
+                    img_roi = frame[y:y + h , x:x + w ]
+                    
+
+        ret, buffer = cv2.imencode('.jpg', frame)
+        frame_bytes = buffer.tobytes()
+        
+        yield (b'--frame\r\n'
+               b'Content-Type: image/jpeg\r\n\r\n' + frame_bytes + b'\r\n')
+@app.route('/')
+def index():
+    return "Hello World"
+
+
+@app.route('/video')
+def video():
+    return Response(generate_frames(0), mimetype='multipart/x-mixed-replace; boundary=frame')
+
+
+
+
+@app.route('/save', methods=['POST'])
+def save():
+
+    global counter
+    global img_roi
+    cv2.imwrite(f"/Users/prudhviraj/Documents/CSE443/img/{counter}.png", img_roi)
+
+    image = f"/Users/prudhviraj/Documents/CSE443/img/{counter}.png"
+    pytesseract.pytesseract.tesseract_cmd = '/opt/homebrew/bin/tesseract'
+    extracted_text = pytesseract.image_to_string(image)
+
+    new_text=''
+    alpha="ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+    num="0123456789"
+    for i in extracted_text:
+        if i in alpha or i in num:
+            new_text+=i
+    print(new_text)
+    print("Yes!!!!!")
+    # collection.insert_one({"number":f"{new_text}"})
+    
+    counter += 1
+    return new_text
+
+UPLOAD_FOLDER = '/Users/prudhviraj/Documents/CSE443/img'
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
+
+app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
+@app.route('/upload', methods=['POST'])
+def upload_image():
+    # check if the post request has the file part
+    if 'image' not in request.files:
+        return jsonify({"error": "No image part in the request"}), 400
+
+    image_file = request.files['image']
+
+    # if user does not select file, browser also submits an empty part without filename
+    if image_file.filename == '':
+        return jsonify({"error": "No selected image file"}), 400
+
+    if image_file and allowed_file(image_file.filename):
+        filename = str(uuid.uuid4()) + '.' + image_file.filename.rsplit('.', 1)[1].lower()
+        filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+        image_file.save(filepath)
+        
+        print(filepath)
+        frame = cv2.imread(filepath)
+        # success, frame = image.read()
+
+        img_gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+        number_plates = PLATE_CASCADE.detectMultiScale(img_gray, 1.3, 7)
+
+        for (x, y, w, h) in number_plates:
+                area = w * h
+                if area > MIN_AREA:
+                    # Draw rectangle around the plate
+                    cv2.rectangle(frame, (x, y), (x + w, y + h), COLOR, 2)
+                    # Add text label
+                    cv2.putText(frame, "License Plate", (x, y - 5), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1, COLOR, 2)
+                    # Show region of interest (ROI)
+                    img_roi = frame[y:y + h, x:x + w]
+        pytesseract.pytesseract.tesseract_cmd = '/opt/homebrew/bin/tesseract'
+        extracted_text = pytesseract.image_to_string(img_roi)
+
+        new_text=''
+        alpha="ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+        num="0123456789"
+        for i in extracted_text:
+            if i in alpha or i in num:
+                new_text+=i
+        print(new_text)
+        return jsonify({"success": "Image uploaded successfully", "filename": filename}), 200
+
+    else:
+        return jsonify({"error": "File type not allowed"}), 400
+
+
+
+if __name__ == '__main__':
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,16 @@
+blinker==1.7.0
+click==8.1.7
+dnspython==2.6.1
+Flask==3.0.3
+Flask-Cors==4.0.0
+gunicorn==22.0.0
+itsdangerous==2.1.2
+Jinja2==3.1.3
+MarkupSafe==2.1.5
+numpy==1.26.4
+opencv-contrib-python==4.9.0.80
+packaging==24.0
+pillow==10.3.0
+pymongo==4.6.3
+pytesseract==0.3.10
+Werkzeug==3.0.2