Update main.py

main.py

@@ -43,7 +43,8 @@ def generate_mesh(image_path:str,
                   input_folder:str=None,
                   output_folder:str=None,
                   input_s3_id:str='input_image.png',
-                  output_s3_id:str='output_mesh.obj'
+                  output_s3_id:str='output_mesh.obj',
+                  output_video_s3_id:str=None
                   ) ->  Tuple[Optional[str], Optional[str]] :
 
     print('Process start')
@@ -76,11 +77,18 @@ def generate_mesh(image_path:str,
         output_s3_key = output_folder + '/' + output_s3_id
     else:
         output_s3_key = output_s3_id
-    print(input_s3_key ,output_s3_key)
+        output_video_s3_key = output_video_s3_id
 
     input_s3_url = upload_to_s3(image_path, bucket_name, input_s3_key)
     output_s3_url = upload_to_s3(output_file_path, bucket_name, output_s3_key)
+
+    if output_video_path != None:
+        if output_folder != None:
+            output_video_s3_key = output_folder + '/' + output_video_s3_id
+        else:
+            output_video_s3_key = output_video_s3_id
+        output_video_s3_url = upload_to_s3(output_video_path, bucket_name, output_video_s3_key)
     
-    print(f'Files uploaded to S3:\nInput Image: {input_s3_url}\nOutput Mesh: {output_s3_url}')
+    print(f'Files uploaded to S3:\nInput Image: {input_s3_url}\nOutput Mesh: {output_s3_url}\nOutput Video: {output_video_s3_url}')
  
     return output_file_path ,output_video_path