server.py

import os
import random
import string
import time
from flask import Flask, flash, request, redirect, url_for, send_from_directory, jsonify
from flask_cors import CORS, cross_origin
from werkzeug.utils import secure_filename

COST = 300
UPLOAD_FOLDER = os.path.join(os.getcwd(), "static")
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}

app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
app.config['CORS_HEADERS'] = 'Content-Type'
app.config['SECRET_KEY'] = 'myverysecretkey'

cors = CORS(app)

def random_string(stringLength=10):
    letters = string.ascii_lowercase
    return ''.join(random.choice(letters) for i in range(stringLength))

def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

def get_extension(filename):
    return filename.rsplit('.', 1)[1].lower()

def get_quality(good, not_good):
    return (good * 100 / (good + not_good))

def get_price(quality):
    return (COST * quality) / 100

@app.route('/', methods=['GET', 'POST'])
@cross_origin(origin='localhost',headers=['Content-Type','Authorization'])
def upload_file():
    if request.method == 'POST':
        
        # check if the post request has the file part
        if 'file' not in request.files:
            flash('No file part')
            return jsonify({
                "message": "No file part"
            })
        
        file = request.files['file']

        # if user does not select file, browser also
        # submit an empty part without filename
        if file.filename == '':
            flash('No selected file')
            return jsonify({
                "message": "No selected file"
            })

        if file and allowed_file(file.filename):
            
            # save uploaded file to static with a random name
            filename = secure_filename(file.filename)
            ex = get_extension(filename)
            new_filename = random_string(4) + "." + ex

            # TODO: change image resolution
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], new_filename))
            
            from wheat_quality_predictor import predict
            path_file = os.path.join(app.config['UPLOAD_FOLDER'], new_filename)
            print("Predicting", path_file)
            good, not_good = predict(path_file)
            print("Prediction Done")
    
            # return payload
            response = jsonify({
                "url": url_for('uploaded_file', filename=new_filename),
                "quality": get_quality(good, not_good),
                "price": get_price(get_quality(good, not_good))
            })
            return response


    elif request.method == 'GET':
        return "AgroAI API"

# Test route
@app.route('/image', methods=['GET', 'POST'])
@cross_origin(origin='localhost',headers=['Content-Type','Authorization'])
def test_image():
    print("Getting image")

    # check if the post request has the file part
    if 'file' not in request.files:
        flash('No file part')
        return jsonify({
            "message": "No file part"
        })
    
    file = request.files['file']
    print(file)

    # if user does not select file, browser also
    # submit an empty part without filename
    if file.filename == '':
        flash('No selected file')
        return jsonify({
            "message": "No selected file"
        })

    if file and allowed_file(file.filename):
        
        # save uploaded file to static with a random name
        filename = secure_filename(file.filename)
        ex = get_extension(filename)
        new_filename = random_string(4) + "." + ex

        file.save(os.path.join(app.config['UPLOAD_FOLDER'], new_filename))
        print("Testing image successfuly")
        time.sleep(10)
        response = jsonify({
            "url": url_for('uploaded_file', filename=new_filename),
            "quality": 50,
            "price": 100,
        })
        return response

# Route to fetch uploaded images
@app.route('/uploads/<filename>')
def uploaded_file(filename):
    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5000, debug=True)