ImageProcessing/backend

1from flask import Flask, request, jsonify2 3import requests4from imageEmbedding.index import get_image_embedding5from imageToText.index import extract_text6 7app = Flask(__name__)8 9 10@app.route('/extract_image_text_and_embedding', methods=['POST'])11def get_image_embedding_route():12    try:13        image_url = request.json.get("imageUrl")14        print(image_url)15        response = requests.get(image_url)16        if response.status_code != 200:17            return jsonify({"error": "Failed to download image"}), 50018        binary_data = response.content19        extracted_text = extract_text(binary_data)20        image_embedding = get_image_embedding(binary_data)21        image_embedding_list = image_embedding.tolist()22        return jsonify({"image_embedding": image_embedding_list,"extracted_text":extracted_text}), 20023 24    except Exception as e:25        return jsonify({"error": str(e)}), 50026 27 28@app.route('/extract_image_text_and_embedding_binary_data', methods=['POST'])29def get_image_embedding_binary_route():30    try:31        # Assuming the image data is sent in the request body32        binary_data = request.get_data()33        if not binary_data:34            return jsonify({"error": "No image data provided"}), 40035 36        extracted_text = extract_text(binary_data)37        image_embedding = get_image_embedding(binary_data)38        image_embedding_list = image_embedding.tolist()39        return jsonify({"image_embedding": image_embedding_list, "extracted_text": extracted_text}), 20040 41    except Exception as e:42        return jsonify({"error": str(e)}), 50043    44