Update main.py

main.py

@@ -6,6 +6,60 @@ import io
 import random
 import string
 from mimetypes import guess_type
+import requests
+import time
+
+def read_image_from_stream(image_data, subscription_key, endpoint):
+
+    headers = {
+        'Ocp-Apim-Subscription-Key': subscription_key,
+        'Content-Type': 'application/octet-stream'
+    }
+
+    response = requests.post(f"{endpoint}/vision/v3.2/read/analyze", headers=headers, data=image_data)
+
+    if response.status_code == 202:
+        read_response_headers = response.headers
+        operation_location = read_response_headers["Operation-Location"]
+    else:
+        raise Exception(f"Unexpected response status: {response.status_code}")
+
+    return operation_location
+
+def get_read_result(operation_location, subscription_key):
+    headers = {
+        'Ocp-Apim-Subscription-Key': subscription_key
+    }
+    
+    while True:
+        response = requests.get(operation_location, headers=headers)
+        if response.status_code == 200:  # HTTP 200 indicates success
+            read_result = response.json()
+            status = read_result['status']
+            if status == 'succeeded':
+                break
+            elif status in ['failed', 'notStarted', 'running']:
+                time.sleep(1)  # Wait before polling again
+            else:
+                raise Exception(f"Unexpected status: {status}")
+        else:
+            raise Exception(f"Unexpected response status: {response.status_code}")
+    
+    return read_result
+
+def process_image(image_path, subscription_key, endpoint):
+    operation_location = read_image_from_stream(image_path, subscription_key, endpoint)
+    operation_id = operation_location.split("/")[-1]
+    operation_location = f"{endpoint}/vision/v3.2/read/analyzeResults/{operation_id}"
+    read_result = get_read_result(operation_location, subscription_key)
+    if read_result['status'] == 'succeeded':
+        output = []
+        for text_result in read_result['analyzeResult']['readResults']:
+            for line in text_result['lines']:
+                output.append(line['text'])
+        return " ".join(output).replace("\n", " ")  # Join lines and replace newlines with spaces
+    else:
+        return "Processing failed or did not succeed."
 
 def get_answer(image: io.BytesIO, question: str):
     filename = image.name
@@ -55,30 +109,7 @@ def valuate_qna(p: str, r: str) -> str:
     return hug.output.get("data", [None])[0]
 
 def ocr(image: io.BytesIO) -> str:
-    spid3 = os.getenv('SPID3')
-    hug = hugging(spid3)
-    filelink = hug.upload(image)
-    image.seek(0, 2)
-    file_size = image.tell()
-    image.seek(0)
-    image_mime_type, _ = guess_type(image.name)
-    data = [
-		{
-			"meta": {
-				"_type": "gradio.FileData"
-			},
-			"mime_type": image_mime_type,
-			"orig_name": image.name,
-			"path": filelink.split("=", 1)[1],
-			"size": file_size,
-			"url": filelink
-		},
-		os.getenv('AZKEY'),
-		os.getenv('AZURL')
-	]
-    eventid = hug.filnal_setup(data, 0, 13)
-    hug.start()
-    return data, hug.output #.get("data", [None])[0]
+    return process_image(image, os.getenv('AZKEY'), os.getenv('AZURL'))
 
 app = Flask(__name__)
 
@@ -132,8 +163,8 @@ def task_ocr():
     headers = dict(request.headers)
     if not headers.get("KEY") != os.getenv("KEY"):
         return jsonify({"status" : False, "msg" : "Invalid API Key"}), 404
-    eventid, answer = ocr(file_content)
-    return jsonify({"status" : True, "data" : answer, "other" : eventid})
+    answer = ocr(file_content)
+    return jsonify({"status" : True, "data" : answer})
 
 @app.route('/check', methods=['POST', 'GET'])
 def check_f():