Update app.py

app.py

@@ -3,6 +3,7 @@ import cv2
 import numpy as np
 import uvicorn
 from fastapi import FastAPI
+from fastapi.responses import PlainTextResponse
 from pydantic import BaseModel
 from paddleocr import PaddleOCR, PPStructureV3
 
@@ -112,35 +113,42 @@ def run_ocr(ocr_input: OcrInput):
     except Exception as e:
         return {"error": f"An error occurred in /ocr: {str(e)}"}
         
-@app.post("/structure")
+@app.post("/structure", response_class=PlainTextResponse)
 def run_structure_analysis(ocr_input: OcrInput):
     """Endpoint for layout analysis, table recognition, and text extraction (PPStructure)."""
     try:
-        print ("** /structure called. Decoding image **")
-        
+        print("** /structure called. Decoding image **")
+
         img = decode_image(ocr_input.image_base64)
         if img is None:
-            return {"error": "Invalid image data. Could not decode."}
+            # For errors, you can still return a PlainTextResponse
+            return PlainTextResponse("Error: Invalid image data. Could not decode.", status_code=400)
+
+        print("** Request for structure analysis received. Running structure_engine.**")
         
-        print ("** Request for structure analysis received. Running structure_engine.**")
         # Use the structure_engine here
         raw_result = structure_engine.predict(img)
-        print (raw_result)
-
-        print ("** We have got the result back. Now converting to JSON **")
+        print(raw_result)
+        print("** We have got the result back. Now converting to text **")
         
         # --- CRITICAL STEP: Clean the result before returning ---
         json_safe_result = clean_ppstructure_result(raw_result)
-        print (json_safe_result)
+        # Convert the processed result (which is a Python object/dict) into a JSON string
+        # to ensure it's properly formatted text.
+        import json
+        text_result = json.dumps({"result": json_safe_result}, indent=2)
         # --------------------------------------------------------
-
-        print ("** Structure Analysis Complete. Returning the JSON result.**")
-        # print(json_safe_result) # Helpful for debugging, but can be verbose in production logs
-
-        return {"result": json_safe_result}
         
+        print("** Structure Analysis Complete. Returning the text result.**")
+        
+        # Return the string directly. PlainTextResponse will handle setting 
+        # the Content-Type header to text/plain.
+        return text_result
+    
     except Exception as e:
-        return {"error": f"An error occurred in /structure: {str(e)}"}
+        # Return error as plain text with an appropriate status code
+        error_message = f"An error occurred in /structure: {str(e)}"
+        return PlainTextResponse(f"Error: {error_message}", status_code=500)
 # --- Run the App ---
 
 if __name__ == "__main__":