edits

app.py

@@ -1,11 +1,10 @@
 from fastapi import FastAPI, UploadFile, File, Response,Header, BackgroundTasks,Body
 from fastapi.staticfiles import StaticFiles
 from vitpose import VitPose
-import os 
 from dotenv import load_dotenv
 from tasks import process_video
 from fastapi.responses import JSONResponse
-from config import API_KEY
+from config import AI_API_TOKEN
 import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -13,7 +12,7 @@ app = FastAPI()
 vitpose = VitPose()
 # vitpose.pipeline.warmup()
 
-load_dotenv()
+load_dotenv() 
 
 
 app.mount("/static", StaticFiles())
@@ -30,10 +29,10 @@ def test():
 async def upload(background_tasks: BackgroundTasks,
                  file: UploadFile = File(...), 
                  token: str = Header(...),
-                 user_id: str = Body(...)):
+                 user_id: str = Body(...),
+                 player_id: str = Body(...)):
     
-    if token != API_KEY:
-        logger.info(f"Unauthorized {token} {API_KEY}")
+    if token != AI_API_TOKEN:
         return JSONResponse(content={"message": "Unauthorized", "status": 401})
     
     logger.info("reading contents")
@@ -47,7 +46,7 @@ async def upload(background_tasks: BackgroundTasks,
     logger.info(f"file saved {file.filename}")
     
     # Create a clone of the file with content already read
-    background_tasks.add_task(process_video, file.filename, vitpose, user_id)  
+    background_tasks.add_task(process_video, file.filename, vitpose, user_id,player_id)  
 
     # Return the file as a response
     return JSONResponse(content={"message": "Video uploaded successfully", "status": 200})

config.py

@@ -4,4 +4,5 @@ from dotenv import load_dotenv
 load_dotenv()
 
 API_URL = os.getenv("API_URL")
-API_KEY = os.getenv("API_KEY")
+API_KEY = os.getenv("API_KEY")
+AI_API_TOKEN = os.getenv("AI_API_TOKEN")

tasks.py

@@ -1,38 +1,46 @@
 from vitpose import VitPose
 import requests
 import os
+from config import API_URL,API_KEY
 from fastapi import UploadFile
-from config import API_URL
-import time
 import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-def process_video(video_path: str,vitpose: VitPose,user_id: str):
+def process_video(file_name: str,vitpose: VitPose,user_id: str,player_id: str):
     
-    logger.info(f"starting task {video_path}")
+    video_path = file_name
+
+    contents = open(video_path, "rb").read()
+
+    with open(video_path, "wb") as f:
+        f.write(contents)
     
+    logger.info(f"file saved {video_path}")
+   
+    logger.info(f"starting task {video_path}")
     
-    new_file_name = video_path.split(".")[0] + '.' + video_path.split(".")[1]
-    new_file_name = os.path.join("static", new_file_name)
+    new_file_name = os.path.join("static", video_path)
     logger.info(f"new file name {new_file_name}")
     
     vitpose.output_video_path = new_file_name
     annotated_frames = vitpose.run(video_path)
     
+    vitpose.frames_to_video(annotated_frames)
     
+    logger.info(f"Video processed {video_path}")
     
-    
-    annotated_video_path = vitpose.frames_to_video(annotated_frames)
-    
-    logger.info(f"Video processed {annotated_video_path}")
-    
-    with open(annotated_video_path, "rb") as f:
+    with open(new_file_name, "rb") as f:
         contents = f.read()
         
     url = API_URL+ "/excercises/webhooks/video-processed"
     logger.info(f"Sending video to {url}")
-    files = {"file": (annotated_video_path, contents, "video/mp4")}
-    response = requests.post(url, files=files, data={"user_id":user_id,"typeMessage":"video_processed","file_name":annotated_video_path}, stream=True)
+    files = {"file": (video_path, contents, "video/mp4")}
+
+    response = requests.post(url, files=files, 
+                             data={"user_id":user_id,"typeMessage":"video_processed","file_name":video_path,
+                                   "player_id":player_id}, 
+                             stream=True,
+                             headers={"token":API_KEY})
     logger.info(f"Video sent to {url}")
-    print(response.json())
+    

vitpose.py

@@ -96,7 +96,6 @@ class VitPose:
         else:
             print(f"Dimensions: {width}x{height}")
             out = cv2.VideoWriter(self.output_video_path, fourcc, self.video_metadata["fps"], (width, height))
-        
         for frame in frames:
             if rotate:
                 # Rotate landscape videos 90 degrees to make them vertical
@@ -104,7 +103,5 @@ class VitPose:
                 out.write(rotated_frame)
             else:
                 # Already vertical, no rotation needed
-                out.write(frame)
-            
-        out.release()
-        return self.output_video_path
+                out.write(frame)    
+        out.release()