app.py

from flask_cors import CORS
from flask import Flask, request, jsonify, send_file, request
from werkzeug.utils import secure_filename
import os
import classBased as cb
import huffmanFiles as hfile
import json
import uuid

hf = cb.Huffboth()
decode_dir = 'decoded'
encodedCompressed_dir = 'encodedCompressed'
encodedUncompressed_dir = 'encodedUncompressed'

app = Flask(__name__)
CORS(app)

###############################################################
# Test Routes


@app.route('/')
def index():
    return "<h1>Welcome to our server !!</h1>"


@app.route('/home')
def index2():
    return "<h1>Welcome to Home server !!</h1>"


@app.route('/api/home')
def index3():
    return jsonify({
        "home": "This is home sweet home",
    })

###############################################################
# Text Compression Decompression Routes


@app.route("/api/huffman-text-encode", methods=["POST"])
def huffmanTextEncode():
    # print(request.get_json())
    obj = request.get_json()
    input = obj["payload"]
    # print(input)
    inputSize = len(input) * 8
    encoded = hf.Encode(input)
    outputSize = len(encoded)
    tree = hf.getTree()
    compressionRatio = inputSize/outputSize
    return jsonify({
        "output": encoded,
        'inputSize': inputSize,
        "outputSize": outputSize,
        "tree": tree,
        "compressionRatio": round(compressionRatio, 2)
    })


@app.route("/api/huffman-text-decode", methods=["POST"])
def huffmanTextDecode():
    # print(request.get_json())
    obj = request.get_json()
    input = obj["payload"]
    inputSize = len(input)
    # print(input)
    decoded = hf.Decode(input)
    tree = hf.getTree()
    outputSize = len(decoded) * 8
    compressionRatio = inputSize/outputSize
    return jsonify({
        "output": decoded,
        'inputSize': inputSize,
        "outputSize": outputSize,
        "compressionRatio": round(compressionRatio, 2),
        "tree": tree,
    })


@app.route("/api/get-tree", methods=["GET"])
def sendHuffmanTree():
    tree = hf.getTree()
    return jsonify({
        "tree": tree,
    })
###############################################################
# File Upload routes for normal and binary files


@app.route('/api/upload-normal-file', methods=["POST"])
def uploadNormalFile():
    if request.method == 'POST':
        fileID = uuid.uuid4()
        f = request.files['file']
        f.save(os.path.join(decode_dir, f"{fileID}.txt"))

        hfile.encodeHuffFile(fileID)
        return 'normal file uploaded successfully'

    return "Normal File Upload Route"


@app.route('/api/upload-encoded-compressed-file', methods=["POST"])
def uploadEncodedCompressedFile():
    if request.method == 'POST':
        fileID = uuid.uuid4()
        f = request.files['file']
        f.save(os.path.join(encodedCompressed_dir, f"{fileID}.bin"))
        hfile.decodeHuffFile(fileID)
        return 'binary file uploaded successfully'
    return "Binary File Upload Route"

###############################################################
# Get the files by ID


@app.route('/decoded/<ID>')
def getDecodedFile(ID):
    filename = f'{decode_dir}/image-{ID}.png'
    return send_file(filename, mimetype='image/png')


@app.route('/encoded-compressed/<ID>')
def getEncodedCompressedFile(ID):
    filename = f'{encodedCompressed_dir}/image-{ID}.png'
    return send_file(filename, mimetype='image/png')


@app.route('/encoded-uncompressed/<ID>')
def getEncodedUnompressedFile(ID):
    filename = f'{encodedUncompressed_dir}/image-{ID}.png'
    return send_file(filename, mimetype='image/png')


#################################################################
if __name__ == '__main__':
    app.config['ENV'] = 'development'
    app.config['DEBUG'] = True
    app.run(threaded=True, port=5000, debug=True)