A bunch of mess

server.py

@@ -1,20 +1,42 @@
 from flask import Flask, request, jsonify
-import os
+from flask_sqlalchemy import SQLAlchemy
+from flask_migrate import Migrate
+from datetime import datetime
 from pyplatex import ANPR
-import torch
 from ultralytics.nn.tasks import DetectionModel
+import os
+import torch
 import asyncio
 
 torch.serialization.add_safe_globals({"DetectionModel": DetectionModel})
 
 # Web Server
 app = Flask(__name__)
+app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///data.sqlite"
+app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
+app.config["SECRET_KEY"] = "bvjchsygvduycgsyugc"
+
+# ORM
+db = SQLAlchemy()
+db.init_app(app)
+
+migrate = Migrate(app, db)
 
 # Saving images locally
 UPLOAD_FOLDER = "uploads"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
 
+# db classes
+class Plate(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    plate = db.Column(db.String(40), nullable=False)
+
+
+class LoggedItem(Plate):
+    dateLogged = db.Column(db.DateTime, default=datetime.now)
+
+
 # Default app route
 @app.route("/")
 def home():
@@ -37,12 +59,13 @@ def data():
     if file.filename.lower().endswith(".jpg"):
         filepath = os.path.join(UPLOAD_FOLDER, file.filename)
         file.save(filepath)
-        print(asyncio.run(process_image(filepath)))
+        plate = asyncio.run(process_image(filepath))
         return jsonify(
             {
                 "message": "File uploaded successfully",
                 "filename": file.filename,
                 "status": True,
+                "anpr": plate,
             }
         )
 
@@ -51,8 +74,13 @@ def data():
 
 async def process_image(file: str):
     anpr = ANPR()
-    plates = await anpr.detect(file)
-    return plates
+    anpr_info = await anpr.detect(file)
+    number_plate = anpr_info["plate_number"]
+    if number_plate:
+        db.session.add(LoggedItem(plate=number_plate))
+        db.session.commit()
+        return number_plate
+    return "ERROR"
 
 
 if __name__ == "__main__":