Update app.py

app.py

@@ -1,10 +1,28 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 from transformers import pipeline
+import logging
 
 app = FastAPI()
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-id")
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+try:
+    logger.info("Loading translation model...")
+    translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-id")
+    logger.info("Model loaded successfully")
+except Exception as e:
+    logger.error(f"Failed to load model: {str(e)}")
+    raise Exception(f"Model initialization failed: {str(e)}")
 
 @app.post("/translate")
 async def translate(text: str):
-    result = translator(text)
-    return {"translated_text": result[0]["translation_text"]}
+    if not text:
+        raise HTTPException(status_code=400, detail="Text input is required")
+    try:
+        result = translator(text)
+        return {"translated_text": result[0]["translation_text"]}
+    except Exception as e:
+        logger.error(f"Translation failed: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Translation failed: {str(e)}")