Update app/main.py

app/main.py

@@ -1,9 +1,10 @@
 # main.py
 from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import JSONResponse, StreamingResponse
+from fastapi.responses import JSONResponse
 from typing import Dict, Any
 import io
-from annotations import analyze_pdf
+import base64  # Import base64 for encoding
+from .annotations import analyze_pdf  # Use relative import
 
 app = FastAPI(
     title="PDF Language Issue Analyzer",
@@ -31,7 +32,9 @@ async def analyze_pdf_endpoint(file: UploadFile = File(...)):
         }
 
         if annotated_pdf:
-            response["annotated_pdf"] = "data:application/pdf;base64," + io.BytesIO(annotated_pdf).getvalue().decode('latin1')  # Adjust encoding as needed
+            # Properly encode the annotated PDF in Base64
+            encoded_pdf = base64.b64encode(annotated_pdf).decode('utf-8')
+            response["annotated_pdf"] = f"data:application/pdf;base64,{encoded_pdf}"
 
         return JSONResponse(content=response)
     except Exception as e: