Upload new_app.py

new_app.py

@@ -0,0 +1,62 @@
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.responses import JSONResponse
+import uuid
+import io
+from PIL import Image
+from utils.ocr_utils import extract_and_translate_chunk
+
+app = FastAPI(title="Manga Translation API", description="FastAPI implementation for Manga Translation with OCR and Azure")
+
+uploaded_images = {}
+
+@app.post("/upload_panel/")
+
+async def upload_panel(file: UploadFile = File(...)):
+    '''Takes an JEPG/PNG manga panel image and returns a image_id  for reference'''
+    
+    if file.content_type not in ["image/jpeg", "image/png"]:
+        raise HTTPException(status_code=400, detail="Only JPEG and PNG files are supported")
+    
+    image_id = str(uuid.uuid4())
+    
+    content = await file.read()
+    uploaded_images[image_id] = content
+
+    return JSONResponse(status_code=200, content={"image_id": image_id})
+
+@app.get("/translate_panel")
+
+async def translate_panel(image_id: str):
+    '''Takes an image_id and returns a translated text from OCR in JSON format'''
+
+    # Loads the img
+    if image_id not in uploaded_images:
+        raise HTTPException(status_code=404, detail="Image not found")
+    
+    image_content = uploaded_images[image_id]
+
+    image = Image.open(io.BytesIO(image_content)).convert("RGB")
+
+    # Extract with OCR and translating
+    translated_text = extract_and_translate_chunk(image)
+
+    results = []
+
+    for translation in translated_text:
+        results.append({
+            "polygon": translation['polygon'],
+            "original_text": translation['original'],
+            "translated_text": translation['translated'],
+            "language": "ja",
+            "confidence": 0.95  # Placeholder confidence value
+        })
+
+    return JSONResponse(status_code=200, content={"image_id": image_id, "results": results})
+
+@app.get("/")
+async def root():
+    return {"message": "Welcome to the Manga Translation API"}
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)