Update app.py

app.py

@@ -1,122 +1,114 @@
-import os
-import cv2
-from paddleocr import PaddleOCR
-import re
-from flask import Flask, request, jsonify
-from PIL import Image
-import pandas as pd
-import requests
-from datetime import datetime
-import random
-
-app = Flask(__name__)
-app.config['UPLOAD_FOLDER'] = 'uploads/'
-
-os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
-
-def convert_image_format(input_path, output_format='PNG'):
-    try:
-        image = Image.open(input_path)
-        if hasattr(image, '_getexif'):
-            exif = image._getexif()
-            if exif and 0x112 in exif:
-                orientation = exif[0x112]
-                rotations = {3: 180, 6: 270, 8: 90}
-                if orientation in rotations:
-                    image = image.rotate(rotations[orientation], expand=True)
-        output_path = os.path.splitext(input_path)[0] + '.' + output_format.lower()
-        image.convert("RGB").save(output_path, format=output_format)
-        return output_path
-    except Exception as e:
-        return input_path
-
-def extract_text_with_paddleocr(file_path):
-    ocr = PaddleOCR(use_angle_cls=True, lang='en', show_log=False)
-    result = ocr.ocr(file_path, cls=True)
-    return result
-
-def extract_numbers_from_paddleocr(result):
-    detected_numbers = []
-    for line in result:
-        for res in line:
-            text = res[1][0]
-            numbers = re.findall(r'\d+', text)
-            if numbers:
-                number_str = ''.join(numbers)
-                detected_numbers.append(number_str)
-    return detected_numbers
-
-def compare_numbers_with_excel(detected_numbers, excel_path='data/test.xlsx'):
-    df = pd.read_excel(excel_path)
-    matched_names = []
-    for number in detected_numbers:
-        try:
-            match = df[df['ID'] == int(number)]
-            if not match.empty:
-                name = match['Name'].values[0]
-                matched_names.append({"ID": number, "name": name})
-            else:
-                matched_names.append({"ID": number, "name": "The student is not in the database"})
-        except ValueError:
-            pass
-    return matched_names
-
-
-
-def send_attendance_to_external_api(student_id):
-    url = 'http://54.242.19.19:3000/api/attendance'
-    random_id = random.randint(10000, 99999)  # Generate random ID
-
-    payload = {
-        "id": str(random_id),  # 👈 Send random ID
-        "studentId": str(student_id),
-        "status": "present",
-        "date": datetime.now().strftime("%m/%d/%Y %I:%M:%S %p +00:00")
-    }
-
-    print(f"Sending payload: {payload}")
-    try:
-        response = requests.post(url, json=payload)
-        try:
-            return response.status_code, response.json()
-        except ValueError:
-            return response.status_code, {"error": "Response is not valid JSON", "raw_response": response.text}
-    except Exception as e:
-        print(f"Failed to send attendance for {student_id}: {e}")
-        return 500, {"error": str(e)}
-
-
-@app.route('/detect_numbers', methods=['POST'])
-def detect_numbers():
-    if 'file' not in request.files:
-        return jsonify({"error": "No file part"}), 400
-
-    file = request.files['file']
-    if file.filename == '':
-        return jsonify({"error": "No selected file"}), 400
-
-    if file:
-        filename = file.filename
-        file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-        try:
-            file.save(file_path)
-            file_path = convert_image_format(file_path, 'PNG')
-            result = extract_text_with_paddleocr(file_path)
-            detected_numbers = extract_numbers_from_paddleocr(result)
-            if detected_numbers:
-                matched_names = compare_numbers_with_excel(detected_numbers)
-                for entry in matched_names:
-                    print(f"ID: {entry['ID']}, Name: {entry['name']}")
-                    if entry['name'] != "The student is not in the database":
-                        status_code, response = send_attendance_to_external_api(entry['ID'])
-                        print(f"External API response: {status_code} - {response}")
-                return jsonify(matched_names)
-            else:
-                print("No numbers detected.")
-                return jsonify({"message": "No numbers detected."})
-        except Exception as e:
-            print(f"Error processing the image: {e}")
-            return jsonify({"error": str(e)}), 500
-
-if __name__ == "__main__":
-    app.run(host='0.0.0.0', port=7860)
+import os
+import cv2
+from paddleocr import PaddleOCR
+import re
+from flask import Flask, request, jsonify
+from PIL import Image
+import pandas as pd
+import requests
+from datetime import datetime
+import random
+
+app = Flask(__name__)
+app.config['UPLOAD_FOLDER'] = 'uploads/'
+os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
+
+def convert_image_format(input_path, output_format='PNG'):
+    try:
+        image = Image.open(input_path)
+        if hasattr(image, '_getexif'):
+            exif = image._getexif()
+            if exif and 0x112 in exif:
+                orientation = exif[0x112]
+                rotations = {3: 180, 6: 270, 8: 90}
+                if orientation in rotations:
+                    image = image.rotate(rotations[orientation], expand=True)
+        output_path = os.path.splitext(input_path)[0] + '.' + output_format.lower()
+        image.convert("RGB").save(output_path, format=output_format)
+        return output_path
+    except Exception as e:
+        return input_path
+
+def extract_text_with_paddleocr(file_path):
+    ocr = PaddleOCR(use_angle_cls=True, lang='en', show_log=False)
+    result = ocr.ocr(file_path, cls=True)
+    return result
+
+def extract_numbers_from_paddleocr(result):
+    detected_numbers = []
+    for line in result:
+        for res in line:
+            text = res[1][0]
+            numbers = re.findall(r'\d+', text)
+            if numbers:
+                number_str = ''.join(numbers)
+                detected_numbers.append(number_str)
+    return detected_numbers
+
+def compare_numbers_with_excel(detected_numbers, excel_path='data/test.xlsx'):
+    df = pd.read_excel(excel_path)
+    matched_names = []
+    for number in detected_numbers:
+        try:
+            match = df[df['ID'] == int(number)]
+            if not match.empty:
+                name = match['Name'].values[0]
+                matched_names.append({"ID": number, "name": name})
+            else:
+                matched_names.append({"ID": number, "name": "The student is not in the database"})
+        except ValueError:
+            pass
+    return matched_names
+
+def send_attendance_to_external_api(student_id):
+    url = 'http://54.242.19.19:3000/api/attendance'
+    random_id = random.randint(10000, 99999)
+    payload = {
+        "id": str(random_id),
+        "studentId": str(student_id),
+        "status": "present",
+        "date": datetime.now().strftime("%m/%d/%Y %I:%M:%S %p +00:00")
+    }
+    print(f"Sending payload: {payload}")
+    try:
+        response = requests.post(url, json=payload)
+        try:
+            return response.status_code, response.json()
+        except ValueError:
+            return response.status_code, {"error": "Response is not valid JSON", "raw_response": response.text}
+    except Exception as e:
+        print(f"Failed to send attendance for {student_id}: {e}")
+        return 500, {"error": str(e)}
+
+@app.route('/detect_numbers', methods=['POST'])
+def detect_numbers():
+    if 'file' not in request.files:
+        return jsonify({"error": "No file part"}), 400
+    file = request.files['file']
+    if file.filename == '':
+        return jsonify({"error": "No selected file"}), 400
+    if file:
+        filename = file.filename
+        file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+        try:
+            file.save(file_path)
+            file_path = convert_image_format(file_path, 'PNG')
+            result = extract_text_with_paddleocr(file_path)
+            detected_numbers = extract_numbers_from_paddleocr(result)
+            if detected_numbers:
+                matched_names = compare_numbers_with_excel(detected_numbers)
+                for entry in matched_names:
+                    print(f"ID: {entry['ID']}, Name: {entry['name']}")
+                    if entry['name'] != "The student is not in the database":
+                        status_code, response = send_attendance_to_external_api(entry['ID'])
+                        print(f"External API response: {status_code} - {response}")
+                return jsonify(matched_names)
+            else:
+                print("No numbers detected.")
+                return jsonify({"message": "No numbers detected."})
+        except Exception as e:
+            print(f"Error processing the image: {e}")
+            return jsonify({"error": str(e)}), 500
+
+if __name__ == "__main__":
+    app.run(host='0.0.0.0', port=7860)