Update app.py

app.py

@@ -1,45 +1,43 @@
-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
-    
-if __name__ == '__main__':  
-    app.run(port=4000)
+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
+