loggers

Dockerfile

@@ -17,5 +17,6 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 RUN --mount=type=secret,id=API_KEY,mode=0444,required=true 
 RUN --mount=type=secret,id=API_URL,mode=0444,required=true 
 
+EXPOSE 7860
 COPY --chown=user . /app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -6,6 +6,9 @@ from dotenv import load_dotenv
 from tasks import process_video
 from fastapi.responses import JSONResponse
 from config import API_KEY
+import logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 app = FastAPI()
 vitpose = VitPose()
 # vitpose.pipeline.warmup()
@@ -37,7 +40,7 @@ async def upload(background_tasks: BackgroundTasks,
     with open(file.filename, "wb") as f:
         f.write(contents)
     
-
+    logger.info(f"Uploading video {file.filename}")
     
     # Create a clone of the file with content already read
     background_tasks.add_task(process_video, file.filename, vitpose, user_id)  

tasks.py

@@ -11,9 +11,11 @@ logger = logging.getLogger(__name__)
 def process_video(video_path: str,vitpose: VitPose,user_id: str):
     
     logger.info(f"Processing video {video_path}")
-        
+    
+    
     new_file_name = video_path.split(".")[0] + "edited." + video_path.split(".")[1]
     new_file_name = os.path.join("static", new_file_name)
+    logger.info(f"New file name {new_file_name}")
     
     vitpose.output_video_path = new_file_name
     annotated_frames = vitpose.run(video_path)