Update app/main.py

app/main.py

@@ -8,7 +8,7 @@ from pydantic import BaseModel, HttpUrl, root_validator, ValidationError
 from fastapi.encoders import jsonable_encoder
 
 from app.lens_images_core import translate_lens
-from app.lens_text_core   import translate_lens_text
+# from app.lens_text_core   import translate_lens_text
 
 PORT              = int(os.getenv("PORT", 8080))
 MAX_WORKERS       = int(os.getenv("MAX_WORKERS", 8))
@@ -115,7 +115,9 @@ async def health():
 @app.post("/translate")
 async def translate(job: Job):
     await ensure_workers_started()
-    if job.mode not in ("lens_images", "lens_text"):
+    # if job.mode not in ("lens_images", "lens_text"):
+    #     raise HTTPException(400, "unsupported mode")
+    if job.mode != "lens_images":
         raise HTTPException(400, "unsupported mode")
     jid = uuid.uuid4().hex
     job.metadata.pipeline.append(PipelineEvent(stage="received_rest", at=datetime.utcnow()))
@@ -155,8 +157,8 @@ async def ws_endpoint(ws: WebSocket):
                     
                     if msg.payload.mode == "lens_images":
                         await jobq_img.put((jid, msg.payload))
-                    elif msg.payload.mode == "lens_text":
-                        await jobq_text.put((jid, msg.payload))
+                    # elif msg.payload.mode == "lens_text":
+                    #     await jobq_text.put((jid, msg.payload))
                     else:
                         await ws.send_json({"type": "error","detail": "unsupported_mode"})
                         pending_ws.pop(jid, None)
@@ -182,8 +184,8 @@ async def worker(mode: str, q: asyncio.Queue):
             log.info("worker start %s mode=%s src=%s", jid, job.mode, job.src)
             if mode == "lens_images":
                 res = await translate_lens(str(job.src), job.lang)
-            elif mode == "lens_text":
-                res = await translate_lens_text(str(job.src))
+            # elif mode == "lens_text":
+            #     res = await translate_lens_text(str(job.src))
             else:
                 raise RuntimeError(f"unsupported mode {mode}")
 
@@ -234,8 +236,8 @@ async def startup():
     if ENABLE_BACKGROUND_WORKERS:
         for _ in range(MAX_WORKERS_IMAGES):
             asyncio.create_task(worker("lens_images", jobq_img))
-        for _ in range(MAX_WORKERS_TEXT):
-            asyncio.create_task(worker("lens_text", jobq_text))
+        # for _ in range(MAX_WORKERS_TEXT):
+        #     asyncio.create_task(worker("lens_text", jobq_text))
     asyncio.create_task(cleanup())
     log.info(
         "startup OK – %d image workers, %d text workers, TTL=%ds (workers_enabled=%s)",