Merge pull request #1 from SentryCoderDev/feature/pytesseract_plate

added feature/pytesseract

Author: SentryCoderDev

app.py

@@ -1,30 +1,30 @@
 from flask import Flask, render_template, Response, request, jsonify
 import cv2
+import pytesseract
 import time
 import threading
 import serial
 
-
 app = Flask(__name__)
 
-# Kameralar
+# All Cameras
 outside_garage_cam = cv2.VideoCapture(0)  
 inside_garage_cam = cv2.VideoCapture(1)   
 entrance_cam = cv2.VideoCapture(2)        
 
-# Seri port bağlantısı (Arduino)
 ser = serial.Serial('/dev/ttyUSB0', 115200)  # Arduino's serial port
-
 result_message = ""
 
-# Kullanıcı şifreleri
 user_passwords = {
     "user1": "1234",
     "user2": "5678"
     # Add more users
 }
 
-def gen_frames(cam):  # A generator function for streaming video
+# Pytesseract configuration
+pytesseract.pytesseract.tesseract_cmd = '/usr/bin/tesseract'  # Path to Tesseract OCR executable
+
+def gen_frames(cam):
     while True:
         success, frame = cam.read()
         if not success:
@@ -39,22 +39,25 @@ def garage_control():
     while True:
         ret, frame = outside_garage_cam.read()
         if ret:
-            # Motion detection and target plate recognition code will be here
-            # Send RF signal if target plate detected
-            ser.write(b'OPEN_GARAGE\n')
+            gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
+            # Perform OCR using Pytesseract
+            text = pytesseract.image_to_string(gray)
+            print("OCR Text:", text)
+            # Check if the target plate number is recognized
+            if "your_target_plate" in text:
+                ser.write(b'OPEN_GARAGE\n')
+                print("Opening garage for plate 'your_target_plate'")
         time.sleep(1)
 
 def audio_detection():
     while True:
         if detect_doorbell():
             print("ringtone detected!")
             # Start camera streaming at home entrance
-            # In this case, the user can be notified and view the camera
-            # In this section, you can send information to the client side using Flask routes.
+            # Notify and view the camera as needed
         time.sleep(1)
 
 def detect_doorbell():
-    # Here will be the ringtone detection algorithm
     return False
 
 def read_from_arduino():
@@ -78,7 +81,7 @@ def video_feed(cam_id):
 
 @app.route('/open_door', methods=['POST'])
 def open_door():
-    ser.write(b'OPEN_DOOR\n')  # Command to send signal to Arduino
+    ser.write(b'OPEN_DOOR\n')
     return "Door Opened", 200
 
 @app.route('/validate_password', methods=['POST'])
@@ -103,8 +106,8 @@ def index():
     return render_template('index.html')
 
 if __name__ == '__main__':
-    # Arka planda çalışan görevler
     threading.Thread(target=garage_control).start()
     threading.Thread(target=audio_detection).start()
     threading.Thread(target=read_from_arduino).start()
     app.run(host='0.0.0.0', port=5000, debug=True)
+

arduino/arduino_code/arduino_code.ino

@@ -6,16 +6,17 @@
 // RFID
 #define SS_PIN 10
 #define RST_PIN 9
+#define solenoidPin 11
 MFRC522 rfid(SS_PIN, RST_PIN);
 
 // Keypad
 const byte ROWS = 4; // Four rows
 const byte COLS = 3; // Three columns
 char keys[ROWS][COLS] = {
-  {'1', '2', '3'},
-  {'4', '5', '6'},
-  {'7', '8', '9'},
-  {'*', '0', '#'}
+    {'1', '2', '3'},
+    {'4', '5', '6'},
+    {'7', '8', '9'},
+    {'*', '0', '#'}
 };
 byte rowPins[ROWS] = {9, 8, 7, 6}; // Connect to the row pinouts of the keypad
 byte colPins[COLS] = {5, 4, 3}; // Connect to the column pinouts of the keypad
@@ -29,96 +30,98 @@ String userNames[] = {"X", "Y"};
 String enteredPassword = "";
 
 void setup() {
-  Serial.begin(115200);
-  SPI.begin();
-  rfid.PCD_Init();
+    Serial.begin(115200);
+    SPI.begin();
+    rfid.PCD_Init();
 }
 
 void loop() {
-  // process incoming commands from serial port
-  if (Serial.available() > 0) {
-    String command = Serial.readStringUntil('\n');
-    if (command == "OPEN_DOOR") {
-      openDoor();
+    // process incoming commands from serial port
+    if (Serial.available() > 0) {
+        String command = Serial.readStringUntil('\n');
+        if (command == "OPEN_DOOR") {
+            openDoor();
+        }
     }
-  }
 
-  // RFID Read
-  if (rfid.PICC_IsNewCardPresent() && rfid.PICC_ReadCardSerial()) {
-    String cardUID = "";
-    for (byte i = 0; i < rfid.uid.size; i++) {
-      cardUID += String(rfid.uid.uidByte[i], HEX);
+    // RFID Read
+    if (rfid.PICC_IsNewCardPresent() && rfid.PICC_ReadCardSerial()) {
+        String cardUID = "";
+        for (byte i = 0; i < rfid.uid.size; i++) {
+            cardUID += String(rfid.uid.uidByte[i], HEX);
+        }
+        if (isValidCard(cardUID)) {
+            String user = getUserByCard(cardUID);
+            openDoor();
+            Serial.print("Hello ");
+            Serial.println(user);
+            sendResultToJetson("Hello " + user);
+        }
+        rfid.PICC_HaltA();
     }
-    if (isValidCard(cardUID)) {
-      String user = getUserByCard(cardUID);
-      openDoor();
-      Serial.print("Hello ");
-      Serial.println(user);
-      sendResultToJetson("Hello " + user);
-    }
-    rfid.PICC_HaltA();
-  }
 
-  // Keypad Read
-  char key = keypad.getKey();
-  if (key) {
-    if (key == '#') {
-      if (isValidPassword(enteredPassword)) {
-        String user = getUserByPassword(enteredPassword);
-        openDoor();
-        Serial.print("Hello ");
-        Serial.println(user);
-        sendResultToJetson("Hello " + user);
-      }
-      enteredPassword = "";
-    } else {
-      enteredPassword += key;
+    // Keypad Read
+    char key = keypad.getKey();
+    if (key) {
+        if (key == '#') {
+            if (isValidPassword(enteredPassword)) {
+                String user = getUserByPassword(enteredPassword);
+                openDoor();
+                Serial.print("Hello ");
+                Serial.println(user);
+                sendResultToJetson("Hello " + user);
+            }
+            enteredPassword = "";
+        } else {
+            enteredPassword += key;
+        }
     }
-  }
 }
 
 bool isValidCard(String uid) {
-  for (String validCard : validCards) {
-    if (validCard == uid) {
-      return true;
+    for (String validCard : validCards) {
+        if (validCard == uid) {
+            return true;
+        }
     }
-  }
-  return false;
+    return false;
 }
 
 bool isValidPassword(String password) {
-  for (String userPassword : userPasswords) {
-    if (userPassword == password) {
-      return true;
+    for (String userPassword : userPasswords) {
+        if (userPassword == password) {
+            return true;
+        }
     }
-  }
-  return false;
+    return false;
 }
 
 String getUserByCard(String uid) {
-  for (int i = 0; i < sizeof(validCards) / sizeof(validCards[0]); i++) {
-    if (validCards[i] == uid) {
-      return userNames[i];
+    for (int i = 0; i < sizeof(validCards) / sizeof(validCards[0]); i++) {
+        if (validCards[i] == uid) {
+            return userNames[i];
+        }
     }
-  }
-  return "Unknown";
+    return "Unknown";
 }
 
 String getUserByPassword(String password) {
-  for (int i = 0; i < sizeof(userPasswords) / sizeof(userPasswords[0]); i++) {
-    if (userPasswords[i] == password) {
-      return userNames[i];
+    for (int i = 0; i < sizeof(userPasswords) / sizeof(userPasswords[0]); i++) {
+        if (userPasswords[i] == password) {
+            return userNames[i];
+        }
     }
-  }
-  return "Unknown";
+    return "Unknown";
 }
 
 void openDoor() {
-  // Solenoidi açma kodu burada olacak
-  // Örneğin: digitalWrite(solenoidPin, HIGH);
+   
+    digitalWrite(solenoidPin, HIGH);
+    sleep(1);
 }
 
 void sendResultToJetson(String message) {
-  Serial.print("RESULT:");
-  Serial.println(message);
+    Serial.print("RESULT:");
+    Serial.println(message);
 }
+