fix

ocr/api/message/openai_request.py

@@ -3,7 +3,7 @@ from ocr.core.wrappers import openai_wrapper
 
 
 @openai_wrapper(model='gpt-4o-mini')
-async def generate_report(content: list):
+async def generate_report(content: str):
     messages = [
         {
             "role": "system",
@@ -18,7 +18,7 @@ async def generate_report(content: list):
 
 
 @openai_wrapper(model='gpt-4o-mini')
-async def extract_original_text(content: list):
+async def extract_original_text(content: str):
     messages = [
         {
             "role": "system",

ocr/api/message/views.py

@@ -22,10 +22,11 @@ async def get_all_chat_messages(
         else:
             raise HTTPException(status_code=400, detail='Unsupported file type.')
         text_content = extract_text_from_images(images)
-        original_text, response = await asyncio.gather(
-            extract_original_text(text_content),
-            generate_report(text_content)
-        )
-        return OcrResponseWrapper(data=OcrResponse(text=clean_response(response), originalText=original_text))
+        # original_text, response = await asyncio.gather(
+        #     extract_original_text(text_content),
+            # generate_report(text_content)
+        # )
+        cleaned_original_text = await extract_original_text(text_content)
+        return OcrResponseWrapper(data=OcrResponse(text=clean_response(text_content), originalText=cleaned_original_text))
     finally:
         await file.close()