app.py

import cv2
from flask import Flask, render_template, request, redirect, url_for, jsonify
import base64
from PIL import Image
import io
import os
from datetime import date, datetime
from roboflow import Roboflow
from flask_sqlalchemy import SQLAlchemy
from datetime import datetime

dateformat = '%Y-%m-%d'




db = SQLAlchemy()
DB_NAME = "database.db"

from app import db
from flask_login import UserMixin
from sqlalchemy.sql import func

class Food_item(db.Model):
    __tablename__ = 'food_item'
    __table_args__ = {'extend_existing': True}  # Add this line
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(150))
    expiry_date = db.Column(db.DateTime(), default=func.now())
    #user_id = db.Column(db.Integer, foreign_key=True)

class User(db.Model, UserMixin):
    __tablename__ = 'user'
    __table_args__ = {'extend_existing': True}  # Add this line
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(150), unique=True)
    password = db.Column(db.String(150))
    user_name = db.Column(db.String(150))
    #food_items = db.relationship('Food_item')

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///{DB_NAME}'
db.init_app(app)

def create_database(app):
    if not os.path.exists("HackAttack/" + DB_NAME):
        with app.app_context():
            db.create_all()
        print('Create Database!')


create_database(app)

# god_user = User(email='god@godmail.com', password='ImbetterUjustsuck', user_name="notgod")

rf = Roboflow(api_key="FBQSwgHbiaU0halM4Nxb")
project = rf.workspace().project("hackattackk")
model = project.version("1").model

@app.route("/")
def index():
    # Render the login page
    return render_template("index.html")

@app.route("/signup", methods=['GET', 'POST'])
def signup():
    if request.method == 'POST':
        # Assuming you have form fields named 'username' and 'password' in your signup.html
        username = request.form['username']
        password = request.form['password']
        # Process the signup form
        # Insert the user's data into your database and perform any necessary validation
        # For demonstration purposes, redirect to the home page
        return redirect(url_for('home'))
    # If the request method is GET, render the signup page
    return render_template("signup.html")

@app.route("/home")
def home():
    # Render the home page
    return render_template("home.html")  # Ensure the path is correct, might be just "home.html" without "templates/"

@app.route("/scanitem")
def scanitem():
    return render_template("cam.html")

@app.route("/login", methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        # Process the login form
        username = request.form['username']
        password = request.form['password']
        # Validate the user's credentials with your database
        # If valid, log the user in and redirect to the home page
        return redirect(url_for('home'))
    # If the request method is GET, render the login page
    return render_template("index.html")

@app.route("/viewInventory")
def view_inventory():
    # Render the viewInventory page
    food_items= Food_item.query.all()
    #print(food_items)
    for item in food_items:
        delta = item.expiry_date.date() - date.today() 
        item.days_until_expiry = delta.days
    return render_template("viewInventory.html",food_items=food_items)


@app.route('/capture', methods=['POST'])
def capture():
    data = request.get_json()
    image_data = data['image'].split(",")[1]  # Split the data URL to get just the base64 part
    image_bytes = base64.b64decode(image_data)
    #date
    expiry_date = data.get('expiryDate') #date
    image = Image.open(io.BytesIO(image_bytes))
    
    # Save the image to a file or process it
    filepath = os.path.join('captured_images', 'captured_image.png')  # Ensure this directory exists
    image.save(filepath)

    img=cv2.imread(filepath)
    results=model.predict(img,confidence=40,overlap=30).json()
    predictions = results['predictions']
    print(predictions)
    if predictions:
        # For example, handling the most confident prediction
        most_confident_prediction = max(predictions, key=lambda x: x['confidence'])
        print(f"Detected: {most_confident_prediction['class']} with confidence: {most_confident_prediction['confidence']}")
        food_item = Food_item(expiry_date=datetime.strptime(expiry_date,dateformat), name=most_confident_prediction['class'])
        db.session.add(food_item)
        db.session.commit()
        return jsonify({'message': 'Detection successful!', 'class': most_confident_prediction['class'], 'confidence': most_confident_prediction['confidence']})
    else:
        print("No detection made.")
        return jsonify({'message': 'No detection made.'})

    # Print expiration date if needed
    # print("Expiration Date:", expiry_date)
    
    # return jsonify({'message': 'Image and expiration date received successfully!'})

if __name__ == "__main__":
    app.run(debug=True)