Spaces:
Sleeping
Sleeping
from flask import Flask, request, jsonify | |
import requests | |
from imageEmbedding.index import get_image_embedding | |
from imageToText.index import extract_text | |
app = Flask(__name__) | |
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 | |
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 | |