nicolasbuitragob commited on
Commit
825522e
·
1 Parent(s): 2d729d2

remove token for a sec

Browse files
Files changed (2) hide show
  1. app.py +10 -4
  2. tasks.py +2 -2
app.py CHANGED
@@ -32,15 +32,21 @@ async def upload(background_tasks: BackgroundTasks,
32
  token: str = Header(...),
33
  user_id: str = Body(...)):
34
 
35
- if token != API_KEY:
36
- return JSONResponse(content={"message": "Unauthorized", "status": 401})
37
 
 
 
 
 
 
38
  contents = await file.read()
39
- # Save the file to the local directory
 
 
40
  with open(file.filename, "wb") as f:
41
  f.write(contents)
42
 
43
- logger.info(f"Uploading video {file.filename}")
44
 
45
  # Create a clone of the file with content already read
46
  background_tasks.add_task(process_video, file.filename, vitpose, user_id)
 
32
  token: str = Header(...),
33
  user_id: str = Body(...)):
34
 
35
+ logger.info(f"Uploading video {file.filename} {token} {user_id}")
 
36
 
37
+ # if token != API_KEY:
38
+ # logger.info(f"Unauthorized {token} {API_KEY}")
39
+ # return JSONResponse(content={"message": "Unauthorized", "status": 401})
40
+
41
+ logger.info("reading contents")
42
  contents = await file.read()
43
+
44
+ # Save the file to the local directory
45
+ logger.info("saving file")
46
  with open(file.filename, "wb") as f:
47
  f.write(contents)
48
 
49
+ logger.info(f"file saved {file.filename}")
50
 
51
  # Create a clone of the file with content already read
52
  background_tasks.add_task(process_video, file.filename, vitpose, user_id)
tasks.py CHANGED
@@ -10,12 +10,12 @@ logger = logging.getLogger(__name__)
10
 
11
  def process_video(video_path: str,vitpose: VitPose,user_id: str):
12
 
13
- logger.info(f"Processing video {video_path}")
14
 
15
 
16
  new_file_name = video_path.split(".")[0] + "edited." + video_path.split(".")[1]
17
  new_file_name = os.path.join("static", new_file_name)
18
- logger.info(f"New file name {new_file_name}")
19
 
20
  vitpose.output_video_path = new_file_name
21
  annotated_frames = vitpose.run(video_path)
 
10
 
11
  def process_video(video_path: str,vitpose: VitPose,user_id: str):
12
 
13
+ logger.info(f"starting task {video_path}")
14
 
15
 
16
  new_file_name = video_path.split(".")[0] + "edited." + video_path.split(".")[1]
17
  new_file_name = os.path.join("static", new_file_name)
18
+ logger.info(f"new file name {new_file_name}")
19
 
20
  vitpose.output_video_path = new_file_name
21
  annotated_frames = vitpose.run(video_path)