reencoding fix + minor changes

src/processing/video_render.py

@@ -1,13 +1,13 @@
+"""
+Video Rendering module for courtline detection and video reencoding.
+"""
 import cv2 as cv
 import random
 import os
 import numpy as np
 import subprocess
 import sys
 
-# pass in homo matrix +  + 
-# implement video reencoding 
-
 class VideoRender:
     def __init__(self, homography):
         self._TRUE_PATH = os.path.join('data','true_map.png')
@@ -58,7 +58,7 @@ def render_video(self,states:list,players:dict,filename:str,fps:int=30):
         height, width, _ = background.shape
 
         # Initialize the video writer
-        fourcc = cv.VideoWriter_fourcc(*'H264')
+        fourcc = cv.VideoWriter_fourcc(*'mp4v')
         video_writer = cv.VideoWriter(filename, fourcc, fps, (width,height))
 
         # Define initial positions for each player
@@ -114,6 +114,7 @@ def render_video(self,states:list,players:dict,filename:str,fps:int=30):
         # Release the video writer
         video_writer.release()
 
+
     def _transform_point(self,x:float,y:float):
         '''
         Applies court homography to single point

src/processrunner.py

@@ -77,8 +77,11 @@ def run(self):
         self.run_general_detect()
         self.run_team_detect()
         self.run_shot_detect()
+        print('G, T, S detect fine')
         self.run_courtline_detect()
+        print('courtline detect fine')
         self.run_video_render()
+        print('video render fine')
 
 
     def get_results(self):

src/view/app.py

@@ -29,6 +29,11 @@
 
 
 def process_video(video_file):
+    '''
+    Takes in a mp4 file at video_file and uploads it to the backend, then stores
+    the processed video name into session state
+    Temporarily: stores the processed video into tmp/user_upload.mp4
+    '''
     if video_file is None:
         return False
     response = requests.post(SERVER_URL+"upload",
@@ -104,6 +109,7 @@ def results_page():
 
     st.button(label='Back to Home', on_click=change_state, args=(0,),type="primary")
 
+
 def tips_page():
     '''
     Loads tips page