Update app.py

app.py

@@ -29,6 +29,10 @@ from transformers import (
     AutoTokenizer,
 )
 
+import logging
+logging.basicConfig(level=logging.INFO
+                    , format='%(asctime)s - %(levelname)s - %(message)s')
+
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
 # load model and processor
@@ -147,12 +151,11 @@ async def translate_audio(
             "srcLang": srcLang, 
             "tgtLang": tgtLang
         } 
-        print(result_dict)
         
         return result_dict
        
     except Exception as error:
-        print("Error: ", str(error))
+        logging.error(f"Error translating audio {file.filename}: {error}")
         raise HTTPException(status_code=500, detail=str(error))
         
     finally:
@@ -169,8 +172,10 @@ async def translate_text(text: str,
                srcLang: str = Form(...),
                tgtLang: str = Form(...)):
     result = translate(text, srcLang, tgtLang)
-    print('Raw: ', text)
-    print('Translated: ', result)
+
+    if not result:
+        logging.error("Translation failed for text: %s", text)
+        raise HTTPException(status_code=500, detail="Translation failed")    
     
     result_dict = {
         "text": text, 
@@ -199,8 +204,9 @@ async def synthesize(text: str):
         temp_file, 
         rate=model.config.sampling_rate,
         data=data_np_squeezed
-        )
-        
+        )    
+    logging.info(f"Synthesizing completed for text: {text}")
+
     return FileResponse(
         temp_file, 
         media_type="audio/wav",