Spaces:
Running
Running
File size: 1,580 Bytes
e09fe9c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
from flask import Flask, request, jsonify
import requests
from imageEmbedding.index import get_image_embedding
from imageToText.index import extract_text
app = Flask(__name__)
@app.route('/extract_image_text_and_embedding', methods=['POST'])
def get_image_embedding_route():
try:
image_url = request.json.get("imageUrl")
print(image_url)
response = requests.get(image_url)
if response.status_code != 200:
return jsonify({"error": "Failed to download image"}), 500
binary_data = response.content
extracted_text = extract_text(binary_data)
image_embedding = get_image_embedding(binary_data)
image_embedding_list = image_embedding.tolist()
return jsonify({"image_embedding": image_embedding_list,"extracted_text":extracted_text}), 200
except Exception as e:
return jsonify({"error": str(e)}), 500
@app.route('/extract_image_text_and_embedding_binary_data', methods=['POST'])
def get_image_embedding_binary_route():
try:
# Assuming the image data is sent in the request body
binary_data = request.get_data()
if not binary_data:
return jsonify({"error": "No image data provided"}), 400
extracted_text = extract_text(binary_data)
image_embedding = get_image_embedding(binary_data)
image_embedding_list = image_embedding.tolist()
return jsonify({"image_embedding": image_embedding_list, "extracted_text": extracted_text}), 200
except Exception as e:
return jsonify({"error": str(e)}), 500
|